Add perft testing
This commit is contained in:
33
test/test_perft.py
Normal file
33
test/test_perft.py
Normal file
@@ -0,0 +1,33 @@
|
||||
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")
|
||||
"""
|
||||
Reference in New Issue
Block a user