25 lines
444 B
Python
25 lines
444 B
Python
from matrixmania import matmul
|
|
|
|
def test_matmul_1():
|
|
a = [[4, 3, 2],
|
|
[1, 2, 3]]
|
|
|
|
b = [[2],
|
|
[3],
|
|
[4]]
|
|
|
|
assert matmul(a, b) == [[25],
|
|
[20]]
|
|
|
|
def test_matmul_2():
|
|
a = [[3, 4, -1, 4],
|
|
[-2, 2, 5, 1]]
|
|
|
|
b = [[1, 3, -2],
|
|
[2, 5, 1],
|
|
[-1, 4, -4],
|
|
[2, 3, 6]]
|
|
|
|
assert matmul(a, b) == [[20, 37, 26],
|
|
[-1, 27, -8]]
|