17 lines
445 B
Python
17 lines
445 B
Python
from matrixmania.compute import matmul
|
|
|
|
def test_matmul():
|
|
matrix_a = [[3, 4, -1, 4],
|
|
[-2, 2, 5, 1]
|
|
]
|
|
matrix_b = [[1, 3, -2],
|
|
[2, 5, 1],
|
|
[-1, 4, -4],
|
|
[2, 3, 6]
|
|
]
|
|
matrix_c = matmul(matrix_a, matrix_b)
|
|
print(f"Matrix A: {matrix_a} mal Matrix B: {matrix_b} = Matrix C: {matrix_c}")
|
|
|
|
assert matrix_c == [[20, 37, 26], [-1, 27, -8]]
|
|
|