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]]