def test_get_links_small(self):
graph = WeightedGraph()
graph.add_direct_link(0, 5, 1)
graph.add_direct_link(0, 1, 1)
graph.add_direct_link(0, 2, 1)
graph.add_direct_link(1, 5, 1)
graph.add_direct_link(2, 5, 1)
graph.add_direct_link(2, 3, 1)
graph.add_direct_link(3, 4, 1)
graph.add_direct_link(3, 5, 1)
graph.add_direct_link(5, 6, 1)
graph.add_direct_link(5, 7, 1)
graph.add_direct_link(6, 9, 1)
graph.add_direct_link(7, 11, 1)
graph.add_direct_link(8, 9, 1)
graph.add_direct_link(9, 10, 1)
graph.add_direct_link(9, 11, 1)
links = sorted(graph.get_links(1))
self.assertEqual(links, [0, 5])
links = sorted(graph.get_links(3))
self.assertEqual(links, [2, 4, 5])
links = sorted(graph.get_links(5))
self.assertEqual(links, [0, 1, 2, 3, 6, 7])
links = sorted(graph.get_links(11))
self.assertEqual(links, [7, 9])
def test_get_links_tiny(self):
graph = WeightedGraph()
graph.add_direct_link(0, 1, 50)
graph.add_direct_link(1, 2, 34)
graph.add_direct_link(1, 3, 42)
graph.add_direct_link(4, 1, 24)
links = sorted(graph.get_links(1))
self.assertEqual(links, [0, 2, 3, 4])
