def test_scenario_4(self): app = Labyrinth(4, 1) for country in app.get_countries(): if country.schengen: self.assertTrue(country.get_posture())
def test_scenario_3_places_5_sleeper_cells(self): app = Labyrinth(3, 1) sleeper_cells = sum( [country.sleeperCells for country in app.get_countries()]) self.assertEqual(5, sleeper_cells)