class TestNormalizedGym:
def setup_method(self):
self.env = MetaRLEnv(
normalize(gym.make('Pendulum-v0'),
normalize_reward=True,
normalize_obs=True,
flatten_obs=True))
def teardown_method(self):
self.env.close()
def test_does_not_modify_action(self):
a = self.env.action_space.sample()
a_copy = a
self.env.reset()
self.env.step(a)
assert a == a_copy
def test_flatten(self):
for _ in range(10):
self.env.reset()
for _ in range(5):
self.env.render()
action = self.env.action_space.sample()
next_obs, _, done, _ = self.env.step(action)
assert next_obs.shape == self.env.observation_space.low.shape
if done:
break
def test_unflatten(self):
for _ in range(10):
self.env.reset()
for _ in range(5):
action = self.env.action_space.sample()
next_obs, _, done, _ = self.env.step(action)
# yapf: disable
assert (self.env.observation_space.flatten(next_obs).shape
== self.env.observation_space.flat_dim)
# yapf: enable
if done:
break
def test_closes_box2d(self):
metarl_env = MetaRLEnv(env_name='CarRacing-v0')
metarl_env.render()
assert metarl_env.env.viewer is not None
metarl_env.close()
assert metarl_env.env.viewer is None
def test_closes_mujoco(self):
metarl_env = MetaRLEnv(env_name='Ant-v2')
metarl_env.render()
assert metarl_env.env.viewer is not None
metarl_env.close()
assert metarl_env.env.viewer is None