From 4528d62a05640d83cab8458b4f4d6385e9be1cbe Mon Sep 17 00:00:00 2001
From: hagrid67 <jdhwatson@gmail.com>
Date: Tue, 2 Jun 2020 20:45:06 +0100
Subject: [PATCH] adding back simple test to check that env.save() still exists
---
tests/test_flatland_envs_rail_env.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/tests/test_flatland_envs_rail_env.py b/tests/test_flatland_envs_rail_env.py
index 664c1155..07cf7789 100644
--- a/tests/test_flatland_envs_rail_env.py
+++ b/tests/test_flatland_envs_rail_env.py
@@ -40,8 +40,11 @@ def test_save_load():
agent_2_pos = env.agents[1].position
agent_2_dir = env.agents[1].direction
agent_2_tar = env.agents[1].target
- #env.save("test_save.dat")
+
+ env.save("test_save_2.pkl")
RailEnvPersister.save(env, "test_save.pkl")
+
+
#env.load("test_save.dat")
env, env_dict = RailEnvPersister.load_new("test_save.pkl")
assert (env.width == 10)
--
GitLab