UserQuerying.getStaticalInfomation() print "\n" elif n.strip() == "4": cityCode = raw_input("Please type the city's code: \n") if UserQuerying.mapCity(cityCode.upper()) == False: print "Sorry, the city's code is invalid\n" elif n.strip() == "5": UserQueryingExpand.addCity() elif n.strip() == "6": UserQueryingExpand.removeCity() elif n.strip() == "7": UserQueryingExpand.addRoute() elif n.strip() == "8": UserQueryingExpand.removeRoute() elif n.strip() == "9": UserQueryingExpand.getRouteInfo() elif n.strip() == "10": UserQueryingExpand.editCity() elif n.strip() == "11": UserQueryingExpand.saveBackToJson() elif n.strip() == "12": UserQueryingExpand.getShortestPath()
def testAddRoute(self): UserQueryingExpand.addRoute() source = "Champaign" dest = "Shanghai" self.assertTrue(dest in UserQuerying.getFlyToCity(source))