fix build

Author: fishercoder1534

src/test/java/com/fishercoder/_200Test.java

@@ -1,7 +1,7 @@
 package com.fishercoder;
 
 import com.fishercoder.solutions._200;
-import org.junit.BeforeClass;
+import org.junit.Before;
 import org.junit.Test;
 
 import static org.junit.Assert.assertEquals;
@@ -11,8 +11,8 @@ public class _200Test {
     private static _200.UnionFindSolution unionFindSolution;
     private static char[][] grid;
 
-    @BeforeClass
-    public static void setup(){
+    @Before
+    public void setup(){
         dfsSolution = new _200.DFSSolution();
         unionFindSolution = new _200.UnionFindSolution();
     }
@@ -25,6 +25,12 @@ public void test1(){
                 {'1','1','1'},
         };
         assertEquals(1, dfsSolution.numIslands(grid));
+
+        grid = new char[][]{
+                {'1','1','1'},
+                {'0','1','0'},
+                {'1','1','1'},
+        };
         assertEquals(1, unionFindSolution.numIslands(grid));
     }
 
@@ -36,8 +42,15 @@ public void test2(){
                 {'1','1','0','0','0'},
                 {'0','0','0','0','0'},
         };
-        assertEquals(0, dfsSolution.numIslands(grid));
-        assertEquals(0, unionFindSolution.numIslands(grid));
+        assertEquals(1, dfsSolution.numIslands(grid));
+
+        grid = new char[][]{
+                {'1','1','1','1','0'},
+                {'1','1','0','1','0'},
+                {'1','1','0','0','0'},
+                {'0','0','0','0','0'},
+        };
+        assertEquals(1, unionFindSolution.numIslands(grid));
     }
 
     @Test
@@ -49,6 +62,13 @@ public void test3(){
                 {'0','0','0','1','1'},
         };
         assertEquals(3, dfsSolution.numIslands(grid));
+
+        grid = new char[][]{
+                {'1','1','0','0','0'},
+                {'1','1','0','0','0'},
+                {'0','0','1','0','0'},
+                {'0','0','0','1','1'},
+        };
         assertEquals(3, unionFindSolution.numIslands(grid));
     }