mirror of
https://github.com/Farama-Foundation/Gymnasium.git
synced 2025-08-19 13:32:03 +00:00
d337f4e571
* fix double reset, as suggested by @jietang * better floors and ceilings * add convenience methods to monitor * add wrappers to gym namespace * allow playing Atari games, with potentially more coming in the future * simplify example in docs * Move play out of the Env * fix tests * no more deprecation warnings * remove env.monitor * monitor simplification * monitor simplifications * monitor related fixes * a few changes suggested by linter * timestep_limit fixes * keep track of gym env variables for future compatibility * timestep_limit => max_episode_timesteps * don't apply TimeLimit wrapper in make for VNC envs * Respect old timestep_limit argument * Pass max_episode_seconds through registration * Don't include deprecation warnings yet
51 lines
1.3 KiB
Python
51 lines
1.3 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.unwrapped, cartpole.CartPoleEnv)
|
|
|
|
def test_make_deprecated():
|
|
try:
|
|
envs.make('Humanoid-v0')
|
|
except error.Error:
|
|
pass
|
|
else:
|
|
assert False
|
|
|
|
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') # must match an env name but not the version above
|
|
except error.DeprecatedEnv:
|
|
pass
|
|
else:
|
|
assert False
|
|
|
|
try:
|
|
registry.spec('Unknown-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 '{}'.format(e), 'Unexpected message: {}'.format(e)
|
|
else:
|
|
assert False
|