puzzle.py

"""
https://github.com/PHentosh/Puzzle
"""

def check_rows(board: list)-> bool:
    """
    Check if in rows are the same numbers
    >>> check_rows(["**** ****", "***1 ****", "**  3****", "* 4 1****", \
    "     9 5 ", " 6  83  *", "3   1  **", "  8  2***", "  2  ****"])
    True
    """
    for i in board:
        numbers = []
        for j in i:
            if (j != "*") and (j != ' '):
                numbers.append(int(j))
        for j in numbers:
            numbers = numbers[1:]
            if j in numbers:
                return False
    return True

def check_columns(board: list)-> bool:
    """
    Check if in colnmns are the same numbers
    >>> check_columns(["**** ****", "***1 ****", "**  3****", "* 4 1****", \
    "     9 5 ", " 6  83  *", "3   1  **", "  8  2***", "  2  ****"])
    False
    """
    new_board = []
    for i in range(len(board)):
        new_board.append('')
    for i in board:
        for j in range(len(i)):
            new_board[j] = i[j] + new_board[j]
    return check_rows(new_board)

def check_colored_cells(board: list)-> bool:
    """
    Check if in colored cells are the same numbers
    >>> check_colored_cells(["**** ****", "***1 ****", "**  3****", "* 4 1****", \
    "     9 5 ", " 6  83  *", "3   1  **", "  8  2***", "  2  ****"])
    True
    """
    color = ['', '', '', '', '']
    for i in range(5):
        for col in range(5):
            color[i] = color[i] + board[8-i-col][i]
        for row in range(5):
            color[i] = color[i] + board[8-i][row+i]
        color[i] = color[i][1:]
    return check_rows(color)

def validate_board(board: list)-> bool:
    """
    Check if board is ready for game
    """
    return check_colored_cells(board) and \
        check_columns(board) and check_rows(board)