def test_board_move_gen(run_movegen_test):
if not run_movegen_test:
pytest.skip("This can take a while.")
board = Chessboard(chess.STARTING_FEN)
perft_results = (1, 20, 400, 8902, 197281, 4865609)
for i in range(int(run_movegen_test)):
assert board.perft(i) == perft_results[i]
def test_board_repr():
board = Chessboard()
board.set_fen(chess.STARTING_FEN)
assert str(board) == ("\nr n b q k b n r "
"\np p p p p p p p "
"\n. . . . . . . . "
"\n. . . . . . . . "
"\n. . . . . . . . "
"\n. . . . . . . . "
"\nP P P P P P P P "
"\nR N B Q K B N R ")
board.set_fen("3q4/kPpP4/4b1p1/2PP1n2/K3B3/1p6/8/4N1bN w - - 0 1")
assert str(board) == ("\n. . . q . . . . "
"\nk P p P . . . . "
"\n. . . . b . p . "
"\n. . P P . n . . "
"\nK . . . B . . . "
"\n. p . . . . . . "
"\n. . . . . . . . "
"\n. . . . N . b N ")
board.set_fen("QBr3N1/1K6/R2ppP2/1n6/P3k3/6p1/1p5P/1b6 w - - 0 1")
assert str(board) == ("\nQ B r . . . N . "
"\n. K . . . . . . "
"\nR . . p p P . . "
"\n. n . . . . . . "
"\nP . . . k . . . "
"\n. . . . . . p . "
"\n. p . . . . . P "
"\n. b . . . . . . ")
def test_board_set_fen():
board = Chessboard()
fen_strings = (
{
"fen": "2k1b3/5P2/PKp2R1R/pp5P/3P4/8/B5r1/2BN4 w - - 0 1",
"turn": chess.WHITE,
},
{
"fen": "r5R1/3B4/3p3P/P7/1pN5/2pn1K2/N1k4r/R4n2 b - - 0 1",
"turn": chess.BLACK,
},
{
"fen": "2N5/3R4/p3k3/1R3pKp/3qp1b1/8/P6P/5NQn w - - 0 1",
"turn": chess.WHITE
},
)
for position in fen_strings:
board.set_fen(position["fen"])
assert board.fen == position["fen"]
assert board.turn == position["turn"]
def test_board_move():
board = Chessboard(chess.STARTING_FEN)
moves = (
chess.Move(
initial_square=96,
target_square=64,
moving_piece="P",
attacked_piece=".",
capture=False,
score=None,
),
chess.Move(
initial_square=16,
target_square=48,
moving_piece="p",
attacked_piece=".",
capture=False,
score=None,
),
chess.Move(
initial_square=112,
target_square=48,
moving_piece="R",
attacked_piece="p",
capture=True,
score=None,
),
)
for move in moves:
board.move(move)
assert move.attacked_piece != board.board[move.target_square]
assert move.moving_piece == board.board[move.target_square]
assert board.board[move.initial_square] == "."
def test_board_undo_move():
board = Chessboard(chess.STARTING_FEN)
og_board = board.board
moves = (
chess.Move(
initial_square=96,
target_square=64,
moving_piece="P",
attacked_piece=".",
capture=False,
score=None,
),
chess.Move(
initial_square=16,
target_square=48,
moving_piece="p",
attacked_piece=".",
capture=False,
score=None,
),
chess.Move(
initial_square=113,
target_square=80,
moving_piece="N",
attacked_piece=".",
capture=False,
score=None,
),
)
for move in moves:
board.move(move)
assert board.board != og_board
board.undo_move()
assert board.board == og_board
def test_board_set_starting_fen_by_default():
board = Chessboard()
assert board.fen == chess.STARTING_FEN
def test_board_is_square_attacked():
fens = {
"8/5K2/8/7k/8/8/8/8 w - - 0 1": (38, 39, 54, 70, 71),
"3Q4/3p2p1/4pp2/K7/bPp1BP1B/2P2p2/8/2k3r1 w - - 0 1": (
34,
38,
39,
49,
51,
52,
53,
54,
70,
81,
83,
86,
97,
98,
99,
100,
102,
113,
115,
116,
117,
119,
),
"3B4/PP2r3/1rp5/P3k1Pb/2p4p/3P2qN/8/K7 w - - 0 1": (
4,
17,
18,
19,
21,
22,
23,
32,
35,
36,
37,
38,
49,
51,
53,
54,
65,
67,
68,
69,
70,
81,
83,
84,
85,
87,
97,
100,
101,
102,
103,
113,
115,
116,
118,
),
}
for fen in fens:
attacked_squares = fens[fen]
board = Chessboard(fen)
for square in [i for i in range(128) if (i & 0x88) == 0]:
if board.is_square_attacked(square):
assert square in attacked_squares
def test_board_generate_fen():
with open("tests/FEN/endgames.txt", "r") as endgames_fen_text_file:
for fen in endgames_fen_text_file.readlines():
board = Chessboard(fen)
assert board.generate_fen().strip() == fen.strip()