def test_pettinzoo_pad_action_space():
_env = simple_world_comm_v2.env()
wrapped_env = pad_action_space_v0(_env)
api_test.api_test(wrapped_env)
seed_test.seed_test(
lambda: sticky_actions_v0(simple_world_comm_v2.env(), 0.5), 100)
def test_pettingzoo_parallel_env():
_env = simple_world_comm_v2.parallel_env()
wrapped_env = pad_action_space_v0(_env)
parallel_test.parallel_play_test(wrapped_env)
def create_env(args):
if args[env_name] == "simple_speaker_listener":
env = simple_speaker_listener_v3.env()
env = supersuit.pad_action_space_v0(env)
env = supersuit.pad_observations_v0(env)
return env
x_size=5,
y_size=10),
supersuit.resize_v0(dtype_v0(knights_archers_zombies_v4.env(), np.uint8),
x_size=5,
y_size=10,
linear_interp=True),
supersuit.dtype_v0(knights_archers_zombies_v4.env(), np.int32),
supersuit.flatten_v0(knights_archers_zombies_v4.env()),
supersuit.reshape_v0(knights_archers_zombies_v4.env(), (512 * 512, 3)),
supersuit.normalize_obs_v0(dtype_v0(knights_archers_zombies_v4.env(),
np.float32),
env_min=-1,
env_max=5.0),
supersuit.frame_stack_v1(knights_archers_zombies_v4.env(), 8),
supersuit.pad_observations_v0(knights_archers_zombies_v4.env()),
supersuit.pad_action_space_v0(knights_archers_zombies_v4.env()),
supersuit.black_death_v0(knights_archers_zombies_v4.env()),
supersuit.agent_indicator_v0(knights_archers_zombies_v4.env(), True),
supersuit.agent_indicator_v0(knights_archers_zombies_v4.env(), False),
supersuit.reward_lambda_v0(knights_archers_zombies_v4.env(),
lambda x: x / 10),
supersuit.clip_reward_v0(knights_archers_zombies_v4.env()),
supersuit.clip_actions_v0(prison_v2.env(continuous=True)),
supersuit.frame_skip_v0(knights_archers_zombies_v4.env(), 4),
supersuit.sticky_actions_v0(knights_archers_zombies_v4.env(), 0.75),
supersuit.delay_observations_v0(knights_archers_zombies_v4.env(), 3),
]
@pytest.mark.parametrize("env", wrappers)
def test_pettingzoo_aec_api(env):
def create_env(args):
env = simple_speaker_listener_v3.env()
env = pad_action_space_v0(env)
env = pad_observations_v0(env)
return env
base_act_spaces = {"a_{}".format(idx): Discrete(5) for idx in range(2)}
return PettingzooWrap(DummyEnv(base_obs, base_obs_space, base_act_spaces))
wrappers = [
supersuit.color_reduction_v0(new_dummy(), "R"),
supersuit.resize_v0(dtype_v0(new_dummy(), np.uint8), x_size=5, y_size=10),
supersuit.resize_v0(dtype_v0(new_dummy(), np.uint8), x_size=5, y_size=10, linear_interp=True),
supersuit.dtype_v0(new_dummy(), np.int32),
supersuit.flatten_v0(new_dummy()),
supersuit.reshape_v0(new_dummy(), (64, 3)),
supersuit.normalize_obs_v0(new_dummy(), env_min=-1, env_max=5.0),
supersuit.frame_stack_v1(new_dummy(), 8),
supersuit.pad_observations_v0(new_dummy()),
supersuit.pad_action_space_v0(new_dummy()),
supersuit.black_death_v1(new_dummy()),
supersuit.agent_indicator_v0(new_dummy(), True),
supersuit.agent_indicator_v0(new_dummy(), False),
supersuit.reward_lambda_v0(new_dummy(), lambda x: x / 10),
supersuit.clip_reward_v0(new_dummy()),
supersuit.clip_actions_v0(new_continuous_dummy()),
supersuit.frame_skip_v0(new_dummy(), 4),
supersuit.sticky_actions_v0(new_dummy(), 0.75),
supersuit.delay_observations_v0(new_dummy(), 3),
supersuit.max_observation_v0(new_dummy(), 3),
]
@pytest.mark.parametrize("env", wrappers)
def test_basic_wrappers(env):
def env(**kwargs):
env = base_env.parallel_env(**kwargs)
env = pad_observations_v0(env)
env = pad_action_space_v0(env)
env = _ParallelPettingZooEnv(env)
return env
supersuit.sticky_actions_v0(generated_agents_parallel_v0.parallel_env(),
0.75),
supersuit.delay_observations_v0(
generated_agents_parallel_v0.parallel_env(), 3),
supersuit.max_observation_v0(generated_agents_parallel_v0.parallel_env(),
3),
]
@pytest.mark.parametrize("env", parallel_wrappers)
def test_pettingzoo_parallel_api_gen(env):
parallel_test.parallel_api_test(env, num_cycles=50)
wrapper_fns = [
lambda: supersuit.pad_action_space_v0(generated_agents_parallel_v0.env()),
lambda: supersuit.pad_observations_v0(generated_agents_parallel_v0.env()),
lambda: supersuit.agent_indicator_v0(generated_agents_parallel_v0.env()),
lambda: supersuit.vectorize_aec_env_v0(generated_agents_parallel_v0.env(),
2),
lambda: supersuit.pad_action_space_v0(generated_agents_parallel_v0.
parallel_env()),
lambda: supersuit.pad_observations_v0(generated_agents_parallel_v0.
parallel_env()),
lambda: supersuit.agent_indicator_v0(generated_agents_parallel_v0.
parallel_env()),
lambda: supersuit.pettingzoo_env_to_vec_env_v1(generated_agents_parallel_v0
.parallel_env()),
]
np.uint8),
x_size=5,
y_size=10,
linear_interp=True,
),
supersuit.dtype_v0(knights_archers_zombies_v10.env(), np.int32),
supersuit.flatten_v0(knights_archers_zombies_v10.env()),
supersuit.reshape_v0(knights_archers_zombies_v10.env(vector_state=False),
(512 * 512, 3)),
supersuit.normalize_obs_v0(dtype_v0(knights_archers_zombies_v10.env(),
np.float32),
env_min=-1,
env_max=5.0),
supersuit.frame_stack_v1(combined_arms_v6.env(), 8),
supersuit.pad_observations_v0(simple_world_comm_v2.env()),
supersuit.pad_action_space_v0(simple_world_comm_v2.env()),
supersuit.black_death_v3(combined_arms_v6.env()),
supersuit.agent_indicator_v0(knights_archers_zombies_v10.env(), True),
supersuit.agent_indicator_v0(knights_archers_zombies_v10.env(), False),
supersuit.reward_lambda_v0(knights_archers_zombies_v10.env(),
lambda x: x / 10),
supersuit.clip_reward_v0(combined_arms_v6.env()),
supersuit.nan_noop_v0(knights_archers_zombies_v10.env(), 0),
supersuit.nan_zeros_v0(knights_archers_zombies_v10.env()),
supersuit.nan_random_v0(chess_v5.env()),
supersuit.nan_random_v0(knights_archers_zombies_v10.env()),
supersuit.frame_skip_v0(combined_arms_v6.env(), 4),
supersuit.sticky_actions_v0(combined_arms_v6.env(), 0.75),
supersuit.delay_observations_v0(combined_arms_v6.env(), 3),
supersuit.max_observation_v0(knights_archers_zombies_v10.env(), 3),
]
