def test_pushCars(self): lanes = 5 for x in range(lanes + 1): road = Road(lanes, 100, None) self.assertEqual(x, road.pushCars(x)) for x in range(lanes + 1, 100): road = Road(lanes, 100, None) self.assertEqual(lanes, road.pushCars(x))