Source code for torchrl.envs.test_env_utils
import pytest
import gym
from torchrl.envs import make_gym_env
[docs]@pytest.mark.parametrize('spec_id', [
'Acrobot-v1',
'CartPole-v1',
'MountainCar-v0',
'MountainCarContinuous-v0',
'Pendulum-v0',
])
def test_make_gym_env(spec_id: str):
env = make_gym_env(spec_id)
assert isinstance(env, gym.Env)
assert env.spec.id == spec_id
env.close()