From a133437bfad6fdb09955d531ef9035b53175e3dd Mon Sep 17 00:00:00 2001
From: Giacomo Spigler <spiglerg@gmail.com>
Date: Wed, 5 Jun 2019 22:33:59 +0200
Subject: [PATCH] fix?
---
tests/test_environments.py | 12 +++++++-----
1 file changed, 7 insertions(+), 5 deletions(-)
diff --git a/tests/test_environments.py b/tests/test_environments.py
index ba885d1..a5d1625 100644
--- a/tests/test_environments.py
+++ b/tests/test_environments.py
@@ -164,6 +164,8 @@ def test_dead_end():
def check_consistency(rail_env):
# We run step to check that trains do not move anymore
# after being done.
+ # TODO: GIACOMO: this is deprecated and should be updated; thenew behavior is that agents keep moving
+ # until they are manually stopped.
for i in range(7):
# prev_pos = rail_env.agents_position[0]
prev_pos = rail_env.agents[0].position
@@ -172,13 +174,13 @@ def test_dead_end():
# it stays where it is.
_ = rail_env.step({0: 1})
_ = rail_env.step({0: 3})
- assert (rail_env.agents[0].position == prev_pos)
+ # assert (rail_env.agents[0].position == prev_pos)
_, _, dones, _ = rail_env.step({0: 2})
- if i < 5:
- assert (not dones[0] and not dones['__all__'])
- else:
- assert (dones[0] and dones['__all__'])
+ # if i < 5:
+ # assert (not dones[0] and not dones['__all__'])
+ # else:
+ # assert (dones[0] and dones['__all__'])
# We try the configuration in the 4 directions:
rail_env.reset()
--
GitLab