import blimpy as bl
import pytest
from tests.data import voyager_fil, voyager_h5


def test_read_fns():
    """ These read functions are currently not implemented. """
    a = bl.Waterfall(voyager_fil)
    b = bl.Waterfall(voyager_h5)
    with pytest.raises(NotImplementedError):
        a.container.read_all()
        a.container.read_row(0)
        a.container.read_rows(0, 2)

        b.container.read_all()
        b.container.read_row(0)
        b.container.read_rows(0, 2)


def test_file_wrapper_open_file():
    from blimpy.file_wrapper import open_file
    open_file(voyager_h5)
    open_file(voyager_fil)

    with pytest.raises(NotImplementedError):
        open_file('run_tests.sh')


if __name__ == "__main__":
    test_read_fns()
    test_file_wrapper_open_file()
import blimpy as bl
import pytest
from tests.data import voyager_fil, voyager_h5, here


def test_read_fns():
    """ These read functions are currently not implemented. """
    a = bl.Waterfall(voyager_fil)
    b = bl.Waterfall(voyager_h5)
    with pytest.raises(NotImplementedError):
        a.container.read_all()
        a.container.read_row(0)
        a.container.read_rows(0, 2)

        b.container.read_all()
        b.container.read_row(0)
        b.container.read_rows(0, 2)


def test_file_wrapper_open_file():
    from blimpy.file_wrapper import open_file
    open_file(voyager_h5)
    open_file(voyager_fil)

    with pytest.raises(NotImplementedError):
        open_file(here + '/run_tests.sh')


if __name__ == "__main__":
    test_read_fns()
    test_file_wrapper_open_file()
make test_file_wrapper_open_file robust