UnionFindUnitTest

Complexity

Total Complexity

1

Size/Duplication

Total Lines	12
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

1 Method

Rating	Name	Duplication	Size	Complexity
A	test_find()	0	11	1

import unittest

from pyalgs.algorithms.commons.union_find import UnionFind, QuickFind, QuickUnion


class UnionFindUnitTest(unittest.TestCase):
    def test_find(self):
        uf = UnionFind.create(10)

        uf.union(1, 3)
        uf.union(2, 4)
        uf.union(1, 5)

        self.assertTrue(uf.connected(1, 3))
        self.assertTrue(uf.connected(3, 5))
        self.assertFalse(uf.connected(1, 2))
        self.assertFalse(uf.connected(1, 4))


class QuickFindUnitTest(unittest.TestCase):
    def test_find(self):
        uf = QuickFind(10)

        uf.union(1, 3)
        uf.union(2, 4)
        uf.union(1, 5)

        self.assertTrue(uf.connected(1, 3))
        self.assertTrue(uf.connected(3, 5))
        self.assertFalse(uf.connected(1, 2))
        self.assertFalse(uf.connected(1, 4))


class QuickUnionUnitTest(unittest.TestCase):
    def test_find(self):
        uf = QuickUnion(10)

        uf.union(1, 3)
        uf.union(2, 4)
        uf.union(1, 5)

        self.assertTrue(uf.connected(1, 3))
        self.assertTrue(uf.connected(3, 5))
        self.assertFalse(uf.connected(1, 2))
        self.assertFalse(uf.connected(1, 4))


if __name__ == '__main__':
    unittest.main()