mirror of
https://github.com/Farama-Foundation/Gymnasium.git
synced 2025-08-18 12:57:38 +00:00
36 lines
1.0 KiB
Python
36 lines
1.0 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
from gym import error, envs
|
||
|
|
from gym.envs import registration
|
||
|
|
from gym.envs.classic_control import cartpole
|
||
|
|
|
||
|
|
def test_make():
|
||
|
|
env = envs.make('CartPole-v0')
|
||
|
|
assert env.spec.id == 'CartPole-v0'
|
||
|
|
assert isinstance(env, cartpole.CartPoleEnv)
|
||
|
|
|
||
|
|
def test_spec():
|
||
|
|
spec = envs.spec('CartPole-v0')
|
||
|
|
assert spec.id == 'CartPole-v0'
|
||
|
|
|
||
|
|
def test_missing_lookup():
|
||
|
|
registry = registration.EnvRegistry()
|
||
|
|
registry.register(id='Test-v0', entry_point=None)
|
||
|
|
registry.register(id='Test-v15', entry_point=None)
|
||
|
|
registry.register(id='Test-v9', entry_point=None)
|
||
|
|
registry.register(id='Other-v100', entry_point=None)
|
||
|
|
try:
|
||
|
|
registry.spec('Test-v1')
|
||
|
|
except error.UnregisteredEnv:
|
||
|
|
pass
|
||
|
|
else:
|
||
|
|
assert False
|
||
|
|
|
||
|
|
def test_malformed_lookup():
|
||
|
|
registry = registration.EnvRegistry()
|
||
|
|
try:
|
||
|
|
registry.spec(u'“Breakout-v0”')
|
||
|
|
except error.Error as e:
|
||
|
|
assert 'malformed environment ID' in e.message, 'Unexpected message: {}'.format(e)
|
||
|
|
else:
|
||
|
|
assert False
|