Gymnasium/gym/tests/test_core.py

from gym import core

class ArgumentEnv(core.Env):
    calls = 0

    def __init__(self, arg):
        self.calls += 1
        self.arg = arg

def test_env_instantiation():
    # This looks like a pretty trivial, but given our usage of
    # __new__, it's worth having.
    env = ArgumentEnv('arg')
    assert env.arg == 'arg'
    assert env.calls == 1
Improve closer implementation and docstrings (#126) * Improve auto close implementation - Register all envs at initialization time, not just ones created via make - Simplify names and add more documentation on interface - Move closer instances into the relevant modules review-requested: @jietang * Close environments in the tests This isn't strictly needed, but means there are fewer Doom subprocesses hanging around while the tests run. * Use 4 space indent in comment * Improve docstrings in core * Don't pass through args to __new__ The __init__ method gets called once __new__ returns, so these arguments are either ignored (Python 2) or result in an error (Python 3). The __init__ method automatically gets called with the correct arguments. * Fixup comments 2016-05-27 12:16:35 -07:00			`from gym import core`

			`class ArgumentEnv(core.Env):`
			`calls = 0`

			`def __init__(self, arg):`
			`self.calls += 1`
			`self.arg = arg`

			`def test_env_instantiation():`
			`# This looks like a pretty trivial, but given our usage of`
			`# __new__, it's worth having.`
			`env = ArgumentEnv('arg')`
			`assert env.arg == 'arg'`
			`assert env.calls == 1`