1 files changed, 83 insertions, 0 deletions
diff --git a/src/test/MergingTablesTest.java b/src/test/MergingTablesTest.java
new file mode 100644
index 0000000..fcb5829
--- /dev/null
+++ b/src/test/MergingTablesTest.java
@@ -0,0 +1,83 @@
+import org.junit.jupiter.api.Test;
+
+import java.io.*;
+
+import static org.junit.jupiter.api.Assertions.*;
+
+class MergingTablesTest {
+    @Test
+    void test() {
+        MergingTables.InputReader reader = new MergingTables.InputReader(System.in);
+        MergingTables.OutputWriter writer = new MergingTables.OutputWriter(System.out);
+        MergingTables mergingTables = new MergingTables(reader, writer);
+        mergingTables.tables = new MergingTables.Table[5];
+
+        MergingTables.Table t0 = new MergingTables.Table(1);
+        MergingTables.Table t1 = new MergingTables.Table(1);
+        MergingTables.Table t2 = new MergingTables.Table(1);
+        MergingTables.Table t3 = new MergingTables.Table(1);
+        MergingTables.Table t4 = new MergingTables.Table(1);
+
+        mergingTables.tables[0] = t0;
+        mergingTables.tables[1] = t1;
+        mergingTables.tables[2] = t2;
+        mergingTables.tables[3] = t3;
+        mergingTables.tables[4] = t4;
+        mergingTables.maximumNumberOfRows = 1;
+        mergingTables.rank = new int[]{0, 0, 0, 0, 0};
+        mergingTables.parent = new int[]{0, 1, 2, 3, 4};
+
+        mergingTables.merge(2, 4);
+        assertEquals(2, mergingTables.maximumNumberOfRows);
+
+        mergingTables.merge(1, 3);
+        assertEquals(2, mergingTables.maximumNumberOfRows);
+
+        mergingTables.merge(0, 3);
+        assertEquals(3, mergingTables.maximumNumberOfRows);
+
+        mergingTables.merge(4, 3);
+        assertEquals(5, mergingTables.maximumNumberOfRows);
+
+        mergingTables.merge(4, 2);
+        assertEquals(5, mergingTables.maximumNumberOfRows);
+    }
+
+    @Test
+    void test1() {
+        MergingTables.InputReader reader = new MergingTables.InputReader(System.in);
+        MergingTables.OutputWriter writer = new MergingTables.OutputWriter(System.out);
+        MergingTables mergingTables = new MergingTables(reader, writer);
+        mergingTables.tables = new MergingTables.Table[6];
+
+        MergingTables.Table t0 = new MergingTables.Table(10);
+        MergingTables.Table t1 = new MergingTables.Table(0);
+        MergingTables.Table t2 = new MergingTables.Table(5);
+        MergingTables.Table t3 = new MergingTables.Table(0);
+        MergingTables.Table t4 = new MergingTables.Table(3);
+        MergingTables.Table t5 = new MergingTables.Table(3);
+
+        mergingTables.tables[0] = t0;
+        mergingTables.tables[1] = t1;
+        mergingTables.tables[2] = t2;
+        mergingTables.tables[3] = t3;
+        mergingTables.tables[4] = t4;
+        mergingTables.tables[5] = t5;
+        mergingTables.maximumNumberOfRows = 10;
+        mergingTables.rank = new int[]{0, 0, 0, 0, 0, 0};
+        mergingTables.parent = new int[]{0, 1, 2, 3, 4, 5};
+
+        mergingTables.merge(5, 5);
+        assertEquals(10, mergingTables.maximumNumberOfRows);
+
+        mergingTables.merge(5, 4);
+        assertEquals(10, mergingTables.maximumNumberOfRows);
+
+        mergingTables.merge(4, 3);
+        assertEquals(10, mergingTables.maximumNumberOfRows);
+
+        mergingTables.merge(3, 2);
+        assertEquals(11, mergingTables.maximumNumberOfRows);
+    }
+
+}
+\ No newline at end of file