def testTrondheimData(self):
xml_data = open("testdata/shortairlinelist.xml").read()
airlines = AirlineParser.parse_airlines(xml_data)
self.assertEqual(3, len(airlines))
expected = [ ("AA", "American Airlines"),
("DL","Delta Airlines"),
("DY", "Norwegian") ]
for e, a in zip(expected, airlines):
self.assertEqual(e[0], a.code)
self.assertEqual(e[1], a.name)
def generate_airline_factory():
airlines_xml = download_airline_xml()
airlines = AirlineParser.parse_airlines(airlines_xml)
airline_factory = AirlineFactory(airlines)
return airline_factory
dest="arrivals_only",
help="Print information about only arrivals.",
default=False,
)
(options, args) = parser.parse_args()
airportname = args[0]
airports_xml = FlightInformationService.download_airport_xml()
airports = AirPortParser.parse_airports(airports_xml)
airport_factory = AirPortFactory(airports)
airlines_xml = FlightInformationService.download_airline_xml()
airlines = AirlineParser.parse_airlines(airlines_xml)
airline_factory = AirlineFactory(airlines)
status_xml = FlightInformationService.download_flight_status_xml()
statuses = FlightStatusParser.parse_statuses(status_xml)
status_factory = FlightStatusFactory(statuses)
airport = airport_factory.get_airport_by_code(airportname)
query = Query(airport)
xml = FlightInformationService.download_flight_xml(query)
flights = FlightParser.parse_flights(xml, airline_factory, airport_factory, status_factory)
print "Flight information for %s (%s)" % (airport.name, airport.code)
def testRealisticData(self):
xml_data = open("testdata/airlinenames.xml").read()
airlines = AirlineParser.parse_airlines(xml_data)
self.assertEqual(709, len(airlines))
