From 8ed5dbf49bc4b363e43380925197fd78c25fffd0 Mon Sep 17 00:00:00 2001
From: hagrid67 <jdhwatson@gmail.com>
Date: Wed, 3 Jun 2020 17:45:14 +0100
Subject: [PATCH] changed the assertRaises to check for ValueError rather than
SystemExit when cities cannot be packed in
---
tests/test_flatland_envs_sparse_rail_generator.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/tests/test_flatland_envs_sparse_rail_generator.py b/tests/test_flatland_envs_sparse_rail_generator.py
index ad2fc431..c6e151b3 100644
--- a/tests/test_flatland_envs_sparse_rail_generator.py
+++ b/tests/test_flatland_envs_sparse_rail_generator.py
@@ -1483,7 +1483,7 @@ def test_sparse_generator_with_illegal_params_aborts():
"""
Test that the constructor aborts if the initial parameters don't allow more than one city to be built.
"""
- with unittest.TestCase.assertRaises(test_sparse_generator_with_illegal_params_aborts, SystemExit):
+ with unittest.TestCase.assertRaises(test_sparse_generator_with_illegal_params_aborts, ValueError):
RailEnv(width=6, height=6, rail_generator=sparse_rail_generator(
max_num_cities=100,
max_rails_between_cities=3,
@@ -1492,7 +1492,7 @@ def test_sparse_generator_with_illegal_params_aborts():
), schedule_generator=sparse_schedule_generator(), number_of_agents=10,
obs_builder_object=GlobalObsForRailEnv()).reset()
- with unittest.TestCase.assertRaises(test_sparse_generator_with_illegal_params_aborts, SystemExit):
+ with unittest.TestCase.assertRaises(test_sparse_generator_with_illegal_params_aborts, ValueError):
RailEnv(width=60, height=60, rail_generator=sparse_rail_generator(
max_num_cities=1,
max_rails_between_cities=3,
--
GitLab