def test_read_direct():
dset = ReplayDataset("test.hdf5", STATE_SHAPE, dset_size=1000,
overwrite=True)
batch_size = 32
s = np.zeros((batch_size,)+STATE_SHAPE)
a = np.zeros(batch_size)
r = np.zeros(batch_size)
sp = np.zeros((batch_size,)+STATE_SHAPE)
states = random_fill(dset, 100)
dset.sample_direct(s, a, r, sp, batch_size)
diff = [np.linalg.norm(s[0] - states[i]) for i in range(100)]
assert(min(diff) == 0)
def test_read_direct():
dset = ReplayDataset("test.hdf5",
STATE_SHAPE,
dset_size=1000,
overwrite=True)
batch_size = 32
s = np.zeros((batch_size, ) + STATE_SHAPE)
a = np.zeros(batch_size)
r = np.zeros(batch_size)
sp = np.zeros((batch_size, ) + STATE_SHAPE)
states = random_fill(dset, 100)
dset.sample_direct(s, a, r, sp, batch_size)
diff = [np.linalg.norm(s[0] - states[i]) for i in range(100)]
assert (min(diff) == 0)
