From b515a1c4a625aed8277be7846e98bb2be9070cc8 Mon Sep 17 00:00:00 2001
From: spiglerg <spiglerg@gmail.com>
Date: Mon, 15 Apr 2019 14:12:05 +0200
Subject: [PATCH] fixed tests
---
tests/test_environments.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/tests/test_environments.py b/tests/test_environments.py
index 925874ba..c3329a12 100644
--- a/tests/test_environments.py
+++ b/tests/test_environments.py
@@ -3,6 +3,7 @@
from flatland.core.env import RailEnv
from flatland.core.transitions import Grid4Transitions
+from flatland.core.transitionmap import GridTransitionMap
import numpy as np
"""Tests for `flatland` package."""
@@ -44,7 +45,9 @@ def test_rail_environment_single_agent():
north_west_turn]],
dtype=np.uint16)
- rail_env = RailEnv(rail_map, number_of_agents=1)
+ rail = GridTransitionMap(width=3, height=3, transitions=transitions)
+ rail.grid = rail_map
+ rail_env = RailEnv(rail, number_of_agents=1)
for _ in range(200):
_ = rail_env.reset()
--
GitLab