@@ -49,18 +49,18 @@ def assert_env_name(env, env_name):
4949 assert env .metadata ["name" ].__name__ == env_name
5050
5151
52- # @pytest.mark.unittest
53- # def test_async_env():
54- # env_name = "IdentityEnv"
55- # env = AsyncVectorEnv(init_envs(), shared_memory=True)
56- # assert (
57- # env._env_name == env_name
58- # ), "AsyncVectorEnv should have the same metadata as the wrapped env"
59- # env.exec_func(assert_env_name, indices=None, env_name=env_name)
60- # env.call("render")
61- # env_name_new = "IdentityEnvNew"
62- # env.set_attr("metadata", {"name": env_name_new})
63- # env.exec_func(assert_env_name, indices=None, env_name=env_name_new)
52+ @pytest .mark .unittest
53+ def test_async_env ():
54+ env_name = "IdentityEnv"
55+ env = AsyncVectorEnv (init_envs (), shared_memory = True )
56+ assert (
57+ env ._env_name == env_name
58+ ), "AsyncVectorEnv should have the same metadata as the wrapped env"
59+ env .exec_func (assert_env_name , indices = None , env_name = env_name )
60+ env .call ("render" )
61+ env_name_new = "IdentityEnvNew"
62+ env .set_attr ("metadata" , {"name" : env_name_new })
63+ env .exec_func (assert_env_name , indices = None , env_name = env_name_new )
6464
6565
6666def main_control (parent_pipe , child_pipe ):
0 commit comments