def test_simulate(self):
env_name = "l2rpn_case14_sandbox"
with warnings.catch_warnings():
warnings.filterwarnings("ignore")
env1 = grid2op.make(env_name, test=True)
env2 = grid2op.make(env_name, test=True)
multi_env = BaseMultiProcessEnvironment([env1, env2])
obss = multi_env.reset()
# simulate
actions = [env1.action_space(), env2.action_space()]
sim_obss, sim_rs, sim_ds, sim_is = multi_env.simulate(actions)
multi_env.close()
env1.close()
env2.close()
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()