11package com .fishercoder ;
22
33import com .fishercoder .common .classes .TreeNode ;
4+ import com .fishercoder .common .utils .TreeUtils ;
45import com .fishercoder .solutions ._536 ;
56import org .junit .Before ;
67import org .junit .BeforeClass ;
78import org .junit .Test ;
89
10+ import java .util .Arrays ;
11+
912import static junit .framework .Assert .assertEquals ;
1013
1114public class _536Test {
1215private static _536 .Solution1 solution1 ;
1316private static _536 .Solution2 solution2 ;
1417private static TreeNode expected ;
15- private static TreeNode actual ;
1618private static String s ;
1719
1820@ BeforeClass
@@ -23,19 +25,21 @@ public static void setup() {
2325
2426@ Before
2527public void setupForEachTest () {
26- expected =null ;
27- actual =null ;
28- s ="" ;
2928 }
3029
3130@ Test
3231public void test1 () {
33- s ="" ;
34- expected =null ;
35- actual =solution1 .str2tree (s );
36- assertEquals (expected ,actual );
32+ s ="4(2(3)(1))(6(5))" ;
33+ expected =TreeUtils .constructBinaryTree (Arrays .asList (4 ,2 ,6 ,3 ,1 ,5 ));
34+ assertEquals (expected ,solution1 .str2tree (s ));
35+ assertEquals (expected ,solution2 .str2tree (s ));
36+ }
3737
38- actual =solution2 .str2tree (s );
39- assertEquals (expected ,actual );
38+ @ Test
39+ public void test2 () {
40+ s ="51(232)(434)" ;
41+ expected =TreeUtils .constructBinaryTree (Arrays .asList (51 ,232 ,434 ));
42+ assertEquals (expected ,solution1 .str2tree (s ));
43+ assertEquals (expected ,solution2 .str2tree (s ));
4044 }
4145}