From dde6d690cd53bf26a5596c35ad68812bce8588b6 Mon Sep 17 00:00:00 2001
From: hagrid67 <jdhwatson@gmail.com>
Date: Tue, 28 May 2019 23:19:34 +0100
Subject: [PATCH] added test to load msgpack env and add agent_static
---
tests/test_env_edit.py | 15 +++++++++++++++
1 file changed, 15 insertions(+)
create mode 100644 tests/test_env_edit.py
diff --git a/tests/test_env_edit.py b/tests/test_env_edit.py
new file mode 100644
index 0000000..2d8a4e0
--- /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
+
+
--
GitLab