File size: 1,101 Bytes
05c9ac2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
from mlagents_envs.envs.unity_aec_env import UnityAECEnv
from mlagents_envs.envs.unity_parallel_env import UnityParallelEnv
from simple_test_envs import SimpleEnvironment, MultiAgentEnvironment
from pettingzoo.test import api_test, parallel_api_test
NUM_TEST_CYCLES = 100
def test_single_agent_aec():
unity_env = SimpleEnvironment(["test_single"])
env = UnityAECEnv(unity_env)
api_test(env, num_cycles=NUM_TEST_CYCLES, verbose_progress=False)
def test_multi_agent_aec():
unity_env = MultiAgentEnvironment(["test_multi_1", "test_multi_2"], num_agents=2)
env = UnityAECEnv(unity_env)
api_test(env, num_cycles=NUM_TEST_CYCLES, verbose_progress=False)
def test_single_agent_parallel():
unity_env = SimpleEnvironment(["test_single"])
env = UnityParallelEnv(unity_env)
parallel_api_test(env, num_cycles=NUM_TEST_CYCLES)
def test_multi_agent_parallel():
unity_env = MultiAgentEnvironment(
["test_multi_1", "test_multi_2", "test_multi_3"], num_agents=3
)
env = UnityParallelEnv(unity_env)
parallel_api_test(env, num_cycles=NUM_TEST_CYCLES)
|