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))
