Python EnvRegistry Examples

Example #1

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

Example #2

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

Example #3

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

Example #4

File: test_registration.py Project: MatildeDP/DDPG-and-PID-in-Lunar-Lander-Env

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

Example #5

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")

Example #6

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")

Example #7

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

Example #8

File: __init__.py Project: datavaluepeople/truman

"""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