From 8db9cb3fcf8ee6fe0327c059c4c307f7c1010c62 Mon Sep 17 00:00:00 2001 From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch> Date: Fri, 16 Aug 2019 14:02:23 +0200 Subject: [PATCH] tests --- tests/test_flatland_env_sparse_rail_generator.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_flatland_env_sparse_rail_generator.py b/tests/test_flatland_env_sparse_rail_generator.py index 7079a4f9..49739297 100644 --- a/tests/test_flatland_env_sparse_rail_generator.py +++ b/tests/test_flatland_env_sparse_rail_generator.py @@ -8,7 +8,7 @@ from flatland.utils.rendertools import RenderTool import numpy as np def test_realistic_rail_generator(): - for test_loop in range(20): + for test_loop in range(5): num_agents = np.random.randint(10,30) env = RailEnv(width=np.random.randint(40,80), height=np.random.randint(10,20), @@ -18,7 +18,7 @@ def test_realistic_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) - time.sleep(1) + time.sleep(2) env_renderer.close_window() def test_sparse_rail_generator(): -- GitLab