def testOneRoute(self):
adjList = AdjacencyList()
addingRoute = Route(orig = 1, dest = 2, distance = 20)
adjList.add(addingRoute)
queue = adjList.getPQ()
onlyRoute = queue.get()
self.assertEqual(addingRoute, onlyRoute)
self.assertTrue(queue.empty())
def testMultipleRoutes(self):
adjList = AdjacencyList()
addRoute1 = Route(orig = 1, dest = 2, distance = 20)
addRoute2 = Route(orig = 2, dest = 3, distance = 50)
addRoute3 = Route(orig = 2, dest = 4, distance = 30)
#Sorted by ascending distance
routesByDist = [addRoute1, addRoute3, addRoute2]
adjList.add(addRoute1)
adjList.add(addRoute2)
adjList.add(addRoute3)
queue = adjList.getPQ()
#Match the routes and ordering of those routes
for i in range(0, len(routesByDist)):
pqEntry = queue.get()
self.assertEqual(pqEntry, routesByDist[i])
#There should be any additional routes in the PQ.
self.assertTrue(queue.empty())
def testNoRoutes(self):
adjList = AdjacencyList()
queue = adjList.getPQ()
self.assertTrue(queue.empty())
