diff options
Diffstat (limited to 'tests/test_monteCarlo.py')
-rw-r--r-- | tests/test_monteCarlo.py | 30 |
1 files changed, 30 insertions, 0 deletions
diff --git a/tests/test_monteCarlo.py b/tests/test_monteCarlo.py new file mode 100644 index 0000000..b217cf9 --- /dev/null +++ b/tests/test_monteCarlo.py @@ -0,0 +1,30 @@ +"""Tests for the MonteCarlo algorithm.""" + +import unittest + +from imago.gameLogic.gameState import GameState +from imago.engine.monteCarlo import MCTS +from imago.engine.monteCarlo import MCTSNode + +TEST_BOARD_SIZE = 9 + +class TestMonteCarlo(unittest.TestCase): + """Test MonteCarlo algorithm.""" + + def testPickMove(self): + """Test picking a move.""" + state = GameState(TEST_BOARD_SIZE) + tree = MCTS(state.lastMove) + #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() |