refactor 116

fishercoder1534 · fishercoder1534 · commit 829f5a55c323 · 2021-10-10T16:41:26.000-07:00
diff --git a/src/test/java/com/fishercoder/_116Test.java b/src/test/java/com/fishercoder/_116Test.java
@@ -6,23 +6,37 @@
 import org.junit.Test;
 
 public class _116Test {
-  private static _116.Solution1 solution1;
-  private static TreeLinkNode root;
+    private static _116.Solution1 solution1;
+    private static _116.Solution2 solution2;
+    private static TreeLinkNode root;
 
-  @BeforeClass
-  public static void setup() {
-    solution1 = new _116.Solution1();
-  }
+    @BeforeClass
+    public static void setup() {
+        solution1 = new _116.Solution1();
+        solution2 = new _116.Solution2();
+    }
 
-  @Test
-  public void test1() {
-    root = new TreeLinkNode(1);
-    root.left = new TreeLinkNode(2);
-    root.right = new TreeLinkNode(3);
-    root.left.left = new TreeLinkNode(4);
-    root.left.right = new TreeLinkNode(5);
-    root.right.right = new TreeLinkNode(7);
+    @Test
+    public void test1() {
+        root = new TreeLinkNode(1);
+        root.left = new TreeLinkNode(2);
+        root.right = new TreeLinkNode(3);
+        root.left.left = new TreeLinkNode(4);
+        root.left.right = new TreeLinkNode(5);
+        root.right.right = new TreeLinkNode(7);
 
-    solution1.connect(root);
-  }
+        solution1.connect(root);
+    }
+
+    @Test
+    public void test2() {
+        root = new TreeLinkNode(1);
+        root.left = new TreeLinkNode(2);
+        root.right = new TreeLinkNode(3);
+        root.left.left = new TreeLinkNode(4);
+        root.left.right = new TreeLinkNode(5);
+        root.right.right = new TreeLinkNode(7);
+
+        solution2.connect(root);
+    }
 }
