class TestAirport(unittest.TestCase):
def setUp(self):
self.airport = Airport()
self.flight = Flight()
self.flight2 = Flight(start_time=Date(day=29,
month=11,
year=2016,
hour='17:30'),
from_dest="Vancouver",
to_dest="New York")
self.terminal = Terminal()
self.airport.add_terminal(self.terminal)
def test_init(self): # passed
self.assertEqual(self.airport.name, "Sofia")
self.assertEqual(self.airport.terminals, [self.terminal])
def test_add_terminal(self):
self.assertEqual(self.airport.add_terminal(self.terminal),
[self.terminal, self.terminal])
def test_passengers_from_terminal(self):
passenger = Passenger(first_name="Georgi",
second_name="Atanasov",
age=20,
flight=self.flight)
passenger2 = Passenger(flight=self.flight2)
self.flight.add_passenger(passenger)
self.flight2.add_passenger(passenger2)
self.terminal.add_flight(self.flight)
self.terminal.add_flight(self.flight2)
self.assertEqual(self.airport.passengers_from_terminal(self.terminal),
[passenger, passenger2])
class TestTerminal(unittest.TestCase):
def setUp(self):
self.terminal = Terminal()
self.flight = Flight()
self.flight2 = Flight(start_time=Date(day=29,
month=11, year=2016, hour='17:30'),
from_dest="Vancouver", to_dest="New York",
end_time=Date(day=29, month=11,
year=2016, hour='20:40'))
self.terminal.add_flight(self.flight)
self.terminal.add_flight(self.flight2)
def test_init(self):
self.assertEqual(self.terminal.number, 1)
self.assertEqual(self.terminal.max_flights, 20)
def test_add_flight(self):
flight = Flight()
self.assertEqual(self.terminal.add_flight(flight),
[self.flight, self.flight2, flight])
def test_str(self):
self.assertEqual(str(self.terminal), "Terminal number 1")
def test_get_terminal_flights_on(self):
date = Date()
self.assertEqual(self.terminal.get_terminal_flights_on(date),
self.terminal.flights)
def test_get_terminal_flights(self):
self.assertEqual(self.terminal.get_terminal_flights(),
self.terminal.flights)
def test_terminal_flights_to_dest(self):
destination = "London"
self.assertEqual(self.terminal.terminal_flights_to_dest(destination),
[self.flight])
def test_get_flights_for(self): # passed
date = Date()
self.assertEqual(self.terminal.get_flights_for(date), 2)
def test_get_flights_before(self):
date = Date()
hour = '16:30'
self.assertEqual(self.terminal.get_flights_before(date, hour),
[self.flight])
def test_get_flight_from(self):
destination = self.flight.from_dest
self.assertEqual(self.terminal.get_flight_from(destination),
[self.flight])
def test_get_flight_to(self):
destination = self.flight.to_dest
self.assertEqual(self.terminal.get_flight_to(destination),
[self.flight])
def test_get_flight_to_hour(self):
destination = self.flight.to_dest
date = Date()
hour = '12:31'
self.assertEqual(self.terminal.
get_flight_to_hour(destination, date, hour),
[self.flight])
def test_get_flight_from_hour(self):
destination = self.flight.from_dest
date = Date()
hour = '12:31'
self.assertEqual(self.terminal.
get_flight_from_hour(destination, date, hour),
[self.flight])
def test_flights_on_date_lt_hours(self):
hours = 3.0
date = Date()
self.assertEqual(self.terminal.flights_on_date_lt_hours(date, hours),
[self.flight])
def test_flights_within_duration(self):
start_time = Date(day=29, month=11, year=2016, hour='17:30')
end_time = Date(day=29, month=11, year=2016, hour='20:40')
self.assertEqual(self.terminal.flights_within_duration(start_time,
end_time),
[self.flight2])
def test_passengers_to_dest(self):
destination = self.flight.to_dest
passenger = Passenger(first_name="Georgi", second_name="Atanasov",
age=20, flight=self.flight)
passenger2 = Passenger(flight=self.flight2)
self.flight.add_passenger(passenger)
self.flight2.add_passenger(passenger2)
self.assertEqual(self.terminal.passengers_to_dest(destination),
[passenger])
def test_flights_with_passengers_gt(self):
size = 0
passenger = Passenger(first_name="Georgi", second_name="Atanasov",
age=20, flight=self.flight)
passenger2 = Passenger(flight=self.flight2)
self.flight.add_passenger(passenger)
self.flight2.add_passenger(passenger2)
self.assertEqual(self.terminal.flights_with_passengers_gt(size),
[self.flight, self.flight2])
def test_reservations_to_destination(destination):
destination = "London"
pass
