def testMultipleNeighbors(self): adjList = AdjacencyList() newRoute1 = Route(orig = 1, dest = 2, distance = 567) newRoute2 = Route(orig = 1, dest = 77, distance = 99) newRoute3 = Route(orig = 1, dest = 3222, distance = 0.54) adjList.add(newRoute1) adjList.add(newRoute2) adjList.add(newRoute3) self.assertEqual(adjList.findAllNeighbors(1), [2, 77, 3222])
def testOneNeighbor(self): adjList = AdjacencyList() newRoute = Route(orig = 1, dest = 2, distance = 567) adjList.add(newRoute) self.assertEqual(adjList.findAllNeighbors(1), [2])
def testNoNeighbors(self): adjList = AdjacencyList() self.assertEqual(adjList.findAllNeighbors(1), [])