More tests for getPiecePositions.

skytreader · Oct 19, 2014 · bcc9e81 · bcc9e81
1 parent 596a16d
commit bcc9e81
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/src/test/java/net/skytreader/kode/chesstemplar/GridBoardTest.java b/src/test/java/net/skytreader/kode/chesstemplar/GridBoardTest.java
@@ -117,6 +117,26 @@ public void testGetPiecePositions(){
         }
 
         Assert.assertEquals(initialConfig, testBoard.getPiecePositions());
+
+        // Remove the 5s
+        testBoard.removePiece(0, 5);
+        testBoard.removePiece(1, 5);
+        testBoard.removePiece(6, 5);
+        testBoard.removePiece(7, 5);
+
+        initialConfig.remove(new Point(0, 5));
+        initialConfig.remove(new Point(1, 5));
+        initialConfig.remove(new Point(6, 5));
+        initialConfig.remove(new Point(7, 5));
+
+        Assert.assertEquals(initialConfig, testBoard.getPiecePositions());
+
+        // Move some pawns
+        testBoard.move(6, 4, 4, 4);
+        initialConfig.remove(new Point(6, 4));
+        initialConfig.add(new Point(4, 4));
+
+        Assert.assertEquals(initialConfig, testBoard.getPiecePositions());
     }
 
     /**