def test_check_small(): board = Board() king = King(BLACK, 0, 0) rook = Rook(WHITE, 2, 2) board._pieces = {king, rook} assert not board.checked(BLACK) board.move(rook, 2, 0) assert board.checked(BLACK)
def test_check_medium(): board = Board() king = King(BLACK, 0, 0) rook = Rook(WHITE, 2, 2) pawn = Pawn(WHITE, 7, 3) bishop = Bishop(WHITE, 1, 3) enemy_king = King(WHITE, 6, 6) board._pieces = {king, rook, pawn, bishop, enemy_king} assert not board.checked(BLACK) board.move(rook, 2, 0) assert board.checked(BLACK) assert not board.checked(WHITE)