From fb5e8bb0cbb5fe394ee9954465efa950c2f1fb60 Mon Sep 17 00:00:00 2001
From: Josh <josh@joshuaschuett.com>
Date: Sun, 17 Aug 2025 19:02:00 -0400
Subject: [PATCH] Add knight tests

---
 test/test_movegen.py | 69 +++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 68 insertions(+), 1 deletion(-)

diff --git a/test/test_movegen.py b/test/test_movegen.py
index 36048ca..dbbfccd 100644
--- a/test/test_movegen.py
+++ b/test/test_movegen.py
@@ -4,6 +4,7 @@ from test.chess_ffi import gen_white_pawn_quiet_pushes
 from test.chess_ffi import gen_white_pawn_push_promotions
 from test.chess_ffi import gen_white_pawn_capture_promotions
 from test.chess_ffi import gen_white_pawn_captures
+from test.chess_ffi import gen_knight_moves
 from test.chess_ffi import Move
 from test.chess_ffi import Board
 
@@ -11,7 +12,73 @@ from test.chess_ffi import Board
 MAX_MOVES = 256
 
 
-class TestMoveGeneration(ChessLibTestBase):
+class TestPseudoMoveGeneration(ChessLibTestBase):
+    def test_knight_move_gen(self):
+        all_move_types = [
+            ("8/8/8/8/3N4/8/8/8 w - - 0 1", 8, "center d4: 8 moves"),
+            ("8/8/8/8/N7/8/8/8 w - - 0 1",   4, "edge a4: 4 moves"),
+            ("8/8/8/8/8/8/8/N7 w - - 0 1",   2, "corner a1: 2 moves"),
+            ("8/8/8/8/8/8/8/1N6 w - - 0 1",  3, "b1: 3 moves"),
+            
+            # own pieces on destinations reduce count
+            ("8/8/8/5P2/3N4/1P6/2P5/8 w - - 0 1", 5, "d4 with own pawns on f5,b3,c2 -> 8-3=5"),
+            
+            # two knights
+            ("8/8/8/8/8/8/8/1N4N1 w - - 0 1", 6, "b1 & g1, empty board: 3+3"),
+            
+            # all 8 destinations occupied by black (still 8 total moves, all as captures)
+            ("8/8/2p1p3/1p3p2/3N4/1p3p2/2p1p3/8 w - - 0 1", 8, "enemy on all 8 targets"),
+            
+            # corner with one capture and one friendly block
+            ("8/8/8/8/8/1p6/2P5/N7 w - - 0 1", 1, "a1, b3 black (capture), c2 white (blocked)"),
+        ]
+
+        captures_only = False
+        for fen, expected, msg in all_move_types:
+            with self.subTest(msg=msg, fen=fen):
+                cnt = ctypes.c_int(0)
+                moves = (Move * MAX_MOVES)()
+                b = Board()
+
+                self.load_fen(fen, board=b)
+                gen_knight_moves(b, moves, ctypes.byref(cnt), captures_only)
+
+                self.assertEqual(expected, cnt.value)
+
+        cases_knight_captures = [
+            ("8/8/8/8/3N4/8/8/8 w - - 0 1", 0, "center d4 but no enemies"),
+            ("8/8/2p5/1p6/3N4/5p2/8/8 w - - 0 1", 3, "d4 capturing c6,b5,f3"),
+            ("8/8/8/8/3N4/1P3p2/4P3/8 w - - 0 1", 1, "d4 with f3 black (1 cap), b3/e2 white (not capturable)"),
+            ("8/8/2p1p3/1p3p2/3N4/1p3p2/2p1p3/8 w - - 0 1", 8, "d4 with all 8 targets black: 8 captures"),
+        ]
+
+        captures_only = True
+        for fen, expected, msg in cases_knight_captures:
+            with self.subTest(msg=msg, fen=fen):
+                cnt = ctypes.c_int(0)
+                moves = (Move * MAX_MOVES)()
+                b = Board()
+
+                self.load_fen(fen, board=b)
+                gen_knight_moves(b, moves, ctypes.byref(cnt), captures_only)
+
+                self.assertEqual(expected, cnt.value)
+
+
+    def test_bishop_move_gen(self):
+        pass
+
+
+    def test_rook_move_gen(self):
+        pass
+
+
+    def test_queen_move_gen(self):
+        pass
+
+
+    def test_king_move_gen(self):
+        pass
 
 
     def test_quiet_pawn_pushes_white(self):