add tests for 814

fishercoder1534 · fishercoder1534 · commit ba7dbba75217 · 2021-07-23T10:00:01.000-07:00
diff --git a/src/main/java/com/fishercoder/solutions/_814.java b/src/main/java/com/fishercoder/solutions/_814.java
@@ -19,4 +19,5 @@ public TreeNode pruneTree(TreeNode root) {
             return root;
         }
     }
+
 }
diff --git a/src/test/java/com/fishercoder/_814Test.java b/src/test/java/com/fishercoder/_814Test.java
@@ -0,0 +1,31 @@
+package com.fishercoder;
+
+import com.fishercoder.common.classes.TreeNode;
+import com.fishercoder.common.utils.TreeUtils;
+import com.fishercoder.solutions._814;
+import org.junit.BeforeClass;
+import org.junit.Test;
+
+import java.util.Arrays;
+
+public class _814Test {
+    private static _814.Solution1 solution1;
+
+    @BeforeClass
+    public static void setup() {
+        solution1 = new _814.Solution1();
+    }
+
+    @Test
+    public void test1() {
+        TreeUtils.printBinaryTree(solution1.pruneTree(TreeUtils.constructBinaryTree(Arrays.asList(1, null, 0, 0, 1))));
+    }
+
+    @Test
+    public void test2() {
+        TreeNode root = TreeUtils.constructBinaryTree(Arrays.asList(1, 0, 1, 0, 0, 0, 1));
+        TreeUtils.printBinaryTree(root);
+        TreeUtils.printBinaryTree(solution1.pruneTree(root));
+    }
+
+}

Original file line number	Diff line number	Diff line change
`@@ -19,4 +19,5 @@ public TreeNode pruneTree(TreeNode root) {`
`19`	`19`	`return root;`
`20`	`20`	`}`
`21`	`21`	`}`
	`22`	`+`
`22`	`23`	`}`