From 2853afb506c696233bea2cbb648096d35d311528 Mon Sep 17 00:00:00 2001 From: johndoknjas Date: Wed, 28 Aug 2024 17:35:45 -0700 Subject: [PATCH] test with 1 top move --- tests/stockfish/test_models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/stockfish/test_models.py b/tests/stockfish/test_models.py index 5e682ca..dd758ec 100644 --- a/tests/stockfish/test_models.py +++ b/tests/stockfish/test_models.py @@ -23,7 +23,7 @@ def test_get_best_move_remaining_time_not_first_move(self, stockfish: Stockfish) stockfish.make_moves_from_current_position(["e2e4", "e7e6"]) best_move = stockfish.get_best_move(wtime=1000) assert best_move in ("d2d4", "a2a3", "d1e2", "b1c3") - assert stockfish.get_top_moves(num_nodes=203)[0]["Move"] in ("d2d4", "b1c3") + assert stockfish.get_top_moves(num_top_moves=1, btime=1000)[0]["Move"] in ("d2d4", "b1c3") best_move = stockfish.get_best_move(wtime=1000, btime=1000) assert best_move in ("d2d4", "b1c3", "g1f3") best_move = stockfish.get_best_move(wtime=5 * 60 * 1000, btime=1000)