class TestValidateKnightMoves(unittest.TestCase):
def setUp(self):
self.chess_board = ChessBoard()
def test_move_knight(self):
result = self.chess_board.move((0, 1), (2, 0))
assert result is True
assert self.chess_board.board[(2, 0)].kind == 'knight'
def test_move_knight_on_top_of_a_pawn(self):
result = self.chess_board.move((1, 0), (3, 1))
assert result is False
assert self.chess_board.board[(1, 3)].kind == 'pawn'
def setUp(self):
self.board = ChessBoard()
def __init__(self, renderer: IRenderer = None):
self._finished = False
self._board = ChessBoard()
self._renderer = renderer
self._status = ChessGame.STATUS_WHITE_MOVE
def setUp(self):
reset_option(ALL)
self.board = ChessBoard()
class TestBoard(unittest.TestCase):
"""Chess unit test class."""
def setUp(self):
self.chess_board = ChessBoard()
@staticmethod
def convert_default_white_spaces_to_black(white_positions):
return [(7 - row, column) for row, column in white_positions]
def verify_pieces_at_locations_are_correct_piece_and_color(self, white_positions, piece):
for location in white_positions:
assert self.chess_board.board[location].kind == piece
assert self.chess_board.board[location].color == "white"
black_positions = self.convert_default_white_spaces_to_black(white_positions)
for location in black_positions:
assert self.chess_board.board[location].kind == piece
assert self.chess_board.board[location].color == "black"
def test_init(self):
assert len(self.chess_board.board) == 64
def test_has_pawn_at_1_3(self):
piece = self.chess_board.board[(1, 3)]
assert len(piece.moves) > 0
def test_pawn_can_move_forward(self):
assert self.chess_board.board[(2, 3)] is None
ends = self.chess_board.end_locations_for_piece_at_location((1, 3))
assert ends == [(2, 3)]
def test_initial_pawn_positions(self):
expected_white_pawn_positions = [(1, 0), (1, 1), (1, 2), (1, 3), (1, 4), (1, 5), (1, 6), (1, 7)]
self.verify_pieces_at_locations_are_correct_piece_and_color(expected_white_pawn_positions, "pawn")
def test_initial_knight_positions(self):
self.verify_pieces_at_locations_are_correct_piece_and_color([(0, 1), (0, 6)], "knight")
def test_initial_rook_positions(self):
self.verify_pieces_at_locations_are_correct_piece_and_color([(0, 0), (0, 7)], "rook")
def test_initial_bishop_positions(self):
self.verify_pieces_at_locations_are_correct_piece_and_color([(0, 2), (0, 5)], 'bishop')
def test_initial_queen_positions(self):
self.verify_pieces_at_locations_are_correct_piece_and_color([(0, 3)], "queen")
def test_initial_king_positions(self):
self.verify_pieces_at_locations_are_correct_piece_and_color([(0, 4)], "king")
def test_starting_board_fen_export(self):
assert self.chess_board.generate_fen() == starting_fen_board
def test_one_move_fen_export(self):
self.chess_board.board[(6, 0)] = None
expected_fen = "rnbqkbnr/1ppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR w KQkq - 0 1"
assert expected_fen == self.chess_board.generate_fen()
def test_clear_board_removes_all_pieces(self):
self.chess_board.clear_board()
for location in self.chess_board.board:
assert self.chess_board.board[location] is None
def test_starting_board_fen_import(self):
self.chess_board.clear_board()
self.chess_board.import_fen_board(starting_fen_board)
expected_white_pawn_positions = [(1, 0), (1, 1), (1, 2), (1, 3), (1, 4), (1, 5), (1, 6), (1, 7)]
self.verify_pieces_at_locations_are_correct_piece_and_color(expected_white_pawn_positions, "pawn")
self.verify_pieces_at_locations_are_correct_piece_and_color([(0, 1), (0, 6)], "knight")
self.verify_pieces_at_locations_are_correct_piece_and_color([(0, 0), (0, 7)], "rook")
self.verify_pieces_at_locations_are_correct_piece_and_color([(0, 2), (0, 5)], 'bishop')
self.verify_pieces_at_locations_are_correct_piece_and_color([(0, 3)], "queen")
self.verify_pieces_at_locations_are_correct_piece_and_color([(0, 4)], "king")
def test_starting_board_custom_export(self):
expected_json = self.chess_board.load_json()
exported_json = self.chess_board.export()
self.compare_boards(exported_json['board'], expected_json['board'])
exported_json.pop('board')
for key, value in exported_json.items():
assert expected_json[key] == value
assert len(expected_json) == len(exported_json) + 1
@staticmethod
def compare_boards(board1, board2):
for player in board1:
assert player in board2
for piece in board1[player]:
assert piece in board2[player]
piece_locations1 = board1[player][piece]
piece_locations2 = board2[player][piece]
assert len(piece_locations1) == len(piece_locations2)
for location in piece_locations1:
assert location in piece_locations2
def setUp(self):
self.chess_board = ChessBoard()
def setUp(self):
self.chess_board = ChessBoard()
