1 files changed, 17 insertions, 9 deletions
diff --git a/tests/test_monteCarlo.py b/tests/test_monteCarlo.py
index 8fe6f47..283c657 100644
--- a/tests/test_monteCarlo.py
+++ b/tests/test_monteCarlo.py
@@ -2,21 +2,29 @@
 
 import unittest
 
-from imago.gameLogic.gameBoard import GameBoard
-from imago.gameLogic.gameMove import GameMove
+from imago.gameLogic.gameState import GameState
+from imago.engine.monteCarlo import MCTS
 from imago.engine.monteCarlo import MCTSNode
 
-TEST_BOARD_SIZE = 19
+TEST_BOARD_SIZE = 9
 
 class TestMonteCarlo(unittest.TestCase):
     """Test MonteCarlo algorithm."""
 
-    def testSimulation(self):
-        """Test calculation of group liberties."""
-        board = GameBoard(TEST_BOARD_SIZE, TEST_BOARD_SIZE)
-        move = GameMove(board)
-        node = MCTSNode(move, None)
-        node.simulation()
+    def testPickMove(self):
+        """Test picking a move."""
+        state = GameState(TEST_BOARD_SIZE)
+        tree = MCTS(state)
+        print(tree.pickMove().toString())
+
+    #def testSimulation(self):
+    #    """Test calculation of group liberties."""
+    #    board = GameBoard(TEST_BOARD_SIZE, TEST_BOARD_SIZE)
+    #    move = GameMove(board)
+    #    node = MCTSNode(move, None)
+    #    nMatches = 100
+    #    scoreDiffHeur = 15
+    #    node.simulation(nMatches, scoreDiffHeur)
 
 if __name__ == '__main__':
     unittest.main()