def test_can_pick_up_passenger_from_bus_stop(self): person_1 = Person("Guido van Rossum", 64) person_2 = Person("Carol Willing", 50) bus_stop = BusStop("Waverly Station") bus_stop.add_to_queue(person_1) bus_stop.add_to_queue(person_2) self.bus.pick_up_from_stop(bus_stop) self.assertEqual(2, self.bus.passenger_count())
def setUp(self): self.bus_stop = BusStop("Waverly Station") self.person = Person("Guido van Rossum", 64)