Add Pseudo Move Generator (#12)

work on #7

Reviewed-on: #12
Co-authored-by: Josh <josh@joshuaschuett.com>
Co-committed-by: Josh <josh@joshuaschuett.com>

test/base.py

@@ -1,79 +1,33 @@
-import ctypes
-import platform
 import unittest
-from pathlib import Path
-
-
-WHITE, BLACK = 0, 1
-FILES = {c:i for i,c in enumerate("abcdefgh")}
-
-
-def sq(name: str) -> int:
-    f = FILES[name[0].lower()]
-    r = int(name[1]) - 1
-    return f + 8 * r
-
-
-def bb_from(*algebraic):
-    m = 0
-    for s in algebraic:
-        m |= (1 << sq(s))
-    return m
-
-
-def popcount(x: int) -> int:
-    return x.bit_count()
-
-
-def draw_bb(mask: int, origin: int | None = None) -> str:
-    print("\n")
-    lines = []
-    for r in range(7, -1, -1):
-        row = []
-        for f in range(8):
-            sqi = r * 8 + f
-            bit = (mask >> sqi) & 1
-            if origin is not None and sqi == origin:
-                ch = 'O'
-            elif bit:
-                ch = 'x'
-            else:
-                ch = '.'
-            row.append(ch)
-        lines.append(f"{r+1} " + " ".join(row))
-    lines.append("  " + " ".join(FILES))
-    lines = "\n".join(lines)
-    print(lines, "\n")
+from test.chess_ffi import Board
+from test.chess_ffi import KING_ATTACKS
+from test.chess_ffi import KNIGHT_ATTACKS
+from test.chess_ffi import PAWN_ATTACKS
+from test.chess_ffi import gen_moves
+from test.chess_ffi import init_attack_caches
+from test.chess_ffi import load_fen
 
 
 class ChessLibTestBase(unittest.TestCase):
     @classmethod
     def setUpClass(cls):
-        root = Path(__file__).resolve().parents[1]
-        libname = "libchess.so"
-        libpath = root / "build" / libname
-        cls.lib = ctypes.CDLL(str(libpath))
+        init_attack_caches()
 
-        attack_cache_functions = [
-            "create_knight_attack_cache",
-            "create_pawn_attack_cache",
-            "create_king_attack_cache",
-        ]
+        cls.KNIGHT_ATTACKS = KNIGHT_ATTACKS
+        cls.KING_ATTACKS   = KING_ATTACKS
+        cls.PAWN_ATTACKS   = PAWN_ATTACKS
 
-        # init functions
-        for fn in attack_cache_functions:
-            getattr(cls.lib, fn).argtypes = []
-            getattr(cls.lib, fn).restype = None
-     
-        cls.lib.create_knight_attack_cache()
-        cls.lib.create_pawn_attack_cache()
-        cls.lib.create_king_attack_cache()
 
-        KnightArr = ctypes.c_uint64 * 64
-        KingArr   = ctypes.c_uint64 * 64
-        PawnRow   = ctypes.c_uint64 * 64
-        PawnArr   = PawnRow * 2
+    def setUp(self):
+        # This should be an empty board for each test in the suite.
+        self.board = Board()
 
-        cls.KNIGHT_ATTACKS = KnightArr.in_dll(cls.lib, "KNIGHT_ATTACKS")
-        cls.KING_ATTACKS   = KingArr.in_dll(cls.lib, "KING_ATTACKS")
-        cls.PAWN_ATTACKS   = PawnArr.in_dll(cls.lib, "PAWN_ATTACKS")
+
+    def load_fen(self, fen, board=None):
+        if board:
+            return load_fen(board, fen)
+        return load_fen(self.board, fen)
+
+
+    def gen(self, captures_only: bool = False, cap: int = 256):
+        return gen_moves(self.board, captures_only=captures_only, cap=cap)