def test_trace_load():
""" Test loading Trace """
t = Trace(variables=['step', 'r', 'q'])
np.random.seed(42)
qq = np.random.rand(5)
t.record(step=1, r=[1.0, 2.0], q=qq)
fname = t.save('trace', final=True)
# load in a new trace instance
t2 = Trace()
t2.load(fname)
assert 'r' in t2.vars
assert 'q' in t2.vars
assert len(t2['r']) == 1
assert_equal(t2['r'][0], [1.0, 2.0])
assert_equal(t2['q'][0], qq)
os.remove(fname)
def test_trace_save():
""" Test saving Trace """
t = Trace(variables=['step', 'r', 'q'])
np.random.seed(42)
qq = np.random.rand(5)
t.record(step=1, r=[1.0, 2.0], q=qq)
fname = t.save('trace')
f = h5py.File(fname, 'r')
for k in t.vars:
assert k in f
assert_equal(f['r'][0], [1.0, 2.0])
assert_equal(f['q'][0], qq)
os.remove(fname)
