From e9a65ca20d2d9c13a0b3ca69002d5208cf415256 Mon Sep 17 00:00:00 2001
From: Guillaume Mollard <guillaume.mollard2@gmail.com>
Date: Wed, 29 May 2019 10:30:25 +0200
Subject: [PATCH] corrected env observation test
---
tests/test_env_observation_builder.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/tests/test_env_observation_builder.py b/tests/test_env_observation_builder.py
index 4038fe67..ad393634 100644
--- a/tests/test_env_observation_builder.py
+++ b/tests/test_env_observation_builder.py
@@ -82,7 +82,7 @@ def test_global_obs():
# If this assertion is wrong, it means that the observation returned
# places the agent on an empty cell
- assert(np.sum(rail_map * global_obs[0][1][:, :, 0]) > 0)
+ assert(np.sum(rail_map * global_obs[0][1][:, :, :4].sum(2)) > 0)
def main():
--
GitLab