Josh
6005741b10
All checks were successful
Python tests (make) / test (push) Successful in 14s
Reviewed-on: #29 Co-authored-by: Josh <josh@joshuaschuett.com> Co-committed-by: Josh <josh@joshuaschuett.com>
95 lines
2.9 KiB
Python
95 lines
2.9 KiB
Python
from test.base import ChessLibTestBase
|
|
from binding.python_c_ffi import Board
|
|
|
|
|
|
class TestPerftQuick(ChessLibTestBase):
|
|
def test_perft_1(self):
|
|
b = Board()
|
|
fen = "rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR w KQkq - 0 1"
|
|
self.chess_ffi.load_fen(b, fen)
|
|
node_count = {
|
|
1: 20,
|
|
2: 400,
|
|
3: 8902,
|
|
4: 197281,
|
|
5: 4865609,
|
|
}
|
|
for node, count in node_count.items():
|
|
actual = int(self.chess_ffi._c_perft(b, node))
|
|
self.assertEqual(count, actual, f"perft({node}) start pos")
|
|
|
|
|
|
def test_perft_2(self):
|
|
b = Board()
|
|
fen = "r3k2r/p1ppqpb1/bn2pnp1/3PN3/1p2P3/2N2Q1p/PPPBBPPP/R3K2R w KQkq - 0 1"
|
|
self.chess_ffi.load_fen(b, fen)
|
|
node_count = {
|
|
1: 48,
|
|
2: 2039,
|
|
3: 97862,
|
|
4: 4085603,
|
|
}
|
|
for node, count in node_count.items():
|
|
actual = int(self.chess_ffi._c_perft(b, node))
|
|
self.assertEqual(count, actual, f"perft({node}) start pos")
|
|
|
|
|
|
def test_perft_3(self):
|
|
b = Board()
|
|
fen = "8/2p5/3p4/KP5r/1R3p1k/8/4P1P1/8 w - - 0 1"
|
|
self.chess_ffi.load_fen(b, fen)
|
|
node_count = {
|
|
1: 14,
|
|
2: 191,
|
|
3: 2812,
|
|
4: 43238,
|
|
5: 674624,
|
|
}
|
|
for node, count in node_count.items():
|
|
actual = int(self.chess_ffi._c_perft(b, node))
|
|
self.assertEqual(count, actual, f"perft({node}) start pos")
|
|
|
|
|
|
def test_perft_4(self):
|
|
b = Board()
|
|
fen = "r3k2r/Pppp1ppp/1b3nbN/nP6/BBP1P3/q4N2/Pp1P2PP/R2Q1RK1 w kq - 0 1"
|
|
self.chess_ffi.load_fen(b, fen)
|
|
node_count = {
|
|
1: 6,
|
|
2: 264,
|
|
3: 9467,
|
|
4: 422333,
|
|
}
|
|
for node, count in node_count.items():
|
|
actual = int(self.chess_ffi._c_perft(b, node))
|
|
self.assertEqual(count, actual, f"perft({node}) start pos")
|
|
|
|
|
|
def test_perft_5(self):
|
|
b = Board()
|
|
fen = "rnbq1k1r/pp1Pbppp/2p5/8/2B5/8/PPP1NnPP/RNBQK2R w KQ - 1 8"
|
|
self.chess_ffi.load_fen(b, fen)
|
|
node_count = {
|
|
1: 44,
|
|
2: 1486,
|
|
3: 62379,
|
|
4: 2103487,
|
|
}
|
|
for node, count in node_count.items():
|
|
actual = int(self.chess_ffi._c_perft(b, node))
|
|
self.assertEqual(count, actual, f"perft({node}) start pos")
|
|
|
|
|
|
def test_perft_6(self):
|
|
b = Board()
|
|
fen = "r4rk1/1pp1qppp/p1np1n2/2b1p1B1/2B1P1b1/P1NP1N2/1PP1QPPP/R4RK1 w - - 0 10"
|
|
self.chess_ffi.load_fen(b, fen)
|
|
node_count = {
|
|
1: 46,
|
|
2: 2079,
|
|
3: 89890,
|
|
4: 3894594,
|
|
}
|
|
for node, count in node_count.items():
|
|
actual = int(self.chess_ffi._c_perft(b, node))
|
|
self.assertEqual(count, actual, f"perft({node}) start pos") |