def is_matrix_unitary_ref()

in tutorials/LinearAlgebra/testing.py [0:0]

• 10 lines of code
• 6 McCabe index (conditional complexity)

def is_matrix_unitary_ref(a):
    n = len(a)
    prod = matrix_mult_ref(a, adjoint_ref(a))
    for i in range(n):
        for j in range(n):
            if i == j:
                if prod[i][j] != approx(1): return False
            else:
                if prod[i][j] != approx(0): return False
    return True