def simple_tests():
    np.random.seed(222)
    dataset = DataSet(width=2, height=3,
                      rng=np.random.RandomState(42),
                      max_steps=6, phi_length=4)
    for i in range(10):
        img = np.random.randint(0, 256, size=(3, 2))
        action = np.random.randint(16)
        reward = np.random.random()
        terminal = False
        if .05 < np.random.random():
            terminal = True
        print 'img', img
        dataset.add_sample(img, action, reward, terminal)
        print "I", dataset.imgs
        print "A", dataset.actions
        print "R", dataset.rewards
        print "T", dataset.terminal
        print "SIZE", dataset.size
        print
    print "LAST PHI", dataset.last_phi()
    print
    print 'BATCH', dataset.random_batch(2)
dataset/ETHPy150Open spragunr/deep_q_rl/deep_q_rl/ale_data_set.py/simple_tests/swapped operands of `<`