test corrected

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,

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/")