def test_reset_timeout_async_vector_env(shared_memory): env_fns = [make_slow_env(0.3, i) for i in range(4)] with pytest.raises(TimeoutError): try: env = AsyncVectorEnv(env_fns, shared_memory=shared_memory) env.reset_async() env.reset_wait(timeout=0.1) finally: env.close(terminate=True)
def test_reset_out_of_order_async_vector_env(shared_memory): env_fns = [make_env("CartPole-v1", i) for i in range(4)] with pytest.raises(NoAsyncCallError): try: env = AsyncVectorEnv(env_fns, shared_memory=shared_memory) env.reset_wait() except NoAsyncCallError as exception: assert exception.name == "reset" raise finally: env.close(terminate=True) with pytest.raises(AlreadyPendingCallError): try: env = AsyncVectorEnv(env_fns, shared_memory=shared_memory) actions = env.action_space.sample() env.reset() env.step_async(actions) env.reset_async() except NoAsyncCallError as exception: assert exception.name == "step" raise finally: env.close(terminate=True)