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)
评论列表
文章目录