sudoku.py

import math

def convert_blocks_to_lists(list):
    block_length = int(math.sqrt(len(list)))

    squares = []
    for i in range(0,len(list),block_length):
        for j in range(0,len(list),block_length):
            squares.append([item for x in range (i,i+block_length) for item in list[x][j:j+block_length]])

    return squares

def main(sudoku_list):
    length_of_sudoku_board = len(sudoku_list[0])
    check_sum = sum([i for i in range(1,length_of_sudoku_board+1)])
    blocks = convert_blocks_to_lists(sudoku_list)

    for i in range(length_of_sudoku_board):
        if sum(sudoku_list[i]) != check_sum:
            return False
        elif sum([item[i] for item in sudoku_list]) != check_sum:
            return False
        elif sum(blocks[i]) != check_sum:
            return False
    return True

if __name__ == "__main__":
    sudoku = [
        [1,2,3,4],
        [3,4,2,1],
        [4,3,1,2],
        [2,1,4,3]
    ]
    print main(sudoku)