1 files changed, 78 insertions, 0 deletions
diff --git a/tests/test_gameMove.py b/tests/test_gameMove.py
new file mode 100644
index 0000000..6569c5b
--- /dev/null
+++ b/tests/test_gameMove.py
@@ -0,0 +1,78 @@
+"""Tests for gameMove module."""
+
+import unittest
+
+from imago.data.enums import Player
+from imago.gameLogic.gameBoard import GameBoard
+from imago.gameLogic.gameMove import GameMove
+
+TEST_BOARD_SIZE = 19
+
+class TestGameMove(unittest.TestCase):
+    """Test gameMove module."""
+
+    def testAddMove(self):
+        """Test adding new moves to existing moves."""
+        board = GameBoard(TEST_BOARD_SIZE, TEST_BOARD_SIZE)
+        firstMove = GameMove(board)
+
+        self.assertIsNone(firstMove.coords)
+
+        secondMove = firstMove.addMove(1, 2)
+
+        self.assertIsNone(firstMove.coords)
+        self.assertEqual(secondMove.coords[0], 1)
+        self.assertEqual(secondMove.coords[1], 2)
+
+        thirdMove = secondMove.addMove(5, 7)
+
+        self.assertIsNone(firstMove.coords)
+        self.assertIsNone(thirdMove.previousMove.previousMove.coords)
+
+        self.assertEqual(secondMove.coords[0], 1)
+        self.assertEqual(secondMove.coords[1], 2)
+        self.assertEqual(thirdMove.previousMove.coords[0], 1)
+        self.assertEqual(thirdMove.previousMove.coords[1], 2)
+
+        self.assertEqual(thirdMove.coords[0], 5)
+        self.assertEqual(thirdMove.coords[1], 7)
+        self.assertEqual(firstMove
+                .nextMoves[0]
+                .nextMoves[0]
+                .coords[0], 5)
+        self.assertEqual(firstMove
+                .nextMoves[0]
+                .nextMoves[0]
+                .coords[1], 7)
+
+        self.assertEqual(firstMove.board.getBoard()[1][2], Player.EMPTY)
+        self.assertEqual(secondMove.board.getBoard()[1][2], Player.BLACK)
+        self.assertEqual(thirdMove.board.getBoard()[1][2], Player.BLACK)
+
+        self.assertEqual(firstMove.board.getBoard()[5][7], Player.EMPTY)
+        self.assertEqual(secondMove.board.getBoard()[5][7], Player.EMPTY)
+        self.assertEqual(thirdMove.board.getBoard()[5][7], Player.WHITE)
+
+    def testGetPlayableVertices(self):
+        """Test getting the set of valid moves."""
+        boardSize = 3
+        board = GameBoard(boardSize, boardSize)
+
+        firstMove = GameMove(board)
+        self.assertSetEqual(
+            firstMove.getPlayableVertices(),
+            set(((0,0), (0,1), (0,2),
+            (1,0), (1,1), (1,2),
+            (2,0), (2,1), (2,2)))
+        )
+
+        secondMove = firstMove.addMove(1, 2)
+        self.assertSetEqual(
+            secondMove.getPlayableVertices(),
+            set(((0,0), (0,1), (0,2),
+            (1,0), (1,1),
+            (2,0), (2,1), (2,2)))
+        )
+
+if __name__ == '__main__':
+    unittest.main()