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 testNonMatchingRoute(self): adjList = AdjacencyList() adjList.add(Route(orig = 1, dest = 2, distance = 20)) self.assertEqual(adjList.removeFromList(orig = 3, dest = 4), adjList)
def testNoRoutes(self): adjList = AdjacencyList() self.assertTrue(adjList.removeFromList(orig = 1, dest = 2).empty())