class TestAirport(unittest.TestCase):
def setUp(self):
self.airport = Airport(flights, passengers, reservations)
def test_get_flights_for_date(self):
date = Date(18, 11, 2016)
flights = [flight_1, flight_3]
self.assertEqual(self.airport.get_flights_for_date(date), flights)
def test_get_flights_before(self):
date = Date(19, 11, 2016)
flights = [flight_1, flight_2, flight_3, flight_4]
self.assertEqual(self.airport.get_flights_before(date, "23:59"),
flights)
def test_get_flights_from(self):
flights = [flight_1, flight_2, flight_3, flight_4, flight_5]
self.assertEqual(self.airport.get_flights_from("Sofia"), flights)
def test_get_flights_to(self):
flights = [flight_2]
self.assertEqual(self.airport.get_flights_to("London"), flights)
def test_flight_from(self):
date = Date(18, 11, 2016)
flights = [flight_1, flight_3]
self.assertEqual(self.airport.get_flight_from("Sofia", date), flights)
def test_flight_to(self):
date = Date(18, 11, 2016)
flights = [flight_1]
self.assertEqual(self.airport.get_flight_to("Madrid", date), flights)
def test_flights_on_date_lt_hours(self):
date = Date(18, 11, 2016)
flights = [flight_1]
self.assertEqual(self.airport.flights_on_date_lt_hours(date, "04:25"),
flights)
def test_flights_within_duration(self):
date = Date(18, 11, 2016)
flights = [flight_1]
self.assertEqual(self.airport.flights_on_date_lt_hours(date, "04:25"),
flights)
def test_passengers_under_18(self):
self.assertEqual(self.airport.passengers_under_18(flight_3), [psng_3])
def test_passengers_to_dest(self):
passengers = [psng_2, psng_7]
self.assertEqual(self.airport.passengers_to_dest("London"), passengers)
def test_passengers_from_terminal(self):
passengers = [psng_1, psng_2, psng_6, psng_7]
self.assertEqual(self.airport.passengers_from_terminal(terminal_1),
passengers)
def test_flights_with_passengers(self):
flights = [flight_1, flight_4, flight_5]
self.assertEqual(self.airport.flights_with_passengers(90), flights)
def test_passengers_reservations(self):
reservations = [res_1, res_6]
self.assertEqual(self.airport.passengers_reservations(flight_1),
reservations)
def test_reservations_to_destionation(self):
reservations = [res_2, res_7]
self.assertEqual(self.airport.reservations_to_destination("London"),
reservations)
