def test_parallel_generated_agents_conversions():
parallel_api_test(aec_to_parallel(generated_agents_parallel_v0.env()))
api_test(parallel_to_aec(generated_agents_parallel_v0.parallel_env()))
env1 = parallel_to_aec(generated_agents_parallel_v0.parallel_env())
env2 = parallel_to_aec(
aec_to_parallel(
parallel_to_aec(generated_agents_parallel_v0.parallel_env())))
check_environment_deterministic(env1, env2, 500)
def test_module(name, env_module, kwargs):
_env = env_module.env(**kwargs)
api_test(_env)
# some atari environments fail this test
if "atari/" not in name:
seed_test(lambda: env_module.env(**kwargs), 50)
render_test(lambda: env_module.env(**kwargs))
if hasattr(env_module, "parallel_env"):
par_env = env_module.parallel_env(**kwargs)
try:
_env.state()
state_test(_env, par_env)
except NotImplementedError:
# no issue if state is simply not implemented
pass
def test_module(name, env_module):
_env = env_module.env()
assert str(_env) == os.path.basename(name)
api_test(_env)
if "classic/" not in name:
parallel_api_test(env_module.parallel_env())
seed_test(env_module.env, 50)
if "classic/" not in name:
max_cycles_test(env_module)
if ("butterfly/" in name) or ("mpe/" in name) or ("magent/" in name):
state_test(_env, env_module.parallel_env())
recreated_env = pickle.loads(pickle.dumps(_env))
api_test(recreated_env)
def test_module(name, env_module):
_env = env_module.env()
assert str(_env) == os.path.basename(name)
api_test(_env)
if "classic/" not in name:
parallel_api_test(env_module.parallel_env())
# seed_test(env_module.env, 50)
# some atari environments fail this test
if "atari/" not in name:
seed_test(env_module.env, 50)
render_test(env_module.env)
if "classic/" not in name:
max_cycles_test(env_module)
if ("butterfly/" in name) or ("mpe/" in name) or ("magent/" in name):
state_test(_env, env_module.parallel_env())
def test_module(env_constr, kwargs):
_env = env_constr(**kwargs)
print(kwargs)
api_test(_env)
def test_generated_agents_parallel():
parallel_api_test(generated_agents_parallel_v0.parallel_env())
api_test(generated_agents_parallel_v0.env())
def test_generated_agents_aec():
api_test(generated_agents_env_v0.env())