Example #1
0
def test_read_structured_omf_file():
    """Takes a file as an input and returns the vecorfield of data it contains
    """
    for i in range(len(filenames)):
        file = filenames[i]
        byte = bytes[i]
        nodes = filenames_nodes[i]
        data = filenames_data_types[i]
        # actual result
        act = omf.read_structured_omf_file(file)
        # all files should be retuned as numpy arrays
        assert type(act) == np.ndarray
        assert len(act) == nodes[0] * nodes[1] * nodes[2]
        assert [int(dim) for dim in act.shape] == [len(act), 3]

        # check binary4, binary8 files return expected data values
        if i < 6:
            exp = omf.read_structured_binary_oommf_data(
                file, byte, nodes, data)
        # check ascii files return expected data values
        else:
            exp = omf.read_structured_ascii_oommf_data(file, byte, nodes)
        assert exp.all() == act.all()

        # test print statements show if debug
        if sysvers == 2:
            debug = [0, 1]
            for db in debug:
                result = StringIO()
                sys.stdout = result
                omf.read_structured_omf_file(file, db)
                result_string = result.getvalue()
                # binary4 & binary8 files
                if debug == 1 and i < 6:
                    floatsizes = [4, 4, 8, 8, 4, 4]
                    floatsize = floatsizes[i]
                    assert result_string == """Number of cells \
(Nx={},Ny={},Nz={})
Expect floats of length {} bytes.
Expect to find data in file {} at position {}.
verification_tag is okay (=> reading byte order correctly)\n"""\
.format(nodes[0], nodes[1], nodes[2], floatsize, file, byte)
                # ascii files
                elif debug == 1 and i > 5:
                    assert result_string == """Number of cells \
(Nx={},Ny={},Nz={})\n""".format(nodes[0], nodes[1], nodes[2])
                # debug = False
                elif debug == 0:
                    assert result_string == ''
Example #2
0
def test_read_structured_omf_file():
    """Takes a file as an input and returns the vecorfield of data it contains
    """
    for i in range(len(filenames)):
        file = filenames[i]
        byte = bytes[i]
        nodes = filenames_nodes[i]
        data = filenames_data_types[i]
        # actual result
        act = omf.read_structured_omf_file(file)
        # all files should be retuned as numpy arrays
        assert type(act) == np.ndarray
        assert len(act) == nodes[0] * nodes[1] * nodes[2]
        assert [int(dim) for dim in act.shape] == [len(act), 3]

        # check binary4, binary8 files return expected data values
        if i < 6:
            exp = omf.read_structured_binary_oommf_data(file, byte, nodes,
                                                        data)
        # check ascii files return expected data values
        else:
            exp = omf.read_structured_ascii_oommf_data(file, byte, nodes)
        assert exp.all() == act.all()

        # test print statements show if debug
        if sysvers == 2:
            debug = [0, 1]
            for db in debug:
                result = StringIO()
                sys.stdout = result
                omf.read_structured_omf_file(file, db)
                result_string = result.getvalue()
                # binary4 & binary8 files
                if debug == 1 and i < 6:
                    floatsizes = [4, 4, 8, 8, 4, 4]
                    floatsize = floatsizes[i]
                    assert result_string == """Number of cells \
(Nx={},Ny={},Nz={})
Expect floats of length {} bytes.
Expect to find data in file {} at position {}.
verification_tag is okay (=> reading byte order correctly)\n"""\
.format(nodes[0], nodes[1], nodes[2], floatsize, file, byte)
                # ascii files
                elif debug == 1 and i > 5:
                    assert result_string == """Number of cells \
(Nx={},Ny={},Nz={})\n""".format(nodes[0], nodes[1], nodes[2])
                # debug = False
                elif debug == 0:
                    assert result_string == ''
Example #3
0
def test_read_structured_oommf_data():
    """ function takes a file, the byte at which data starts, the nodes of the
    data and the data type, returns a vectorfield."""
    for i in range(len(filenames)):
        file = filenames[i]
        byte = bytes[i]
        nodes = filenames_nodes[i]
        data = filenames_data_types[i]
        # actual result
        act = omf.read_structured_oommf_data(file, byte, nodes, data)
        # all files retuned as numpy arrays
        assert type(act) == np.ndarray
        assert len(act) == nodes[0] * nodes[1] * nodes[2]
        assert [int(dim) for dim in act.shape] == [len(act), 3]

        # check binary4, binary8 files return expected data values
        if i < 6:
            exp = omf.read_structured_binary_oommf_data(
                file, byte, nodes, data)
        # check ascii files return expected data values
        else:
            exp = omf.read_structured_ascii_oommf_data(file, byte, nodes)
        assert exp.all() == act.all()

        # test function detects unexpected datatype
        if sysvers == 2:
            data_unknown = data + 'unknown'
            result = StringIO()
            sys.stdout = result
        try:
            omf.read_structured_oommf_data(file, byte, nodes, data + 'unknown')
            x = 0
        except Exception:
            x = 1
            if sysvers == 2:
                result_string = result.getvalue()
                assert result_string == """expected ascii or binary4 or binary8 for\
 datatype, but got {}\n""".format(data_unknown)
        assert x == 1
Example #4
0
def test_read_structured_oommf_data():
    """ function takes a file, the byte at which data starts, the nodes of the
    data and the data type, returns a vectorfield."""
    for i in range(len(filenames)):
        file = filenames[i]
        byte = bytes[i]
        nodes = filenames_nodes[i]
        data = filenames_data_types[i]
        # actual result
        act = omf.read_structured_oommf_data(file, byte, nodes, data)
        # all files retuned as numpy arrays
        assert type(act) == np.ndarray
        assert len(act) == nodes[0] * nodes[1] * nodes[2]
        assert [int(dim) for dim in act.shape] == [len(act), 3]

        # check binary4, binary8 files return expected data values
        if i < 6:
            exp = omf.read_structured_binary_oommf_data(file, byte, nodes,
                                                        data)
        # check ascii files return expected data values
        else:
            exp = omf.read_structured_ascii_oommf_data(file, byte, nodes)
        assert exp.all() == act.all()

        # test function detects unexpected datatype
        if sysvers == 2:
            data_unknown = data+'unknown'
            result = StringIO()
            sys.stdout = result
        try:
            omf.read_structured_oommf_data(file, byte, nodes, data+'unknown')
            x = 0
        except Exception:
            x = 1
            if sysvers == 2:
                result_string = result.getvalue()
                assert result_string == """expected ascii or binary4 or binary8 for\
 datatype, but got {}\n""".format(data_unknown)
        assert x == 1
Example #5
0
def test_read_structured_binary_oommf_data():
    """the actual function takes, as inputs, the filename; its byte at which
    data begins; its nodes values in a tuple; and the datatype (binary4,
    binary8 or ascii). Returns an array of the vectorfield.
    Utilises 'read_verification_tag' function."""

    # test if ascii file given, exception is raised and print statement sent
    for i in range(len(ascii_files)):
        if sysvers == 2:
            result = StringIO()
            sys.stdout = result
        try:
            omf.read_structured_binary_oommf_data(ascii_files[i],
                                                  ascii_bytes[i],
                                                  ascii_nodes[i], b'ascii')
            x = 0
        except NotImplementedError:
            x = 1
            if sysvers == 2:
                result_string = result.getvalue()
                assert result_string == "ascii -oommf data not supported \
here\n"

        assert x == 1

    # test that an unknown data type is detected.
    for i in range(len(filenames)):
        if sysvers == 2:
            result = StringIO()
            sys.stdout = result
        try:
            omf.read_structured_binary_oommf_data(
                filenames[i], bytes[i], filenames_nodes[i],
                filenames_data_types[i] + 'unknown')
            x = 0
        except Exception:
            x = 1
            if sysvers == 2:
                result_string = result.getvalue()
                assert result_string == "unknown datatype (expected  'binary4',\
'binary8' [or 'ascii'], but got {}"\
.format(filenames_data_types[i]+'unknown\n')
        assert x == 1

    # test that incorrect verification tag is detected for binary4 and...
    # ...binary8 files. Uses edited versions of files from 'filenames'...
    # ...which have tags 7.27159209092e+31 and 4.91466545592e+252...
    # ...rather than 1234567.0 and 123456789012345.0 respectively.
    b4_b8_files = [
        os.path.join('..', 'Examples', 'cantedvortextest.omf'),
        os.path.join('..', 'Examples', 'ellipsoidwraptest.omf'),
        os.path.join('..', 'Examples', 'h2hleftedgetest.ohf'),
        os.path.join('..', 'Examples', 'yoyoleftedgetest.ohf'),
        os.path.join('..', 'Examples', 'stdprob3v-regtest.omf'),
        os.path.join('..', 'Examples', 'stdprobatest.omf')
    ]

    for i in range(len(b4_b8_files)):
        if sysvers == 2:
            result = StringIO()
            sys.stdout = result
        try:
            omf.read_structured_binary_oommf_data(b4_b8_files[i], bytes[i],
                                                  filenames_nodes[i],
                                                  filenames_data_types[i])
            x = 0
        except AssertionError:
            x = 1
            if sysvers == 2:
                result_string = result.getvalue()
                # if binary4, tag = 7.27159209092e+31
                if i < 2 or i > 3:
                    assert result_string == """The first item in a binary file is \
meant to be 1234567.0
but it is not. Instead, I read {}.
Cowardly stopping here.\n""".format(7.271592090916425e+31)
            # if binary8, tag = 4.91466545592e+252
                if 1 < i < 4:
                    assert result_string == """The first item in a binary file is \
meant to be 123456789012345.0
but it is not. Instead, I read {}.
Cowardly stopping here.\n""".format(4.91466545592e+252)
        assert x == 1

    # test function output is as expected.
    # actual result
    for i in range(6):
        file = filenames[i]
        byte = bytes[i]
        nodes = filenames_nodes[i]
        data = filenames_data_types[i]
        act = omf.read_structured_binary_oommf_data(file, byte, nodes, data)
        # check result is a numpy array of correct length and shape
        assert type(act) == np.ndarray
        assert len(act) == filenames_nodes[i][0] * filenames_nodes[i][1] *\
            filenames_nodes[i][2]
        assert [int(dim) for dim in act.shape] == [len(act), 3]
        # check data matches that of of orginal function
        exp = omfread_original.read_structured_binary_oommf_data(
            file, byte, nodes, data)
        assert act.all() == exp.all()

    # test verbose effects
    if sysvers == 2:
        for verbose in verboses:
            for i in range(6):
                file = filenames[i]
                byte = bytes[i]
                nodes = filenames_nodes[i]
                data = filenames_data_types[i]
                # the floatsizes of the first six files in 'filenames'
                floatsizes = [4, 4, 8, 8, 4, 4]
                floatsize = floatsizes[i]
                result = StringIO()
                sys.stdout = result
                omf.read_structured_binary_oommf_data(file, byte, nodes, data,
                                                      verbose)
                result_string = result.getvalue()
                if verbose == 0:
                    assert result_string == ''
                else:
                    assert result_string == """Expect floats of length {} bytes.
Expect to find data in file {} at position {}.
verification_tag is okay (=> reading byte order correctly)\n"""\
.format(floatsize, file, byte)
Example #6
0
def test_read_structured_binary_oommf_data():
    """the actual function takes, as inputs, the filename; its byte at which
    data begins; its nodes values in a tuple; and the datatype (binary4,
    binary8 or ascii). Returns an array of the vectorfield.
    Utilises 'read_verification_tag' function."""

    # test if ascii file given, exception is raised and print statement sent
    for i in range(len(ascii_files)):
        if sysvers == 2:
            result = StringIO()
            sys.stdout = result
        try:
            omf.read_structured_binary_oommf_data(ascii_files[i],
                                                  ascii_bytes[i],
                                                  ascii_nodes[i], b'ascii')
            x = 0
        except NotImplementedError:
            x = 1
            if sysvers == 2:
                result_string = result.getvalue()
                assert result_string == "ascii -oommf data not supported \
here\n"
        assert x == 1

    # test that an unknown data type is detected.
    for i in range(len(filenames)):
        if sysvers == 2:
            result = StringIO()
            sys.stdout = result
        try:
            omf.read_structured_binary_oommf_data(filenames[i], bytes[i],
                                                  filenames_nodes[i],
                                                  filenames_data_types[i] +
                                                  'unknown')
            x = 0
        except Exception:
            x = 1
            if sysvers == 2:
                result_string = result.getvalue()
                assert result_string == "unknown datatype (expected  'binary4',\
'binary8' [or 'ascii'], but got {}"\
.format(filenames_data_types[i]+'unknown\n')
        assert x == 1

    # test that incorrect verification tag is detected for binary4 and...
    # ...binary8 files. Uses edited versions of files from 'filenames'...
    # ...which have tags 7.27159209092e+31 and 4.91466545592e+252...
    # ...rather than 1234567.0 and 123456789012345.0 respectively.
    b4_b8_files = [os.path.join('..', 'Examples', 'cantedvortextest.omf'),
                   os.path.join('..', 'Examples', 'ellipsoidwraptest.omf'),
                   os.path.join('..', 'Examples', 'h2hleftedgetest.ohf'),
                   os.path.join('..', 'Examples', 'yoyoleftedgetest.ohf'),
                   os.path.join('..', 'Examples', 'stdprob3v-regtest.omf'),
                   os.path.join('..', 'Examples', 'stdprobatest.omf')]

    for i in range(len(b4_b8_files)):
        if sysvers == 2:
            result = StringIO()
            sys.stdout = result
        try:
            omf.read_structured_binary_oommf_data(b4_b8_files[i], bytes[i],
                                                  filenames_nodes[i],
                                                  filenames_data_types[i])
            x = 0
        except AssertionError:
            x = 1
            if sysvers == 2:
                result_string = result.getvalue()
                # if binary4, tag = 7.27159209092e+31
                if i < 2 or i > 3:
                    assert result_string == """The first item in a binary file is \
meant to be 1234567.0
but it is not. Instead, I read {}.
Cowardly stopping here.\n""".format(7.271592090916425e+31)
            # if binary8, tag = 4.91466545592e+252
                if 1 < i < 4:
                    assert result_string == """The first item in a binary file is \
meant to be 123456789012345.0
but it is not. Instead, I read {}.
Cowardly stopping here.\n""".format(4.91466545592e+252)
        assert x == 1

    # test function output is as expected.
    # actual result
    for i in range(6):
        file = filenames[i]
        byte = bytes[i]
        nodes = filenames_nodes[i]
        data = filenames_data_types[i]
        act = omf.read_structured_binary_oommf_data(file, byte, nodes, data)
        # check result is a numpy array of correct length and shape
        assert type(act) == np.ndarray
        assert len(act) == filenames_nodes[i][0] * filenames_nodes[i][1] *\
            filenames_nodes[i][2]
        assert [int(dim) for dim in act.shape] == [len(act), 3]
        # check data matches that of of orginal function
        exp = omfread_original.read_structured_binary_oommf_data(file, byte,
                                                                 nodes, data)
        assert act.all() == exp.all()

    # test verbose effects
    if sysvers == 2:
        for verbose in verboses:
            for i in range(6):
                file = filenames[i]
                byte = bytes[i]
                nodes = filenames_nodes[i]
                data = filenames_data_types[i]
                # the floatsizes of the first six files in 'filenames'
                floatsizes = [4, 4, 8, 8, 4, 4]
                floatsize = floatsizes[i]
                result = StringIO()
                sys.stdout = result
                omf.read_structured_binary_oommf_data(file, byte, nodes, data,
                                                      verbose)
                result_string = result.getvalue()
                if verbose == 0:
                    assert result_string == ''
                else:
                    assert result_string == """Expect floats of length {} bytes.
Expect to find data in file {} at position {}.
verification_tag is okay (=> reading byte order correctly)\n"""\
.format(floatsize, file, byte)