edit 145

fishercoder1534 · fishercoder1534 · commit b152b2399e24 · 2017-07-31T08:45:28.000-07:00
diff --git a/src/main/java/com/fishercoder/solutions/_145.java b/src/main/java/com/fishercoder/solutions/_145.java
@@ -1,7 +1,6 @@
 package com.fishercoder.solutions;
 
 import com.fishercoder.common.classes.TreeNode;
-import com.fishercoder.common.utils.CommonUtils;
 
 import java.util.ArrayList;
 import java.util.Collections;
@@ -38,20 +37,6 @@ public static List<Integer> postorderTraversal_iterative(TreeNode root) {
         return result;
     }
     
-    public static void main(String...strings){
-//        TreeNode root = new TreeNode(1);
-//        root.left = new TreeNode(2);
-//        root.right = new TreeNode(3);
-        
-//        TreeNode root = new TreeNode(1);
-//        root.left = new TreeNode(2);
-        
-        TreeNode root = new TreeNode(1);
-        root.right = new TreeNode(2);
-        List<Integer> result = postorderTraversal_iterative(root);
-        CommonUtils.printList(result);
-    }
-    
     public List<Integer> postorderTraversal_recursive(TreeNode root) {
         List<Integer> result = new ArrayList();
         return post(root, result);
diff --git a/src/test/java/com/fishercoder/_145Test.java b/src/test/java/com/fishercoder/_145Test.java
@@ -0,0 +1,32 @@
+package com.fishercoder;
+
+import com.fishercoder.common.classes.TreeNode;
+import com.fishercoder.solutions._145;
+import org.junit.BeforeClass;
+import org.junit.Test;
+
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.List;
+
+import static org.junit.Assert.assertEquals;
+
+public class _145Test {
+    private static _145 test;
+    private static TreeNode root;
+    private static List<Integer> expected;
+
+    @BeforeClass
+    public static void setup(){
+        test = new _145();
+    }
+
+    @Test
+    public void test1(){
+        root = new TreeNode(1);
+        root.left = new TreeNode(2);
+        root.right = new TreeNode(3);
+        expected = new ArrayList<>(Arrays.asList(2, 3, 1));
+        assertEquals(expected, test.postorderTraversal_recursive(root));
+    }
+}
