def test_matrix_identities():
A = Matrix.identity(3, 3)
A[0, 1] = 1
assert (A + Matrix.zero(3, 3) == A)
assert (Matrix.zero(3, 3) + A == A)
assert (A * Matrix.identity(3, 3) == A)
assert (Matrix.identity(3, 3) * A == A)
def test_matrix_equality():
A = Matrix(3, 3)
B = Matrix(3, 3)
assert (A == B)
A = Matrix.identity(3, 3)
B = Matrix.identity(3, 3)
assert (A == B)
A = Matrix.zero(3, 3)
B = Matrix.zero(3, 3)
assert (A == B)
def test_approx_jacobian_identity():
def f(x):
return x
x = Vector(0, 0, 0)
assert(approx(approx_jacobian(f, x), Matrix.identity(3)))