def testAddFirstRoute(self):
adjList = AdjacencyList()
newRoute = Route(orig = 1, dest = 2, distance = 567)
self.assertIsNone(adjList.findRoutePair(newRoute.orig, newRoute.dest))
self.assertTrue(adjList.add(newRoute))
routeFoundInList = adjList.findRoutePair(newRoute.orig, newRoute.dest)
self.assertIsNotNone(routeFoundInList)
self.assertEqual(routeFoundInList.orig, newRoute.orig)
self.assertEqual(routeFoundInList.dest, newRoute.dest)
self.assertEqual(routeFoundInList.distance, newRoute.distance)
def testMatchingRoute(self):
adjList = AdjacencyList()
route1 = Route(orig = 1, dest = 2, distance = 20)
route2 = Route(orig = 1, dest = 3, distance = 40)
adjList.add(route1)
adjList.add(route2)
adjList = adjList.removeFromList(orig = 1, dest = 2)
self.assertIsNone(adjList.findRoutePair(orig = 1, dest = 2))
self.assertEqual(adjList.findRoutePair(orig = 1, dest = 3), route2)
def testAddBeyondFirst(self):
adjList = AdjacencyList()
newRoute1 = Route(orig = 1, dest = 2, distance = 567)
adjList.add(newRoute1)
newRoute2 = Route(orig = 2, dest = 3, distance = 784)
adjList.add(newRoute2)
self.assertEqual(adjList.findRoutePair(newRoute1.orig, newRoute1.dest), newRoute1)
self.assertEqual(adjList.findRoutePair(newRoute2.orig, newRoute2.dest), newRoute2)
