(self)
| 559 | class TestGraphPartitioning(unittest.TestCase): |
| 560 | |
| 561 | def setUp(self): |
| 562 | # Create test graph. |
| 563 | self.g = graph.Graph() |
| 564 | self.g.add_edge("a", "b", weight=0.5) |
| 565 | self.g.add_edge("a", "c") |
| 566 | self.g.add_edge("b", "d") |
| 567 | self.g.add_edge("d", "e") |
| 568 | self.g.add_edge("x", "y") |
| 569 | self.g.add_node("z") |
| 570 | |
| 571 | def test_union(self): |
| 572 | self.assertEqual(graph.union([1,2],[2,3]), [1,2,3]) |