def test_black_queen_cannot_move_if_friendly_piece_in_the_way(): # Arrange board = Board.empty() queen = Queen(Player.BLACK) queen_square = Square.at(2, 2) board.set_piece(queen_square, queen) obstructing_square1 = Square.at(1, 1) obstructing_square2 = Square.at(3, 3) obstructing_square3 = Square.at(1, 3) obstructing_square4 = Square.at(3, 1) obstructing_square5 = Square.at(2, 1) obstructing_square6 = Square.at(2, 3) obstructing_square7 = Square.at(1, 2) obstructing_square8 = Square.at(3, 2) 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) board.set_piece(obstructing_square5, obstruction) board.set_piece(obstructing_square6, obstruction) board.set_piece(obstructing_square7, obstruction) board.set_piece(obstructing_square8, obstruction) # Act moves = queen.get_available_moves(board) # Assert assert len(moves) == 0
def test_queen_can_move_diagonally(): # Arrange board = Board.empty() queen = Queen(Player.WHITE) square = Square.at(4, 4) board.set_piece(square, queen) # Act moves = queen.get_available_moves(board) # Assert assert Square.at(6, 6) in moves
def test_queen_can_take_opponent(): # Arrange board = Board.empty() queen = Queen(Player.WHITE) square = Square.at(4, 4) board.set_piece(square, queen) board.set_piece(Square.at(6, 4), Queen(Player.BLACK)) # Act moves = queen.get_available_moves(board) # Assert assert Square.at(6, 4) in moves
def test_queen_cannot_move_off_board(): # Arrange board = Board.empty() queen = Queen(Player.BLACK) queen_square = Square.at(4, 4) board.set_piece(queen_square, queen) # Act moves = queen.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_queens_can_move_right(): # Arrange board = Board.empty() queen = Queen(Player.WHITE) queen_square = Square.at(3, 4) board.set_piece(queen_square, queen) # Act moves = queen.get_available_moves(board) # Assert assert Square.at(3, 5) in moves assert Square.at(3, 6) in moves assert Square.at(3, 7) in moves
def test_queen_can_move_until_blocked_by_teammate(): # Arrange board = Board.empty() queen = Queen(Player.WHITE) square = Square.at(4, 4) board.set_piece(square, queen) board.set_piece(Square.at(6, 4), Queen(Player.WHITE)) # Act moves = queen.get_available_moves(board) # Assert assert Square.at(6, 4) not in moves
def test_black_queen_cannot_move_off_top_of_board(): # Arrange board = Board.empty() queen = Queen(Player.BLACK) square = Square.at(1, 7) board.set_piece(square, queen) # Act moves = queen.get_available_moves(board) # Assert assert Square.at(2, 8) not in moves assert Square.at(0, 8) not in moves assert Square.at(1, 8) not in moves
def test_queen_can_move_1_vertically(): # Arrange board = Board.empty() queen = Queen(Player.BLACK) queen_square = Square.at(4, 4) board.set_piece(queen_square, queen) # Act moves = queen.get_available_moves(board) # Assert assert Square.at(5, 4) in moves assert Square.at(6, 4) in moves assert Square.at(3, 4) in moves assert Square.at(2, 4) in moves
def test_queen_can_move_on_backslash(): # Arrange board = Board.empty() queen = Queen(Player.BLACK) queen_square = Square.at(4, 4) board.set_piece(queen_square, queen) # Act moves = queen.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_queen_can_move_horizontally(): # Arrange board = Board.empty() queen = Queen(Player.BLACK) queen_square = Square.at(4, 4) board.set_piece(queen_square, queen) # Act moves = queen.get_available_moves(board) # Assert assert Square.at(4, 5) in moves assert Square.at(4, 6) in moves assert Square.at(4, 3) in moves assert Square.at(4, 2) in moves
def test_black_queen_can_move_vertically_and_horizontally(): # Arrange board = Board.empty() queen = Queen(Player.BLACK) queen_square = Square.at(1, 3) board.set_piece(queen_square, queen) # Act moves = queen.get_available_moves(board) # Assert assert Square.at(1, 4) in moves assert Square.at(4, 3) in moves assert Square.at(0, 3) in moves assert Square.at(1, 1) in moves
def test_black_queen_can_move_diagonally(): # Arrange board = Board.empty() queen = Queen(Player.BLACK) queen_square = Square.at(3, 3) board.set_piece(queen_square, queen) # Act moves = queen.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_queens_can_capture_right(): # Arrange board = Board.empty() queen = Queen(Player.WHITE) queen_square = Square.at(4, 4) board.set_piece(queen_square, queen) enemy_1 = Pawn(Player.BLACK) enemy_1_square = Square.at(4, 6) board.set_piece(enemy_1_square, enemy_1) # Act moves = queen.get_available_moves(board) # Assert assert Square.at(4, 6) in moves
def test_queens_cannot_move_left_through_pieces(): # Arrange board = Board.empty() queen = Queen(Player.WHITE) queen_square = Square.at(4, 4) board.set_piece(queen_square, queen) enemy_1 = Pawn(Player.BLACK) enemy_1_square = Square.at(4, 3) board.set_piece(enemy_1_square, enemy_1) # Act moves = queen.get_available_moves(board) # Assert assert Square.at(4, 2) not in moves
def test_queen_cannot_move_if_piece_in_front(): # Arrange board = Board.empty() queen = Queen(Player.WHITE) queen_square = Square.at(4, 4) board.set_piece(queen_square, queen) enemy = Pawn(Player.BLACK) enemy_square = Square.at(5, 5) board.set_piece(enemy_square, enemy) # Act moves = queen.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_white_queens_can_move_to_all_possible_squares(): # Arrange board = Board.empty() queen = Queen(Player.WHITE) square = Square.at(1, 4) board.set_piece(square, queen) # Act moves = queen.get_available_moves(board) # Assert assert set(moves) == set([ Square.at(0, 3), Square.at(0, 4), Square.at(0, 5), Square.at(1, 0), Square.at(1, 1), Square.at(1, 2), Square.at(1, 3), Square.at(1, 5), Square.at(1, 6), Square.at(1, 7), Square.at(2, 3), Square.at(2, 4), Square.at(2, 5), Square.at(3, 2), Square.at(3, 4), Square.at(3, 6), Square.at(4, 1), Square.at(4, 4), Square.at(4, 7), Square.at(5, 0), Square.at(5, 4), Square.at(6, 4), Square.at(7, 4) ])
def test_white_queens_can_move_to_all_unoccupied_squares(): # Arrange board = Board.empty() queen = Queen(Player.WHITE) square = Square.at(1, 4) board.set_piece(square, queen) team1 = Pawn(Player.WHITE) team1_square = Square.at(2, 5) board.set_piece(team1_square, team1) enemy1 = Pawn(Player.BLACK) enemy1_square = Square.at(3, 4) board.set_piece(enemy1_square, enemy1) # Act moves = queen.get_available_moves(board) # Assert assert set(moves) == set([ Square.at(0, 3), Square.at(0, 4), Square.at(0, 5), Square.at(1, 0), Square.at(1, 1), Square.at(1, 2), Square.at(1, 3), Square.at(1, 5), Square.at(1, 6), Square.at(1, 7), Square.at(2, 3), Square.at(2, 4), Square.at(3, 2), Square.at(3, 4), Square.at(4, 1), Square.at(5, 0) ])
def test_queens_cannot_move_off_the_board(): # Arrange board = Board.empty() queen = Queen(Player.WHITE) queen_square = Square.at(0, 0) board.set_piece(queen_square, queen) friendly_1 = Pawn(Player.WHITE) friendly_1_square = Square.at(1, 0) board.set_piece(friendly_1_square, friendly_1) friendly_2 = Pawn(Player.WHITE) friendly_2_square = Square.at(0, 1) board.set_piece(friendly_2_square, friendly_2) friendly_3 = Pawn(Player.WHITE) friendly_3_square = Square.at(1, 1) board.set_piece(friendly_3_square, friendly_3) # Act moves = queen.get_available_moves(board) # Assert assert len(moves) == 0