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)
