test with 1 top move

py-stockfish · Aug 29, 2024 · 2853afb · 2853afb
1 parent ebb72a3
commit 2853afb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)