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() elif n.strip() == "13": break else: print "Please give a valid input..."
def testEditCity(self): station1 = "Chicago" station2 = "Atlanta" station3 = "Miami" cost = 622 self.assertEqual(int(UserQueryingExpand.getRouteInfo()), cost)