Add gaviota tests to standard test suite

niklasf · niklasf · commit 8cd2bf8a9b90 · 2015-10-01T13:46:07.000+02:00
diff --git a/chess/gaviota.py b/chess/gaviota.py
@@ -1778,7 +1778,7 @@ def a8toa1_init():
 
 a8toa1 = a8toa1_init()
 
-class PythonTableBase(object):
+class PythonTablebases(object):
     """ Provide access to Gaviota tablebases via full python code."""
 
     def __init__(self,directory):
diff --git a/data/endgame-dm.epd b/data/endgame-dm.epd
diff --git a/test.py b/test.py
@@ -2258,6 +2258,39 @@ def test_native_probe_wdl(self):
         self.assertEqual(self.tablebases.probe_wdl(board), 1)
 
 
+class GaviotaTestCase(unittest.TestCase):
+
+    def setUp(self):
+        self.tablebases = chess.gaviota.open_tablebases("data/gaviota", LibraryLoader=None)
+
+    def tearDown(self):
+        self.tablebases.close()
+
+    def test_dm(self):
+        with open("data/endgame-dm.epd") as epds:
+            for line, epd in enumerate(epds):
+                # Skip empty lines and comments.
+                epd = epd.strip()
+                if not epd or epd.startswith("#"):
+                    continue
+
+                # Parse EPD.
+                board, extra = chess.Board.from_epd(epd)
+
+                # Skip 5 and 6 piece endgames.
+                if chess.pop_count(board.occupied) > 4:
+                    continue
+
+                # Check DTM.
+                if extra["dm"] > 0:
+                    expected = extra["dm"] * 2 - 1
+                else:
+                    expected = extra["dm"] * 2
+                dtm = self.tablebases.probe_dtm(board).dtm
+                self.assertEqual(dtm, expected,
+                    "Expecting dtm {0} for {1}, got {2} (at line {3})".format(expected, board.fen(), dtm, line + 1))
+
+
 if __name__ == "__main__":
     if "-v" in sys.argv or "--verbose" in sys.argv:
         logging.basicConfig(level=logging.DEBUG)
diff --git a/test_gaviota.py b/test_gaviota.py
@@ -5,10 +5,9 @@
 import chess
 import chess.gaviota
 
-#tables = chess.gaviota.open_tablebases("data/gaviota")
-tables = chess.gaviota.PythonTableBase("data/gaviota")
+tables = chess.gaviota.open_tablebases("data/gaviota", LibraryLoader=None)
 
-with open("data/long-endgames.epd") as epds:
+with open("data/endgame-dm.epd") as epds:
     for line, epd in enumerate(epds):
         # Skip empty lines and comments.
         epd = epd.strip()
