(env_id, n_envs, vec_env_cls, wrapper_class)
| 39 | @pytest.mark.parametrize("vec_env_cls", [None, SubprocVecEnv]) |
| 40 | @pytest.mark.parametrize("wrapper_class", [None, gym.wrappers.RecordEpisodeStatistics]) |
| 41 | def test_make_vec_env(env_id, n_envs, vec_env_cls, wrapper_class): |
| 42 | env = make_vec_env(env_id, n_envs, vec_env_cls=vec_env_cls, wrapper_class=wrapper_class, monitor_dir=None, seed=0) |
| 43 | |
| 44 | assert env.num_envs == n_envs |
| 45 | |
| 46 | if vec_env_cls is None: |
| 47 | assert isinstance(env, DummyVecEnv) |
| 48 | if wrapper_class is not None: |
| 49 | assert isinstance(env.envs[0], wrapper_class) |
| 50 | else: |
| 51 | assert isinstance(env.envs[0], Monitor) |
| 52 | else: |
| 53 | assert isinstance(env, SubprocVecEnv) |
| 54 | # Kill subprocesses |
| 55 | env.close() |
| 56 | |
| 57 | |
| 58 | def test_make_vec_env_func_checker(): |
nothing calls this directly
no test coverage detected
searching dependent graphs…