def test_find_2(self):
h = UF_HWQUPC(3, False)
h.connect(0, 1)
self.assertEqual(0, h.find(0))
self.assertEqual(0, h.find(1))
h.connect(2, 1)
self.assertEqual(0, h.find(0))
self.assertEqual(0, h.find(1))
self.assertEqual(0, h.find(2))
def test_find_4(self):
h = UF_HWQUPC(6)
h.connect(0, 1)
h.connect(0, 2)
h.connect(3, 4)
h.connect(3, 5)
self.assertEqual(0, h.find(0))
self.assertEqual(0, h.find(1))
self.assertEqual(0, h.find(2))
self.assertEqual(3, h.find(3))
self.assertEqual(3, h.find(4))
self.assertEqual(3, h.find(5))
h.connect(0, 3)
self.assertEqual(0, h.find(0))
self.assertEqual(0, h.find(1))
self.assertEqual(0, h.find(2))
self.assertEqual(0, h.find(3))
self.assertEqual(0, h.find(4))
self.assertEqual(0, h.find(5))
self.assertEqual(0, h.get_parent(4))
self.assertEqual(0, h.get_parent(5))
def test_find_1(self):
h = UF_HWQUPC(2)
h.connect(0, 1)
self.assertEqual(0, h.find(0))
self.assertEqual(0, h.find(1))
def test_connect_02(self):
h = UF_HWQUPC(2)
h.connect(0, 1)
h.connect(0, 1)
self.assertTrue(h.is_connected(0, 1))