fix stalemate issue

guidopetri · Aug 26, 2024 · df80a98 · df80a98
1 parent 3b004ab
commit df80a98
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/src/pipeline_import/transforms.py b/src/pipeline_import/transforms.py
@@ -78,6 +78,8 @@ def get_sf_evaluation(fen: str,
                 rating = 9999
             else:
                 rating = -9999
+        elif board.is_stalemate():
+            rating = 0
         else:
             raise ValueError('No best move found and not a checkmate position '
                              f'for: {fen=} {sf.info=}')

diff --git a/tests/test_transforms.py b/tests/test_transforms.py
@@ -330,6 +330,21 @@ def test_get_sf_evaluation_checkmate_white():
     assert rating == 9999
 
 
+def test_get_sf_evaluation_in_stalemate():
+
+    fen = '3Q4/8/8/8/8/3QK2P/8/4k3 b - - 0 56'
+
+    cfg = ConfigParser()
+    cfg.read('luigi.cfg')
+    stockfish_loc = cfg['stockfish_cfg']['location']
+
+    depth = 20
+
+    rating = transforms.get_sf_evaluation(fen, stockfish_loc, depth)
+
+    assert rating == 0
+
+
 def test_get_sf_evaluation_in_checkmate():
 
     fen = '4Rb1k/7Q/8/1p4N1/p7/8/1P4PK/8 b - - 4 0'