101 lines
3.8 KiB
Rust
101 lines
3.8 KiB
Rust
use chess_engine::board::Board;
|
|
use chess_engine::movegen::sliders::generate_rook_moves;
|
|
use chess_engine::r#move::MoveList;
|
|
|
|
/// Compares two move list strings ignoring the order of moves.
|
|
fn assert_moves_equal(actual_str: &str, expected_str: &str) {
|
|
let mut actual_moves: Vec<&str> = actual_str.split_whitespace().collect();
|
|
let mut expected_moves: Vec<&str> = expected_str.split_whitespace().collect();
|
|
|
|
actual_moves.sort();
|
|
expected_moves.sort();
|
|
|
|
assert_eq!(actual_moves, expected_moves);
|
|
}
|
|
|
|
#[test]
|
|
fn test_rook_moves_single_rook() {
|
|
let fen_standard = "8/8/8/2b5/2Rb4/2b5/8/8 w - - 0 1";
|
|
let board = Board::from_fen(fen_standard);
|
|
let mut list = MoveList::new();
|
|
generate_rook_moves(&board, &mut list);
|
|
// This FEN has a White Rook at c4, and Black Bishops at c5, d4, and c3.
|
|
// It should be able to capture all three and move left to a4/b4.
|
|
assert_moves_equal(&list.to_string(), "c4a4 c4b4 c4c3 c4c5 c4d4");
|
|
}
|
|
|
|
#[test]
|
|
fn test_rook_moves_empty_board() {
|
|
let fen = "8/8/8/8/3R4/8/8/8 w - - 0 1";
|
|
let board = Board::from_fen(fen);
|
|
let mut list = MoveList::new();
|
|
generate_rook_moves(&board, &mut list);
|
|
let expected = "d4d1 d4d2 d4d3 d4d5 d4d6 d4d7 d4d8 d4a4 d4b4 d4c4 d4e4 d4f4 d4g4 d4h4";
|
|
assert_moves_equal(&list.to_string(), expected);
|
|
}
|
|
|
|
#[test]
|
|
fn test_rook_moves_corner_blocked_black() {
|
|
let fen = "r6k/1p6/8/8/8/8/8/K7 b - - 0 1";
|
|
let board = Board::from_fen(fen);
|
|
let mut list = MoveList::new();
|
|
generate_rook_moves(&board, &mut list);
|
|
// Black rook at a8. Friendly king at h8, friendly pawn at b7.
|
|
// Rook can move down the a-file and right along the 8th rank, stopping before h8.
|
|
let expected = "a8a7 a8a6 a8a5 a8a4 a8a3 a8a2 a8a1 a8b8 a8c8 a8d8 a8e8 a8f8 a8g8";
|
|
assert_moves_equal(&list.to_string(), expected);
|
|
}
|
|
|
|
#[test]
|
|
fn test_rook_moves_double_rooks_friendly_block() {
|
|
let fen = "8/8/8/8/R3R3/8/8/8 w - - 0 1";
|
|
let board = Board::from_fen(fen);
|
|
let mut list = MoveList::new();
|
|
generate_rook_moves(&board, &mut list);
|
|
// Rooks at a4 and e4. They block each other horizontally.
|
|
// Rook a4 moves a1-a8 and b4, c4, d4.
|
|
// Rook e4 moves e1-e8, f4, g4, h4 AND d4, c4, b4.
|
|
let expected = "a4a1 a4a2 a4a3 a4a5 a4a6 a4a7 a4a8 a4b4 a4c4 a4d4 \
|
|
e4e1 e4e2 e4e3 e4e5 e4e6 e4e7 e4e8 e4f4 e4g4 e4h4 \
|
|
e4d4 e4c4 e4b4";
|
|
assert_moves_equal(&list.to_string(), expected);
|
|
}
|
|
|
|
#[test]
|
|
fn test_rook_moves_capture_stops_movegen() {
|
|
let fen = "r7/P7/8/8/8/8/8/8 b - - 0 1";
|
|
let board = Board::from_fen(fen);
|
|
let mut list = MoveList::new();
|
|
generate_rook_moves(&board, &mut list);
|
|
// Black rook at a8, White pawn at a7.
|
|
// The rook can capture at a7, but cannot move past it.
|
|
// It can still move horizontally.
|
|
let expected = "a8a7 a8b8 a8c8 a8d8 a8e8 a8f8 a8g8 a8h8";
|
|
assert_moves_equal(&list.to_string(), expected);
|
|
}
|
|
|
|
#[test]
|
|
fn test_rook_moves_completely_blocked_friendly() {
|
|
let fen = "8/8/8/1P6/PRP5/1P6/8/8 w - - 0 1";
|
|
let board = Board::from_fen(fen);
|
|
let mut list = MoveList::new();
|
|
generate_rook_moves(&board, &mut list);
|
|
// White rook at b4.
|
|
// Blocked by P(b5), P(b3), P(a4), P(c4).
|
|
// Should have 0 moves.
|
|
assert_moves_equal(&list.to_string(), "");
|
|
}
|
|
|
|
#[test]
|
|
fn test_rook_moves_ignores_absolute_pin() {
|
|
let fen = "r3k3/8/8/8/R3K3/8/8/8 w - - 0 1";
|
|
let board = Board::from_fen(fen);
|
|
let mut list = MoveList::new();
|
|
generate_rook_moves(&board, &mut list);
|
|
// White rook at a4 is absolutely pinned to King at e4 by Black rook at a8.
|
|
// A pseudo-legal generator should *ignore* the pin.
|
|
// It should generate vertical moves (including capture at a8)
|
|
// and horizontal moves (stopping before the friendly King at e4).
|
|
let expected = "a4a1 a4a2 a4a3 a4a5 a4a6 a4a7 a4a8 a4b4 a4c4 a4d4";
|
|
assert_moves_equal(&list.to_string(), expected);
|
|
}
|