update _worker test

Author: huangshiyu13

openrl/envs/vec_env/async_venv.py

@@ -757,15 +757,14 @@ def prepare_obs(observation):
             )
             observation = None
         return observation
+
     if parent_pipe is not None:
         parent_pipe.close()
     try:
         while True:
             command, data = pipe.recv()
             print(command)
 
-
-
             if command == "reset":
                 result = env.reset(**data)
 

tests/test_env/test_vec_env/test_async_env.py

@@ -16,16 +16,16 @@
 
 """"""
 
+import multiprocessing as mp
 import os
 import sys
 
 import pytest
 from gymnasium.wrappers import EnvCompatibility
 
 from openrl.envs.toy_envs import make_toy_envs
-from openrl.envs.vec_env.async_venv import AsyncVectorEnv
-from openrl.envs.vec_env.async_venv import _worker
-import multiprocessing as mp
+from openrl.envs.vec_env.async_venv import AsyncVectorEnv, _worker
+
 
 class CustomEnvCompatibility(EnvCompatibility):
     def reset(self, **kwargs):
@@ -62,37 +62,40 @@ def assert_env_name(env, env_name):
 #     env.set_attr("metadata", {"name": env_name_new})
 #     env.exec_func(assert_env_name, indices=None, env_name=env_name_new)
 
-def main_control(parent_pipe,child_pipe):
+
+def main_control(parent_pipe, child_pipe):
     child_pipe.close()
 
-    parent_pipe.send(("reset", {"seed":0}))
+    parent_pipe.send(("reset", {"seed": 0}))
     result, success = parent_pipe.recv()
     assert success, result
 
     parent_pipe.send(("step", [0]))
     result, success = parent_pipe.recv()
     assert success, result
 
-    parent_pipe.send(("_call", ("render",[],{})))
+    parent_pipe.send(("_call", ("render", [], {})))
     result, success = parent_pipe.recv()
     assert success, result
 
     parent_pipe.send(("_setattr", ("metadata", {"name": "IdentityEnvNew"})))
     result, success = parent_pipe.recv()
     assert success, result
 
-    parent_pipe.send(("_func_exec",(assert_env_name,None,[],{"env_name":"IdentityEnvNew"})))
+    parent_pipe.send(
+        ("_func_exec", (assert_env_name, None, [], {"env_name": "IdentityEnvNew"}))
+    )
     result, success = parent_pipe.recv()
     assert success, result
 
-    parent_pipe.send(("close",None))
+    parent_pipe.send(("close", None))
     result, success = parent_pipe.recv()
     assert success, result
 
 
 @pytest.mark.unittest
 def test_worker():
-    for auto_reset in [True,False]:
+    for auto_reset in [True, False]:
         ctx = mp.get_context(None)
         parent_pipe, child_pipe = ctx.Pipe()
 
@@ -101,17 +104,12 @@ def test_worker():
         process = ctx.Process(
             target=main_control,
             name="test",
-            args=(
-                parent_pipe,
-                child_pipe
-            ),
+            args=(parent_pipe, child_pipe),
         )
         process.daemon = True
         process.start()
         _worker(0, init_envs()[0], child_pipe, None, False, error_queue, auto_reset)
 
 
-
-
 if __name__ == "__main__":
     sys.exit(pytest.main(["-sv", os.path.basename(__file__)]))