99import static junit .framework .Assert .assertEquals ;
1010
1111public class _538Test {
12- private static _538 .Solution2 solution2 ;
1312private static _538 .Solution1 solution1 ;
13+ private static _538 .Solution2 solution2 ;
1414private static TreeNode expectedRoot ;
1515private static TreeNode root ;
1616
@@ -20,10 +20,6 @@ public static void setup() {
2020solution2 =new _538 .Solution2 ();
2121 }
2222
23- @ Before
24- public void setupForEachTest () {
25- }
26-
2723@ Test
2824public void test1 () {
2925root =new TreeNode (5 );
@@ -32,15 +28,15 @@ public void test1() {
3228expectedRoot =new TreeNode (18 );
3329expectedRoot .left =new TreeNode (20 );
3430expectedRoot .right =new TreeNode (13 );
35- assertEquals (expectedRoot .toString (),solution2 .convertBST (root ).toString ());
3631assertEquals (expectedRoot .toString (),solution1 .convertBST (root ).toString ());
32+ assertEquals (expectedRoot .toString (),solution2 .convertBST (root ).toString ());
3733 }
3834
3935@ Test
4036public void test2 () {
4137root =null ;
4238expectedRoot =null ;
43- assertEquals (expectedRoot ,solution2 .convertBST (root ));
4439assertEquals (expectedRoot ,solution1 .convertBST (root ));
40+ assertEquals (expectedRoot ,solution2 .convertBST (root ));
4541 }
4642}