added initial mcts

main.py

@@ -0,0 +1,45 @@
+import chess
+import chess.engine
+from mcts import MCTSNode
+import engine
+import eval
+
+
+def test_mcts(seed):
+    fools_mate = "rnbqkbnr/pppp1ppp/4p3/8/5PP1/8/PPPPP2P/RNBQKBNR b KQkq f3 0 2"
+    board = chess.Board(fools_mate)
+    mcts_root = MCTSNode(board)
+    mcts_root.build_tree()
+    sorted_moves = sorted(mcts_root.children, key=lambda x: x.move.uci())
+    for c in sorted_moves:
+        print("move (mcts):", c.move, " with score:", c.score)
+
+
+def test_stockfish(seed):
+    fools_mate = "rnbqkbnr/pppp1ppp/4p3/8/5PP1/8/PPPPP2P/RNBQKBNR b KQkq f3 0 2"
+    board = chess.Board(fools_mate)
+    moves = {}
+    untried_moves = list(board.legal_moves)
+    for move in untried_moves:
+        engine.simulate_game(board, move, 100)
+        moves[move] = board
+        board = chess.Board(fools_mate)
+
+    sorted_moves = dict(sorted(moves.items(), key=lambda x: x[0].uci()))
+    analyze_results(sorted_moves)
+
+
+def analyze_results(moves: dict):
+    for m, b in moves.items():
+        manual_score = eval.score_manual(b)
+        engine_score = eval.score_stockfish(b).white()
+        print(f"score for move {m}: manual_score={manual_score}, engine_score={engine_score}")
+
+
+def main():
+    test_mcts(0)
+    test_stockfish(0)
+
+
+if __name__ == '__main__':
+    main()