def test_arrive(self): call = Call(origin=1, destination=5, size=1, init_time=1) call.board(5) call.arrive(10) self.assertEqual(call.state, 'ARRIVED') self.assertEqual(call.arrival_time, 10)
def test_board(self): call = Call(origin=1, destination=5, size=1, init_time=1) call.board(5) self.assertEqual(call.state, 'BOARDED') self.assertEqual(call.board_time, 5)