From 4e599bf9190098afd8ff285278c07210ddfa112f Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Tue, 20 Aug 2019 10:13:24 +0200
Subject: [PATCH] test corrected
---
flatland/envs/generators.py | 2 +-
tests/test_flatland_env_sparse_rail_generator.py | 10 +++++-----
2 files changed, 6 insertions(+), 6 deletions(-)
diff --git a/flatland/envs/generators.py b/flatland/envs/generators.py
index 45d897cc..8eae9312 100644
--- a/flatland/envs/generators.py
+++ b/flatland/envs/generators.py
@@ -1070,7 +1070,7 @@ def sparse_rail_generator(num_cities=100, num_intersections=10, num_trainstation
width - 1)
while (station_x, station_y) in train_stations or (station_x, station_y) == node_positions[
trainstation_node] or \
- rail_array[(station_x, station_y)] != 0:
+ rail_array[(station_x, station_y)] != 0:
station_x = np.clip(
node_positions[trainstation_node][0] + np.random.randint(-node_radius, node_radius),
0,
diff --git a/tests/test_flatland_env_sparse_rail_generator.py b/tests/test_flatland_env_sparse_rail_generator.py
index 17d2df0e..8f494df8 100644
--- a/tests/test_flatland_env_sparse_rail_generator.py
+++ b/tests/test_flatland_env_sparse_rail_generator.py
@@ -7,9 +7,8 @@ from flatland.envs.observations import GlobalObsForRailEnv
from flatland.envs.rail_env import RailEnv
from flatland.utils.rendertools import RenderTool, AgentRenderVariant
-
def test_realistic_rail_generator(vizualization_folder_name=None):
- for test_loop in range(20):
+ for test_loop in range(5):
print("test_loop", test_loop)
num_agents = np.random.randint(10, 30)
env = RailEnv(width=np.random.randint(40, 80),
@@ -21,8 +20,8 @@ def test_realistic_rail_generator(vizualization_folder_name=None):
obs_builder_object=GlobalObsForRailEnv())
# reset to initialize agents_static
env_renderer = RenderTool(env, gl="PILSVG", agent_render_variant=AgentRenderVariant.ONE_STEP_BEHIND,
- screen_height=1200,
- screen_width=1600)
+ screen_height=600,
+ screen_width=800)
env_renderer.render_env(show=True, show_observations=True, show_predictions=False)
if vizualization_folder_name is not None:
env_renderer.gl.save_image(
@@ -50,5 +49,6 @@ def test_sparse_rail_generator():
# reset to initialize agents_static
env_renderer = RenderTool(env, gl="PILSVG", )
env_renderer.render_env(show=True, show_observations=True, show_predictions=False)
+ env_renderer.close_window()
-test_realistic_rail_generator("./../rendering/")
+# test_realistic_rail_generator("./../rendering/")
--
GitLab