Python make Examples

Example #1

File: test_registration.py Project: Ji4chenLi/Multi-Task-Batch-RL

def test_make_deprecated():
    try:
        envs.make('Humanoid-v0')
    except error.Error:
        pass
    else:
        assert False

Example #2

File: test_envs.py Project: Ji4chenLi/Multi-Task-Batch-RL

def test_random_rollout():
    for env in [envs.make('CartPole-v0'), envs.make('FrozenLake-v0')]:
        agent = lambda ob: env.action_space.sample()
        ob = env.reset()
        for _ in range(10):
            assert env.observation_space.contains(ob)
            a = agent(ob)
            assert env.action_space.contains(a)
            (ob, _reward, done, _info) = env.step(a)
            if done: break

Example #3

File: test_registration.py Project: Ji4chenLi/Multi-Task-Batch-RL

def test_make():
    env = envs.make('CartPole-v0')
    assert env.spec.id == 'CartPole-v0'
    assert isinstance(env.unwrapped, cartpole.CartPoleEnv)