Merge branch 'master' into ease-development

stockfish/models.py

@@ -706,11 +706,18 @@ def is_move_correct(self, move_value: str) -> bool:
         self.info = old_self_info
         return is_move_correct
 
-    def get_wdl_stats(self) -> Optional[List[int]]:
+    def get_wdl_stats(
+        self, get_as_tuple: bool = False
+    ) -> Union[list[int] | tuple[int, int, int] | None]:
         """Returns Stockfish's win/draw/loss stats for the side to move.
 
+        Args:
+            get_as_tuple:
+                Option to return the wdl stats as a tuple instead of a list
+                `Boolean`. Default is `False`.
+
         Returns:
-            A list of three integers, unless the game is over (in which case
+            A list or tuple of three integers, unless the game is over (in which case
             `None` is returned).
         """
 
@@ -730,7 +737,12 @@ def get_wdl_stats(self) -> Optional[List[int]]:
             return None
         split_line = [line.split(" ") for line in lines if " multipv 1 " in line][-1]
         wdl_index = split_line.index("wdl")
-        return [int(split_line[i]) for i in range(wdl_index + 1, wdl_index + 4)]
+
+        wdl_stats = [int(split_line[i]) for i in range(wdl_index + 1, wdl_index + 4)]
+
+        if get_as_tuple:
+            return (wdl_stats[0], wdl_stats[1], wdl_stats[2])
+        return wdl_stats
 
     def does_current_engine_version_have_wdl_option(self) -> bool:
         """Returns whether the user's version of Stockfish has the option

tests/stockfish/test_models.py

@@ -914,6 +914,11 @@ def test_get_wdl_stats(self, stockfish: Stockfish):
             wdl_stats_3 = stockfish.get_wdl_stats()
             assert isinstance(wdl_stats_3, list) and len(wdl_stats_3) == 3
 
+            wdl_stats_4 = stockfish.get_wdl_stats(get_as_tuple=True)
+            assert isinstance(wdl_stats_4, tuple) and len(wdl_stats_4) == 3
+            assert wdl_stats_3 == list(wdl_stats_4)
+            assert tuple(wdl_stats_3) == wdl_stats_4
+
             stockfish.set_fen_position("8/8/8/8/8/3k4/3p4/3K4 w - - 0 1")
             assert stockfish.get_wdl_stats() is None