From a186d58bbd4360d807f645225a14d0a1b6739cbe Mon Sep 17 00:00:00 2001
From: nimishsantosh107 <nimishsantosh107@icloud.com>
Date: Fri, 10 Sep 2021 15:24:05 +0530
Subject: [PATCH] test persistence - added, file name typo in tests fixed
---
tests/test_flatland_envs_persistence.py | 36 +++++++++++++++++++
....py => test_flatland_rail_agent_status.py} | 0
2 files changed, 36 insertions(+)
create mode 100644 tests/test_flatland_envs_persistence.py
rename tests/{test_flaltland_rail_agent_status.py => test_flatland_rail_agent_status.py} (100%)
diff --git a/tests/test_flatland_envs_persistence.py b/tests/test_flatland_envs_persistence.py
new file mode 100644
index 00000000..7e26389f
--- /dev/null
+++ b/tests/test_flatland_envs_persistence.py
@@ -0,0 +1,36 @@
+import numpy as np
+
+from flatland.envs.rail_env import RailEnv
+from flatland.envs.rail_generators import rail_from_grid_transition_map
+from flatland.envs.line_generators import sparse_line_generator
+from flatland.utils.simple_rail import make_simple_rail
+from flatland.envs.persistence import RailEnvPersister
+
+def test_load_new():
+
+ filename = "test_load_new.pkl"
+
+ rail, rail_map, optionals = make_simple_rail()
+ n_agents = 2
+ env_initial = RailEnv(width=rail_map.shape[1], height=rail_map.shape[0], rail_generator=rail_from_grid_transition_map(rail, optionals),
+ line_generator=sparse_line_generator(), number_of_agents=n_agents)
+ env_initial.reset(False, False)
+
+ rails_initial = env_initial.rail.grid
+ agents_initial = env_initial.agents
+
+ RailEnvPersister.save(env_initial, filename)
+
+ env_loaded, _ = RailEnvPersister.load_new(filename)
+
+ rails_loaded = env_loaded.rail.grid
+ agents_loaded = env_loaded.agents
+
+ assert np.all(np.array_equal(rails_initial, rails_loaded))
+ assert agents_initial == agents_loaded
+
+def main():
+ pass
+
+if __name__ == "__main__":
+ main()
diff --git a/tests/test_flaltland_rail_agent_status.py b/tests/test_flatland_rail_agent_status.py
similarity index 100%
rename from tests/test_flaltland_rail_agent_status.py
rename to tests/test_flatland_rail_agent_status.py
--
GitLab