test_create_disp_dat.py 文件源码

def test_write_data(tmpdir):
    """test writing data to disp.dat
    """
    from create_disp_dat import open_dispout
    from create_disp_dat import process_timestep_data
    import struct
    from pytest import approx

    fname = tmpdir.join('testdata.dat')
    dispout = open_dispout(fname.strpath)
    data = []
    data.append([float(0.0), float(0.1), float(0.2), float(0.3)])
    data.append([float(1.0), float(1.1), float(1.2), float(1.3)])
    process_timestep_data(data, dispout, writenode=True)
    dispout.close()

    with open(fname, 'rb') as f:
        d = struct.unpack(8 * 'f', f.read(4 * 8))

    assert d[0] == 0.0
    assert d[1] == 1.0
    assert d[2] == approx(0.1)
    assert d[3] == approx(1.1)
    assert d[7] == approx(1.3)