33 lines
971 B
Python
33 lines
971 B
Python
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")
|
|
""" |