def test_base_digraph(self):
digraph_base = base(self.digraph)
self.assertEqual({6, 7, 8}, digraph_base)
def test_base_graph(self):
graph_base = base(self.graph)
self.assertTrue(any([node in graph_base for node in {1, 2, 5}]))
self.assertTrue(any([7 in graph_base, 8 in graph_base]))
self.assertTrue(11 in graph_base)
self.assertEqual(len(graph_base), 3)
