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)