Skip to content

Commit 8889302

Browse files
committed
update _worker test
1 parent d2ff6b3 commit 8889302

1 file changed

Lines changed: 12 additions & 12 deletions

File tree

tests/test_env/test_vec_env/test_async_env.py

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -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

6666
def main_control(parent_pipe, child_pipe):

0 commit comments

Comments
 (0)