def test_queen_moves():
board = Board()
# Check move generation when queen is on A1
queen = Queen(Square.from_notation("a1"), board)
moves = queen.get_moves()
assert len(moves) == 21
assert all(isinstance(move, Square) for move in moves)
assert Square.from_notation("h1") in moves
assert Square.from_notation("a8") in moves
assert Square.from_notation("h8") in moves
# Check move generation when queen is on D4
queen = Queen(Square.from_notation("d4"), board)
moves = queen.get_moves()
assert len(moves) == 27
assert all(isinstance(move, Square) for move in moves)
assert Square.from_notation("a4") in moves
assert Square.from_notation("h4") in moves
assert Square.from_notation("d8") in moves
assert Square.from_notation("d1") in moves
assert Square.from_notation("b2") in moves
assert Square.from_notation("h8") in moves
assert Square.from_notation("a7") in moves
assert Square.from_notation("g1") in moves
# Check move generation when queen is on D4
# and there is a blockin piece on F4
queen = Queen(Square.from_notation("d4"), board)
blocking_bishop1 = Bishop(Square.from_notation("c4"), board)
blocking_bishop2 = Bishop(Square.from_notation("c3"), board)
board.add_piece(blocking_bishop1)
board.add_piece(blocking_bishop2)
moves = queen.get_moves()
assert len(moves) == 21
assert all(isinstance(move, Square) for move in moves)
assert Square.from_notation("h4") in moves
assert Square.from_notation("e4") in moves
assert Square.from_notation("d1") in moves
assert Square.from_notation("d8") in moves
assert Square.from_notation("f4") in moves
assert Square.from_notation("b6") in moves
assert Square.from_notation("g1") in moves
assert Square.from_notation("a7") in moves
assert Square.from_notation("a1") not in moves
assert Square.from_notation("a4") not in moves
def test_bishop_moves():
board = Board()
# Check move generation when bishop is on A1
bishop = Bishop(Square.from_notation("a1"), board)
moves = bishop.get_moves()
assert len(moves) == 7
assert all(isinstance(move, Square) for move in moves)
assert Square.from_notation("b2") in moves
assert Square.from_notation("h8") in moves
# Check move generation when bishop is on D4
bishop = Bishop(Square.from_notation("d4"), board)
moves = bishop.get_moves()
assert len(moves) == 13
assert all(isinstance(move, Square) for move in moves)
assert Square.from_notation("b2") in moves
assert Square.from_notation("h8") in moves
assert Square.from_notation("a7") in moves
assert Square.from_notation("g1") in moves
# Check move generation when bishop is on E4
# and there is a blockin piece on F5
bishop = Bishop(Square.from_notation("e4"), board)
blocking_bishop = Bishop(Square.from_notation("f5"), board)
board.add_piece(blocking_bishop)
moves = bishop.get_moves()
assert len(moves) == 10
assert all(isinstance(move, Square) for move in moves)
assert Square.from_notation("b1") in moves
assert Square.from_notation("h1") in moves
assert Square.from_notation("a8") in moves
assert Square.from_notation("h7") not in moves
def _set_pieces(self, row, colour):
king = King(colour)
self._board[row][king.get_x_pos()] = king
queen = Queen(colour)
self._board[row][queen.get_x_pos()] = queen
rook = Rook(colour, [row, 0])
self._board[row][0] = rook
rook = Rook(colour, [row, 7])
self._board[row][7] = rook
knight = Knight(colour, [row, 1])
self._board[row][1] = knight
knight = Knight(colour, [row, 6])
self._board[row][6] = knight
bishop = Bishop(colour, [row, 2])
self._board[row][2] = bishop
bishop = Bishop(colour, [row, 5])
self._board[row][5] = bishop
def test_king_moves():
board = Board()
# Check move generation when king is on A1
king = King(Square.from_notation("a1"), board)
moves = king.get_moves()
assert len(moves) == 3
assert all(isinstance(move, Square) for move in moves)
assert Square(file=0, rank=1) in moves
assert Square.from_notation("b1") in moves
assert Square.from_notation("b1") in moves
# Check move generation when king is on D4
king = King(Square.from_notation("d4"), board)
moves = king.get_moves()
assert len(moves) == 8
assert all(isinstance(move, Square) for move in moves)
assert Square.from_notation("c3") in moves
assert Square.from_notation("c4") in moves
assert Square.from_notation("c5") in moves
assert Square.from_notation("d3") in moves
assert Square.from_notation("d5") in moves
assert Square.from_notation("e3") in moves
assert Square.from_notation("e4") in moves
assert Square.from_notation("e5") in moves
# Check move generation when king is on D4
# and there is a blockin piece on C5
king = King(Square.from_notation("d4"), board)
blocking_bishop = Bishop(Square.from_notation("c5"), board)
board.add_piece(blocking_bishop)
moves = king.get_moves()
assert len(moves) == 7
assert all(isinstance(move, Square) for move in moves)
assert Square.from_notation("c3") in moves
assert Square.from_notation("c4") in moves
assert Square.from_notation("c5") not in moves
assert Square.from_notation("d3") in moves
assert Square.from_notation("d5") in moves
assert Square.from_notation("e3") in moves
assert Square.from_notation("e4") in moves
assert Square.from_notation("e5") in moves
def test_knight_moves():
board = Board()
# Check move generation when knight is on A1
knight = Knight(Square.from_notation("a1"), board)
moves = knight.get_moves()
assert len(moves) == 2
assert all(isinstance(move, Square) for move in moves)
assert Square.from_notation("b3") in moves
assert Square.from_notation("c2") in moves
# Check move generation when knight is on D4
knight = Knight(Square.from_notation("d4"), board)
moves = knight.get_moves()
assert len(moves) == 8
assert all(isinstance(move, Square) for move in moves)
assert Square.from_notation("b3") in moves
assert Square.from_notation("b5") in moves
assert Square.from_notation("c6") in moves
assert Square.from_notation("c2") in moves
assert Square.from_notation("e2") in moves
assert Square.from_notation("e6") in moves
assert Square.from_notation("f5") in moves
assert Square.from_notation("f3") in moves
# Check move generation when knight is on D4
# and there is a blockin piece on C2
knight = Knight(Square.from_notation("d4"), board)
blocking_bishop = Bishop(Square.from_notation("c2"), board)
board.add_piece(blocking_bishop)
moves = knight.get_moves()
assert len(moves) == 7
assert all(isinstance(move, Square) for move in moves)
assert Square.from_notation("b3") in moves
assert Square.from_notation("b5") in moves
assert Square.from_notation("c6") in moves
assert Square.from_notation("c2") not in moves
assert Square.from_notation("e2") in moves
assert Square.from_notation("e6") in moves
assert Square.from_notation("f5") in moves
assert Square.from_notation("f3") in moves
def test_get_piece_that_reaches_square():
board = Board()
bishop = Bishop(Square.from_notation("a1"), board)
knight = Knight(Square.from_notation("b1"), board)
board.add_piece(bishop)
board.add_piece(knight)
square = Square.from_notation("a3")
assert isinstance(board.get_piece_that_reaches_square(square), Knight)
square = Square.from_notation("b2")
assert isinstance(board.get_piece_that_reaches_square(square), Bishop)
with raises(ValueError):
square = Square.from_notation("e2")
_ = board.get_piece_that_reaches_square(square)
def test_get_singular_squares():
board = Board()
bishop = Bishop(Square.from_notation("a1"), board)
knight = Knight(Square.from_notation("b1"), board)
board.add_piece(bishop)
board.add_piece(knight)
squares = board.get_singular_squares()
assert isinstance(squares, list)
assert all(isinstance(square, Square) for square in squares)
assert len(squares) == 8
assert Square.from_notation("b2") in squares
assert Square.from_notation("h8") in squares
assert Square.from_notation("c3") not in squares
assert Square.from_notation("a3") in squares
assert Square.from_notation("d2") in squares
def test_rook_moves():
board = Board()
# Check move generation when rook is on A1
rook = Rook(Square.from_notation("a1"), board)
moves = rook.get_moves()
assert len(moves) == 14
assert all(isinstance(move, Square) for move in moves)
assert Square.from_notation("h1") in moves
assert Square.from_notation("a8") in moves
# Check move generation when rook is on D4
rook = Rook(Square.from_notation("d4"), board)
moves = rook.get_moves()
assert len(moves) == 14
assert all(isinstance(move, Square) for move in moves)
assert Square.from_notation("a4") in moves
assert Square.from_notation("h4") in moves
assert Square.from_notation("d8") in moves
assert Square.from_notation("d1") in moves
# Check move generation when rook is on D4
# and there is a blockin piece on F4
rook = Rook(Square.from_notation("d4"), board)
blocking_bishop = Bishop(Square.from_notation("f4"), board)
board.add_piece(blocking_bishop)
moves = rook.get_moves()
assert len(moves) == 11
assert all(isinstance(move, Square) for move in moves)
assert Square.from_notation("a4") in moves
assert Square.from_notation("e4") in moves
assert Square.from_notation("d1") in moves
assert Square.from_notation("d8") in moves
assert Square.from_notation("f4") not in moves