Мини-судоку

def solve_sudoku( matrix ):
    SQ_DIM = 2
    DIM    = SQ_DIM * SQ_DIM
    res    = []
    #==========================================================================
    def res_i():
        return ( len(res) - 1 ) // DIM
    # ==========================================================================
    def res_j():
        return ( len(res) - 1 ) % DIM
    # ==========================================================================
    def get_cur_matr_val():
        return matrix[ res_i() ][ res_j() ]
    # ==========================================================================
    def get_cur_row():
        return res[ res_i()*DIM : res_i()*DIM + DIM ]
    # ==========================================================================
    def cur_matr_value_is_correct():
        return get_cur_matr_val() in ( 0, res[-1] )
    # ==========================================================================
    def get_cur_col():
        return res[ res_j() : res_j() + DIM*3 + 1 : DIM ]
    # ==========================================================================
    def res_val(i, j):
        return res[i*DIM + j]
    # ==========================================================================
    def get_cur_square():
        sq_i = res_i() // SQ_DIM * SQ_DIM
        sq_j = res_j() // SQ_DIM * SQ_DIM
        square = []
        for i in range( sq_i, sq_i + SQ_DIM ):
            for j in range( sq_j, sq_j + SQ_DIM ):
                if i > res_i() or i == res_i() and j > res_j():
                    break
                square.append( res_val(i,j) )
        return square
    # ==========================================================================
    def has_no_duplicate_values( arr ):
        for val in arr:
            if arr.count(val) > 1:
                return False
        return True
    # ==========================================================================
    def cur_row_is_correct():
        return has_no_duplicate_values( get_cur_row() )
    # ==========================================================================
    def cur_col_is_correct():
        return has_no_duplicate_values( get_cur_col() )
    # ==========================================================================
    def cur_square_is_correct():
        return has_no_duplicate_values( get_cur_square() )
    # ==========================================================================
    def is_good_res_val():
        return  cur_matr_value_is_correct () \
                and cur_row_is_correct    () \
                and cur_col_is_correct    () \
                and cur_square_is_correct ()
    # ==========================================================================
    def print_res():
        for i in range(DIM):
            for j in range(DIM):
                print( res_val(i,j), end='\t' )
            print()
    # ==========================================================================
    def good_find(k):
        res.append(1)
        while not is_good_res_val() or k < DIM*DIM and not good_find(k+1):
            if res[-1] < DIM:
                res[-1] += 1
            else:
                res.pop()
                return False
        return True
    # ==========================================================================
    if good_find(1):
        print_res()
    else:
        print( 'Задание некорректно!' )
#==============================================================================
m = [
    [0,0,0,0],
    [0,0,2,0],
    [0,1,0,0],
    [3,0,0,4]
]
solve_sudoku(m)