Fix another missing class of filenames()

niklasf · Mar 18, 2015 · 08bec47 · 08bec47
1 parent c9af8d9
commit 08bec47
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/chess/syzygy.py b/chess/syzygy.py
@@ -299,7 +299,7 @@ def filenames():
 
     for i in range(1, 6):
         for j in range(i, 6):
-            for k in range(j, 6):
+            for k in range(i, 6):
                 for l in range(j if i == k else k, 6):
                     yield "K%c%cvK%c%c" % (PCHR[i], PCHR[j], PCHR[k], PCHR[l])
 

diff --git a/test.py b/test.py
@@ -1400,7 +1400,11 @@ def test_calc_key(self):
 
     def test_filenames(self):
         self.assertTrue("KPPvKN" in chess.syzygy.filenames())
-        self.assertTrue("KPPPvKN" in chess.syzygy.filenames())
+        self.assertTrue("KNNPvKN" in chess.syzygy.filenames())
+        self.assertTrue("KQRNvKR" in chess.syzygy.filenames())
+        self.assertTrue("KRRRvKR" in chess.syzygy.filenames())
+        self.assertTrue("KRRvKRR" in chess.syzygy.filenames())
+        self.assertTrue("KRNvKRP" in chess.syzygy.filenames())
 
     def test_probe_pawnless_wdl_table(self):
         wdl = chess.syzygy.WdlTable("data/syzygy", "KBNvK")