def find_eigenvalue_test()

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

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

def find_eigenvalue_test(fun):
    for i in range(10):
        (a, expected, v) = (None, None, None)
        if i < 3:
            (a, expected) = (edge_matrices[i], edge_values[i])
            v = edge_vectors[i]
        else:
            (a, expected) = gen_eigenmatrix()
            v = find_eigenvector_ref(a, expected)
        actual = fun(a, v)
        if actual == None or actual == ...:
            print("Your function must return a value!")
            return
        if actual != approx(expected):
            print("Wrong eigenvalue!\n"
                  + gen_labeled_message([a, v], ["A: ", "V: "])
                  + "Expected "
                  + "{0:.3f}\n\n".format(expected)
                  + "You returned: {0:.3f}\n\n".format(actual)
                  + "Try again!")
            return
    print("Success!")