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)