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)