fix build

fishercoder1534 · fishercoder1534 · commit 3b6318672851 · 2021-02-09T10:11:01.000-08:00
diff --git a/src/test/java/com/fishercoder/_538Test.java b/src/test/java/com/fishercoder/_538Test.java
@@ -14,8 +14,8 @@ public class _538Test {
     private static TreeNode expectedRoot;
     private static TreeNode root;
 
-    @BeforeClass
-    public static void setup() {
+    @Before
+    public void setup() {
         solution1 = new _538.Solution1();
         solution2 = new _538.Solution2();
     }
@@ -29,11 +29,21 @@ public void test1() {
         expectedRoot.left = new TreeNode(20);
         expectedRoot.right = new TreeNode(13);
         assertEquals(expectedRoot.toString(), solution1.convertBST(root).toString());
-        assertEquals(expectedRoot.toString(), solution2.convertBST(root).toString());
     }
 
     @Test
     public void test2() {
+        root = new TreeNode(5);
+        root.left = new TreeNode(2);
+        root.right = new TreeNode(13);
+        expectedRoot = new TreeNode(18);
+        expectedRoot.left = new TreeNode(20);
+        expectedRoot.right = new TreeNode(13);
+        assertEquals(expectedRoot.toString(), solution2.convertBST(root).toString());
+    }
+
+    @Test
+    public void test3() {
         root = null;
         expectedRoot = null;
         assertEquals(expectedRoot, solution1.convertBST(root));
