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("Test-v1000") except error.UnregisteredEnv: 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("“Breakout-v0”") except error.Error as e: assert "Malformed environment ID" in f"{e}", f"Unexpected message: {e}" 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
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
def test_default_lookups(): registry = registration.EnvRegistry() registry.register("test/Test") with pytest.raises(error.DeprecatedEnv): registry.spec("test/Test-v0") # Lookup default registry.spec("test/Test")
def test_versioned_lookups(): registry = registration.EnvRegistry() registry.register("test/Test-v5") with pytest.raises(error.VersionNotFound): registry.spec("test/Test-v9") with pytest.raises(error.DeprecatedEnv): registry.spec("test/Test-v4") assert registry.spec("test/Test-v5")
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
"""Init and create global env registry.""" from gym.envs import registration from truman import history # noqa from truman._version import get_versions from truman.run.interface import run # noqa # Global registry registry = registration.EnvRegistry() # import all env modules, these contain the `register` calls which register envs to registry from truman import time_period_step # noqa __version__ = get_versions()["version"] del get_versions