Python DisjointSet Examples

Example #1

File: test_union_find.py Project: baiyanhuang/backup

    def test_union_find(self):
        s = DisjointSet(6)
        s.union(1, 2)
        s.union(2, 3)

        s.union(4, 5)
        s.union(4, 0)

        self.assertEqual(s.find(1), s.find(2))
        self.assertEqual(s.find(2), s.find(3))

        self.assertEqual(s.find(0), s.find(4))
        self.assertEqual(s.find(4), s.find(5))
        
        self.assertNotEqual(s.find(4), s.find(1))

Example #2

    def test_union_find(self):
        s = DisjointSet(6)
        s.union(1, 2)
        s.union(2, 3)

        s.union(4, 5)
        s.union(4, 0)

        self.assertEqual(s.find(1), s.find(2))
        self.assertEqual(s.find(2), s.find(3))

        self.assertEqual(s.find(0), s.find(4))
        self.assertEqual(s.find(4), s.find(5))

        self.assertNotEqual(s.find(4), s.find(1))

Example #3

File: test_union_find.py Project: baiyanhuang/backup

    def test_union_find_size_1(self):
        s = DisjointSet(1)

        s.union(0, 0)

        self.assertEqual(0, s.find(0))

Example #4

    def test_union_find_size_1(self):
        s = DisjointSet(1)

        s.union(0, 0)

        self.assertEqual(0, s.find(0))