def test_matrix_mul():
"""Test Matrix addition"""
a = Matrix.identity(3)
b = Matrix.identity(3)
c = a * b
assert c.get_data() == [[1, 0, 0], [0, 1, 0], [0, 0, 1]]
def test_matrix_add():
"""Test Matrix addition"""
a = Matrix.identity(3)
b = Matrix.identity(3)
c = a + b
assert c.get_data() == [[2, 0, 0], [0, 2, 0], [0, 0, 2]]
def test_matrix_sub():
"""Test Matrix addition"""
a = Matrix.identity(3)
b = Matrix.identity(3)
c = a - b
assert c.get_data() == [[0, 0, 0], [0, 0, 0], [0, 0, 0]]
def test_matrix_identity():
"""Test identity matrix constructor"""
a = Matrix.identity(3)
assert a.get_data() == [[1, 0, 0], [0, 1, 0], [0, 0, 1]]