def test_bishop_can_take_opponent(): # Arrange board = Board.empty() bishop = Bishop(Player.WHITE) square = Square.at(4, 4) board.set_piece(square, bishop) board.set_piece(Square.at(6, 6), Bishop(Player.BLACK)) # Act moves = bishop.get_available_moves(board) # Assert assert Square.at(6, 6) in moves
def test_bishop_can_move_until_blocked_by_teammate(): # Arrange board = Board.empty() bishop = Bishop(Player.WHITE) square = Square.at(4, 4) board.set_piece(square, bishop) board.set_piece(Square.at(6, 6), Bishop(Player.WHITE)) # Act moves = bishop.get_available_moves(board) # Assert assert Square.at(6, 6) not in moves
def test_bishop_can_move_diagonally(): # Arrange board = Board.empty() bishop = Bishop(Player.WHITE) square = Square.at(3, 5) board.set_piece(square, bishop) # Act moves = bishop.get_available_moves(board) # Assert expected_moves = [ Square.at(0, 2), Square.at(1, 3), Square.at(2, 4), Square.at(4, 6), Square.at(5, 7), Square.at(1, 7), Square.at(2, 6), Square.at(4, 4), Square.at(5, 3), Square.at(6, 2), Square.at(7, 1) ] assert len(moves) == len(expected_moves) assert set(moves) == set(expected_moves)
def test_bishop_can_move_diagonally(): # Arrange board = Board.empty() bishop = Bishop(Player.WHITE) square = Square.at(4, 4) board.set_piece(square, bishop) # Act moves = bishop.get_available_moves(board) # Assert assert Square.at(6, 6) in moves
def test_black_bishop_cannot_move_off_top_of_board(): # Arrange board = Board.empty() bishop = Bishop(Player.WHITE) square = Square.at(1, 7) board.set_piece(square, bishop) # Act moves = bishop.get_available_moves(board) # Assert assert Square.at(2, 8) not in moves assert Square.at(0, 8) not in moves
def test_white_bishops_moving_to_home_row_do_not_become_queens(): # Arrange board = Board.empty() bishop = Bishop(Player.WHITE) bishop_square = Square.at(1, 0) board.set_piece(bishop_square, bishop) target_square = Square.at(0, 1) # Act board.move_piece(bishop_square, target_square) # Assert assert isinstance(board.get_piece(target_square), Bishop) is True
def test_bishop_cannot_move_off_board(): # Arrange board = Board.empty() bishop = Bishop(Player.BLACK) bishop_square = Square.at(4, 4) board.set_piece(bishop_square, bishop) # Act moves = bishop.get_available_moves(board) # Assert assert Square.at(0, 0) in moves assert Square.at(7, 7) in moves assert Square.at(8, 8) not in moves assert Square.at(-1, -1) not in moves
def test_black_bishops_moving_to_home_row_do_not_become_queens(): # Arrange board = Board.empty() bishop = Bishop(Player.BLACK) bishop_square = Square.at(6, 0) board.set_piece(bishop_square, bishop) target_square = Square.at(7, 1) # Act board.current_player = board.current_player.opponent() board.move_piece(bishop_square, target_square) # Assert assert isinstance(board.get_piece(target_square), Bishop) is True
def test_bishop_can_move_down_and_right(): # Arrange board = Board.empty() bishop = Bishop(Player.WHITE) bishop_square = Square.at(4, 4) board.set_piece(bishop_square, bishop) # Act moves = bishop.get_available_moves(board) # Assert assert Square.at(3, 5) in moves assert Square.at(2, 6) in moves assert Square.at(1, 7) in moves
def test_bishop_can_move_up_and_left(): # Arrange board = Board.empty() bishop = Bishop(Player.WHITE) bishop_square = Square.at(4, 4) board.set_piece(bishop_square, bishop) # Act moves = bishop.get_available_moves(board) # Assert assert Square.at(5, 3) in moves assert Square.at(6, 2) in moves assert Square.at(7, 1) in moves
def test_black_bishop_can_move_diagonally(): # Arrange board = Board.empty() bishop = Bishop(Player.BLACK) bishop_square = Square.at(3, 3) board.set_piece(bishop_square, bishop) # Act moves = bishop.get_available_moves(board) # Assert assert Square.at(4, 4) in moves assert Square.at(2, 2) in moves assert Square.at(4, 2) in moves assert Square.at(2, 4) in moves
def test_bishop_cannot_move_through_pieces_down_and_left(): # Arrange board = Board.empty() bishop = Bishop(Player.WHITE) bishop_square = Square.at(4, 4) board.set_piece(bishop_square, bishop) friendly = Pawn(Player.WHITE) friendly_square = Square.at(3, 3) board.set_piece(friendly_square, friendly) # Act moves = bishop.get_available_moves(board) # Assert assert Square.at(2, 2) not in moves
def test_bishop_can_move_on_backslash(): # Arrange board = Board.empty() bishop = Bishop(Player.BLACK) bishop_square = Square.at(4, 4) board.set_piece(bishop_square, bishop) # Act moves = bishop.get_available_moves(board) # Assert assert Square.at(3, 5) in moves assert Square.at(2, 6) in moves assert Square.at(5, 3) in moves assert Square.at(6, 2) in moves
def test_bishop_cannot_move_off_board(): # Arrange board = Board.empty() bishop = Bishop(Player.WHITE) bishop_square = Square.at(0, 0) board.set_piece(bishop_square, bishop) friendly = Pawn(Player.WHITE) friendly_square = Square.at(1, 1) board.set_piece(friendly_square, friendly) # Act moves = bishop.get_available_moves(board) # Assert assert len(moves) == 0
def test_bishop_is_blocked_by_enemy_pieces(): # Arrange board = Board.empty() bishop = Bishop(Player.WHITE) square = Square.at(3, 5) board.set_piece(square, bishop) enemy = Pawn(Player.BLACK) enemy_square = Square.at(5, 3) board.set_piece(enemy_square, enemy) # Act moves = bishop.get_available_moves(board) # Assert assert Square.at(6, 2) not in moves
def test_bishop_can_capture_enemy_pieces(): # Arrange board = Board.empty() bishop = Bishop(Player.WHITE) square = Square.at(3, 5) board.set_piece(square, bishop) enemy = Pawn(Player.BLACK) enemy_square = Square.at(5, 7) board.set_piece(enemy_square, enemy) # Act moves = bishop.get_available_moves(board) # Assert assert enemy_square in moves
def test_bishop_cannot_capture_friendly_pieces(): # Arrange board = Board.empty() bishop = Bishop(Player.WHITE) square = Square.at(3, 5) board.set_piece(square, bishop) friendly = Pawn(Player.WHITE) friendly_square = Square.at(5, 3) board.set_piece(friendly_square, friendly) # Act moves = bishop.get_available_moves(board) # Assert assert friendly_square not in moves
def handle_promotion(self, moving_piece, to_square): if isinstance(moving_piece, Pawn): if to_square.row == 0 or to_square.row == 7: piece = input( "Select piece to promote pawn to: 'Q', 'K', 'R' or 'B'.") if piece == 'Q': self.set_piece(to_square, Queen(self.current_player)) elif piece == 'K': self.set_piece(to_square, Knight(self.current_player)) elif piece == 'R': self.set_piece(to_square, Rook(self.current_player)) elif piece == 'B': self.set_piece(to_square, Bishop(self.current_player)) else: self.set_piece(to_square, Queen(self.current_player)) return
def test_bishop_can_capture_down_and_left(): # Arrange board = Board.empty() bishop = Bishop(Player.WHITE) bishop_square = Square.at(4, 4) board.set_piece(bishop_square, bishop) enemy = Pawn(Player.BLACK) enemy_square = Square.at(2, 2) board.set_piece(enemy_square, enemy) # Act moves = bishop.get_available_moves(board) # Assert assert Square.at(2, 2) in moves
def test_bishop_is_blocked_by_friendly_pieces(): # Arrange board = Board.empty() bishop = Bishop(Player.WHITE) square = Square.at(3, 5) board.set_piece(square, bishop) friendly = Pawn(Player.WHITE) friendly_square = Square.at(5, 3) board.set_piece(friendly_square, friendly) # Act moves = bishop.get_available_moves(board) # Assert assert Square.at(6, 2) not in moves assert Square.at(7, 7) not in moves
def test_white_king_checked_by_black_bishop(): # Arrange board = Board.empty() king = King(Player.WHITE) king_square = Square.at(2, 3) board.set_piece(king_square, king) enemy = Bishop(Player.BLACK) enemy_square = Square.at(1, 2) board.set_piece(enemy_square, enemy) # Act # check if current square is in check? check = king.is_in_check(board) # Assert assert check is True
def test_bishop_cannot_move_if_piece_in_front(): # Arrange board = Board.empty() bishop = Bishop(Player.WHITE) bishop_square = Square.at(4, 4) board.set_piece(bishop_square, bishop) enemy = Pawn(Player.BLACK) enemy_square = Square.at(5, 5) board.set_piece(enemy_square, enemy) # Act moves = bishop.get_available_moves(board) # Assert print(moves) assert Square.at(5, 5) in moves assert Square.at(6, 6) not in moves assert Square.at(7, 7) not in moves assert Square.at(2, 2) in moves
def test_black_bishop_cannot_move_if_friendly_piece_in_the_way(): # Arrange board = Board.empty() bishop = Bishop(Player.BLACK) bishop_square = Square.at(2, 2) board.set_piece(bishop_square, bishop) obstructing_square1 = Square.at(1, 1) obstructing_square2 = Square.at(3, 3) obstructing_square3 = Square.at(1, 3) obstructing_square4 = Square.at(3, 1) obstruction = Pawn(Player.BLACK) board.set_piece(obstructing_square1, obstruction) board.set_piece(obstructing_square2, obstruction) board.set_piece(obstructing_square3, obstruction) board.set_piece(obstructing_square4, obstruction) # Act moves = bishop.get_available_moves(board) # Assert assert len(moves) == 0
def test_king_cannot_move_to_check_by_bishop(): # Arrange board = Board.empty() king = King(Player.WHITE) king_square = Square.at(2, 3) board.set_piece(king_square, king) enemy = Bishop(Player.BLACK) enemy_square = Square.at(1, 2) board.set_piece(enemy_square, enemy) # Act moves = king.get_available_moves(board) # Assert assert len(moves) == 7 assert Square.at(1, 2) in moves assert Square.at(2, 2) in moves assert Square.at(3, 2) in moves assert Square.at(3, 3) in moves assert Square.at(3, 4) not in moves assert Square.at(2, 4) in moves assert Square.at(1, 4) in moves assert Square.at(1, 3) in moves
def test_white_king_not_in_check(): # Arrange board = Board.empty() king = King(Player.WHITE) king_square = Square.at(2, 3) board.set_piece(king_square, king) enemy_pawn = Pawn(Player.BLACK) enemy_pawn_square = Square.at(3, 3) board.set_piece(enemy_pawn_square, enemy_pawn) enemy_rook = Rook(Player.BLACK) enemy_rook_square = Square.at(3, 4) board.set_piece(enemy_rook_square, enemy_rook) enemy_bishop = Bishop(Player.BLACK) enemy_bishop_square = Square.at(1, 3) board.set_piece(enemy_bishop_square, enemy_bishop) enemy_knight = Knight(Player.BLACK) enemy_knight_square = Square.at(7, 7) board.set_piece(enemy_knight_square, enemy_knight) enemy_queen = Queen(Player.BLACK) enemy_queen_square = Square.at(3, 0) board.set_piece(enemy_queen_square, enemy_queen) enemy_king = King(Player.BLACK) enemy_king_square = Square.at(6, 6) board.set_piece(enemy_king_square, enemy_king) # Act check = king.is_in_check(board) # Assert assert check is False