def test_seeding(self):
nb_env = 2
with warnings.catch_warnings():
warnings.filterwarnings("ignore")
with make("rte_case5_example", test=True) as env:
envs = [env for _ in range(nb_env)]
env.seed(2)
multi_envs1 = BaseMultiProcessEnvironment(envs)
seeds_1 = multi_envs1.get_seeds()
multi_envs1.close()
multi_envs2 = BaseMultiProcessEnvironment(envs)
seeds_2 = multi_envs2.get_seeds()
multi_envs2.close()
env.seed(2)
multi_envs3 = BaseMultiProcessEnvironment(envs)
seeds_3 = multi_envs3.get_seeds()
multi_envs3.close()
assert np.all(seeds_1 == seeds_3)
assert np.any(seeds_1 != seeds_2)
def test_creation_multienv(self):
nb_env = 2
with warnings.catch_warnings():
warnings.filterwarnings("ignore")
with make("rte_case5_example", test=True) as env:
envs = [env for _ in range(nb_env)]
multi_envs = BaseMultiProcessEnvironment(envs)
obss, rewards, dones, infos = multi_envs.step(
[env.action_space() for _ in range(multi_envs.nb_env)])
for ob in obss:
assert isinstance(ob, CompleteObservation)
obss = multi_envs.reset()
for ob in obss:
assert isinstance(ob, CompleteObservation)
# test some actions will not throw errors
multi_envs.set_ff(7 * 288)
multi_envs.set_chunk_size(128)
obss = multi_envs.reset()
seeds = multi_envs.get_seeds()
multi_envs.close()