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):
UserQueryingExpand.editCity()
oldName = "Champaign"
newName = "Urbana"
self.assertTrue(newName in QueryingData.cityDicationary)
