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