add tests for 623

fishercoder1534 · fishercoder1534 · commit 2b0ff7342eb1 · 2021-03-09T11:28:32.000-08:00
diff --git a/src/test/java/com/fishercoder/_623Test.java b/src/test/java/com/fishercoder/_623Test.java
@@ -0,0 +1,52 @@
+package com.fishercoder;
+
+import com.fishercoder.common.classes.TreeNode;
+import com.fishercoder.common.utils.CommonUtils;
+import com.fishercoder.common.utils.TreeUtils;
+import com.fishercoder.solutions._1727;
+import com.fishercoder.solutions._623;
+import org.junit.BeforeClass;
+import org.junit.Test;
+
+import java.util.Arrays;
+
+import static org.junit.Assert.assertEquals;
+
+public class _623Test {
+    private static _623.Solution1 solution1;
+
+    @BeforeClass
+    public static void setup() {
+        solution1 = new _623.Solution1();
+    }
+
+    @Test
+    public void test1() {
+        TreeNode expected = TreeUtils.constructBinaryTree(Arrays.asList(4, 1, 1, 2, null, null, 6, 3, 1, 5));
+        TreeUtils.printBinaryTree(expected);
+        TreeNode inputTree = TreeUtils.constructBinaryTree(Arrays.asList(4, 2, 6, 3, 1, 5));
+        TreeUtils.printBinaryTree(inputTree);
+        assertEquals(expected, solution1.addOneRow(inputTree, 1, 2));
+    }
+
+    @Test
+    public void test2() {
+        TreeNode expected = TreeUtils.constructBinaryTree(Arrays.asList(4, 2, null, 1, 1, 3, null, null, 1));
+        TreeUtils.printBinaryTree(expected);
+        TreeNode inputTree = TreeUtils.constructBinaryTree(Arrays.asList(4, 2, null, 3, 1));
+        TreeUtils.printBinaryTree(inputTree);
+        assertEquals(expected, solution1.addOneRow(inputTree, 1, 3));
+    }
+
+    @Test
+    public void test3() {
+        TreeNode expected = TreeUtils.constructBinaryTree(Arrays.asList(4, 2, 5, 1, 1, 1, 1, 3, null, null, 1, 6, null, null, 7));
+        TreeUtils.printBinaryTree(expected);
+        TreeNode inputTree = TreeUtils.constructBinaryTree(Arrays.asList(4, 2, 5, 3, 1, 6, 7));
+        TreeUtils.printBinaryTree(inputTree);
+        TreeNode actual = solution1.addOneRow(inputTree, 1, 3);
+        TreeUtils.printBinaryTree(actual);
+        assertEquals(expected, actual);
+    }
+
+}
