def test_reserved_seat_count_of_train_of_10_seats_and_one_reserved_seat( self): train_info = TrainInfo( build_seats_data(total_seats=10, reserved_seats=1, booking_reference=TEST_BOOKING_REFERENCE)) self.assertEqual(1, train_info.reserved_seat_count())
def test_reserved_seat_count_of_an_empty_train_of_10_seats(self): train_info = TrainInfo(build_seats_data(total_seats=10)) self.assertEqual(0, train_info.reserved_seat_count())