diff --git a/tests/test_env_edit.py b/tests/test_env_edit.py new file mode 100644 index 0000000000000000000000000000000000000000..2d8a4e087a0ab1d4425c05cf3153abcded8a8ceb --- /dev/null +++ b/tests/test_env_edit.py @@ -0,0 +1,15 @@ + +from flatland.envs.rail_env import RailEnv +# from flatland.envs.agent_utils import EnvAgent +from flatland.envs.agent_utils import EnvAgentStatic + + +def test_load_env(): + env = RailEnv(10, 10) + env.load("env-data/tests/test-10x10.mpk") + + agent_static = EnvAgentStatic((0, 0), 2, (5, 5)) + env.add_agent_static(agent_static) + assert env.get_num_agents() == 1 + +