def test_castling_to_check(self):
"""Ensures that castling is not returned in the King's moved
list if the King would be castling into check."""
board, white, black = Board(), Player(Color.W), Player(Color.B)
king_white = King(white, [4, 7]) # e1
board.add_to_board(king_white)
rook_white = Rook(white, [7, 7]) # h1
board.add_to_board(rook_white)
castle_move = [6, 7] # g1
returned_moves = king_white.get_legal_moves(board, True)
self.assertTrue(castle_move in returned_moves)
rook_black = Rook(black, [6, 0]) # g7
board.add_to_board(rook_black)
returned_moves = king_white.get_legal_moves(board, True)
self.assertFalse(castle_move in returned_moves)
def test_castling_normal(self):
"""Ensures that castling is properly returned in the King's moved
list."""
board, white = Board(), Player(Color.W)
king_white = King(white, [4, 7]) # e1
board.add_to_board(king_white)
rook_white = Rook(white, [7, 7]) # h1
board.add_to_board(rook_white)
castle_move = [6, 7] # g1
returned_moves = king_white.get_legal_moves(board, True)
self.assertTrue(castle_move in returned_moves)
def test_castling_already_moved(self):
"""Ensures that castling is not returned in the King's moved
list if the King has already moved."""
board, white = Board(), Player(Color.W)
king_white = King(white, [4, 7]) # e1
king_white.first_move = False
board.add_to_board(king_white)
rook_white = Rook(white, [7, 7]) # h1
board.add_to_board(rook_white)
castle_move = [6, 7] # g1
returned_moves = king_white.get_legal_moves(board, True)
self.assertFalse(castle_move in returned_moves)
def test_path_impeded(self):
"""Verifies that a bishop's returned moves stops when encountering
another of the bishop's owner's pieces."""
board, white = Board(), Player(Color.W)
king_white = King(white, [3, 4]) # d4
board.add_to_board(king_white)
pawn_white = Pawn(white, [2, 4]) # c4
board.add_to_board(pawn_white)
incorrect_move1 = [2, 4] # c4
returned_moves = king_white.get_legal_moves(board, True)
self.assertTrue(len(returned_moves) == 7)
self.assertFalse(incorrect_move1 in returned_moves)
def test_simple(self):
"""Verifies that a king's returned movelist has the correct
number of moves and that the boundaries of it's movelist is correct."""
board, white = Board(), Player(Color.W)
king_white = King(white, [3, 4]) # d4
board.add_to_board(king_white)
correct_move_list = [[3, 5]] # d3
correct_move_list += [2, 5], # c3
correct_move_list += [4, 5], # e3
correct_move_list += [4, 4], # e4
correct_move_list += [2, 4], # c4
correct_move_list += [4, 3], # e5
correct_move_list += [3, 3], # d5
correct_move_list += [2, 3], # c5
returned_moves = king_white.get_legal_moves(board, True)
self.assertTrue(len(returned_moves) == 8)
for move in correct_move_list:
self.assertTrue(move in returned_moves)
def test_capture(self):
"""Verifies that a king's returned movelist correctly includes
a capture opportunity."""
board, white, black = create_board_and_players()
king_white = King(white, [3, 4]) # d4
board.add_to_board(king_white)
pawn_black = Pawn(black, [2, 3]) # c5
board.add_to_board(pawn_black)
correct_move_list = [[3, 5]] # d3
correct_move_list += [2, 5], # c3
correct_move_list += [4, 5], # e3
correct_move_list += [4, 4], # e4
correct_move_list += [2, 4], # c4
correct_move_list += [4, 3], # e5
correct_move_list += [3, 3], # d5
correct_move_list += [2, 3], # c5
returned_moves = king_white.get_legal_moves(board, True)
self.assertTrue(len(returned_moves) == 8)
for move in correct_move_list:
self.assertTrue(move in returned_moves)
