@@ -14,8 +14,8 @@ public class _538Test {
1414private static TreeNode expectedRoot ;
1515private static TreeNode root ;
1616
17- @ BeforeClass
18- public static void setup () {
17+ @ Before
18+ public void setup () {
1919solution1 =new _538 .Solution1 ();
2020solution2 =new _538 .Solution2 ();
2121 }
@@ -29,11 +29,21 @@ public void test1() {
2929expectedRoot .left =new TreeNode (20 );
3030expectedRoot .right =new TreeNode (13 );
3131assertEquals (expectedRoot .toString (),solution1 .convertBST (root ).toString ());
32- assertEquals (expectedRoot .toString (),solution2 .convertBST (root ).toString ());
3332 }
3433
3534@ Test
3635public void test2 () {
36+ root =new TreeNode (5 );
37+ root .left =new TreeNode (2 );
38+ root .right =new TreeNode (13 );
39+ expectedRoot =new TreeNode (18 );
40+ expectedRoot .left =new TreeNode (20 );
41+ expectedRoot .right =new TreeNode (13 );
42+ assertEquals (expectedRoot .toString (),solution2 .convertBST (root ).toString ());
43+ }
44+
45+ @ Test
46+ public void test3 () {
3747root =null ;
3848expectedRoot =null ;
3949assertEquals (expectedRoot ,solution1 .convertBST (root ));