def testAdditionalFunctions(self): a = MultiGraph(["a", "b", "c"]) a.add_edge("a", "b", "label1") a.add_edge("b", "c", "label1") a.add_edge("b", "a", "label2") self.assertEqual( str(a), "<MultiGraph: 3 node(s), 3 edge(s), 2 unique label(s)>") self.assertListEqual(a.edges("label1"), [("a", "b"), ("b", "c")]) self.assertListEqual(a.labels(), ["label1", "label2"])
def testAdditionalFunctions(self): a = MultiGraph(['a', 'b', 'c']) a.add_edge('a', 'b', 'label1') a.add_edge('b', 'c', 'label1') a.add_edge('b', 'a', 'label2') self.assertTrue( str(a) == "<MultiGraph: 3 node(s), 3 edge(s), 2 unique label(s)>") self.assertListEqual(a.edges('label1'), [('a', 'b'), ('b', 'c')]) self.assertListEqual(a.labels(), ['label1', 'label2'])