added tests for the variadic merging

DarkWingMcQuack · Oct 22, 2021 · d60e8e6 · d60e8e6
1 parent 5f013e9
commit d60e8e6
Showing 1 changed file with 31 additions and 0 deletions.
diff --git a/test/tests.cpp b/test/tests.cpp
@@ -88,6 +88,37 @@ TEST(UnionFindTest, SimpleMergeTest4)
     EXPECT_EQ(uf.numberOfSets(), 1);
 }
 
+TEST(UnionFindTest, VariadicMergeTest1)
+{
+    unionfind::UnionFind uf{5};
+
+    uf.merge(0, 4, 3, 1, 2);
+
+    EXPECT_EQ(uf.find(0), 0);
+    EXPECT_EQ(uf.find(1), 0);
+    EXPECT_EQ(uf.find(2), 0);
+    EXPECT_EQ(uf.find(3), 0);
+    EXPECT_EQ(uf.find(4), 0);
+
+    EXPECT_EQ(uf.numberOfSets(), 1);
+}
+
+TEST(UnionFindTest, VariadicMergeTest2)
+{
+    unionfind::UnionFind uf{5};
+
+    uf.merge(0, 4, 3);
+    uf.merge(1, 2);
+
+    EXPECT_EQ(uf.find(0), 0);
+    EXPECT_EQ(uf.find(1), 1);
+    EXPECT_EQ(uf.find(2), 1);
+    EXPECT_EQ(uf.find(3), 0);
+    EXPECT_EQ(uf.find(4), 0);
+
+    EXPECT_EQ(uf.numberOfSets(), 2);
+}
+
 TEST(UnionFindTest, NumberOfSetsTest)
 {
     unionfind::UnionFind uf{5};