def test_simple(self):
"""Verifies that a queen'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)
queen_white = Queen(white, [3, 4]) # d4
board.add_to_board(queen_white)
correct_move_list = [[7, 0]] # h8
correct_move_list += [6, 7], # g1
correct_move_list += [0, 1], # a7
correct_move_list += [0, 7], # a1
correct_move_list += [0, 4], # a4
correct_move_list += [7, 4], # h4
correct_move_list += [3, 0], # d8
correct_move_list += [3, 7], # d1
returned_moves = queen_white.get_legal_moves(board, True)
self.assertTrue(len(returned_moves) == 27)
for move in correct_move_list:
self.assertTrue(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)
queen_white = Queen(white, [3, 4]) # d4
board.add_to_board(queen_white)
pawn_white1 = Pawn(white, [2, 3]) # c5
board.add_to_board(pawn_white1)
incorrect_move1 = [2, 3] # c6
pawn_white2 = Pawn(white, [2, 4]) # c4
board.add_to_board(pawn_white2)
incorrect_move2 = [2, 4] # c4
returned_moves = queen_white.get_legal_moves(board, True)
self.assertTrue(len(returned_moves) == 21)
self.assertTrue(incorrect_move1 not in returned_moves)
self.assertTrue(incorrect_move2 not in returned_moves)
def test_capture(self):
"""Verifies that a queen's returned movelist correctly includes
a capture opportunity."""
board, white, black = create_board_and_players()
queen_white = Queen(white, [3, 4]) # d4
board.add_to_board(queen_white)
pawn_black = Pawn(black, [2, 3]) # c5
board.add_to_board(pawn_black)
correct_move1 = [2, 5] # c6
pawn_black = Pawn(black, [2, 4]) # c4
board.add_to_board(pawn_black)
correct_move2 = [2, 4] # c4
returned_moves = queen_white.get_legal_moves(board, True)
self.assertTrue(len(returned_moves) == 23)
self.assertTrue(correct_move1 in returned_moves)
self.assertTrue(correct_move2 in returned_moves)
