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)