from test.base import ChessLibTestBase
from test.chess_ffi import Board, perft


class TestPerft(ChessLibTestBase):
    def test_start_position(self):
        b = Board()
        self.load_fen("rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR w KQkq - 0 1", board=b)
        expected = {
            1: 20,
            2: 400,
            3: 8902,
            #4: 197281,
            # 5: 4865609,
            # 6: 119060324
        }
        for d, want in expected.items():
            got = int(perft(b, d))
            self.assertEqual(got, want, f"perft({d}) start pos")

    """
    def test_kiwipete(self):
        b = Board()
        self.load_fen("r3k2r/p1ppqpb1/bn2pnp1/3PN3/1p2P3/2N2Q1p/PPPBBPPP/R3K2R w KQkq - 0 1", board=b)
        expected = {
            1: 48,
            2: 2039,
            3: 97862
        }
        for d, want in expected.items():
            got = int(perft(b, d))
            self.assertEqual(got, want, f"perft({d}) kiwipete")
    """