def test_get_population(self): my_graph = Graph() my_controller = Controller(my_graph) my_controller.parse_file("F:/rshan3/Airline/content/map_test_data.json") self.assertTrue(my_graph.get_biggest_city().code == 'MEX') self.assertTrue(my_graph.get_smallest_city().code == 'SCL')
def test_get_average_distances_of_flight(self): my_graph = Graph() my_controller = Controller(my_graph) my_controller.parse_file("F:/rshan3/Airline/content/map_test_data.json") #print(my_graph.list_metros()) self.assertTrue(my_graph.get_average_distances_of_flight() == (4231+2453)/2.0 )
def test_list_metros(self): my_graph = Graph() my_controller = Controller(my_graph) my_controller.parse_file("F:/rshan3/Airline/content/map_test_data.json") #print(my_graph.list_metros()) self.assertTrue(len(my_graph.list_metros()) == 3 )
def test_get_single_flight(self): my_graph = Graph() my_controller = Controller(my_graph) my_controller.parse_file("F:/rshan3/Airline/content/map_test_data.json") #print(my_graph.list_metros()) self.assertTrue(my_graph.get_longest_single_flight().distance == 4231 ) self.assertTrue(my_graph.get_shortest_single_flight().distance == 2453)
def test_add_edges(self): my_graph = Graph() my_controller = Controller(my_graph) my_controller.parse_file("F:/rshan3/Airline/content/map_test_data.json") self.assertTrue(len(my_graph.routes) == 2*2) self.assertTrue('BOG' not in my_graph.metros)
def test_get_city_info(self): my_graph = Graph() my_controller = Controller(my_graph) my_controller.parse_file("F:/rshan3/Airline/content/map_test_data.json") self.assertTrue(my_graph.get_city_info('LIM') is not None) self.assertTrue(my_graph.get_city_info('LIMM') is None) self.assertTrue(my_graph.get_city_info("Lima") is not None)
from controller.Controller import Controller from model.Graph import Graph __author__ = 'Ruihan' my_graph = Graph() my_controller = Controller(my_graph) my_controller.parse_file("content/map_data.json") my_controller.start()
def test_add_nodes(self): my_graph = Graph() my_controller = Controller(my_graph) my_controller.parse_file("F:/rshan3/Airline/content/map_test_data.json") self.assertTrue(len(my_graph.metros) == 3)
def test_get_hub(self): my_graph = Graph() my_controller = Controller(my_graph) my_controller.parse_file("F:/rshan3/Airline/content/map_test_data.json") self.assertTrue(my_graph.get_hub()[0] == 'Lima')