# Copyright (c) 2021 Graphcore Ltd. All rights reserved.
import matmul_test_broadcasting_base as mtb

# generated test cases
# these are all known to be valid input shapes to np.matmul
shapes_ = (
    ([1], [2, 4, 1, 3]),
    ([4, 3, 1], [1]),
    ([3, 4], [1, 4, 3]),
    ([2, 4], [2, 4, 3]),
    ([2], [4, 1, 2, 3]),
    ([2, 1, 4, 3], [3]),
    ([2, 1, 3], [2, 3, 1]),
    ([3, 1, 4, 2], [4, 2, 1]),
    ([3, 4], [1, 4, 2]),
    ([4], [3, 4, 1]),
    ([3], [1, 4, 3, 2]),
    ([1, 3], [1, 3, 2]),
    ([1, 3], [2, 4, 3, 1]),
    ([4, 3, 1], [1, 2]),
    ([3, 2, 4], [4]),
    ([4, 1, 3], [3]),
    ([3], [2, 1, 3, 4]),
    ([3, 1], [1, 2]),
    ([4, 1], [1, 2]),
)


def test_matmul_broadcasting_2(op_tester):
    mtb._test_matmul_broadcasting_base(op_tester, shapes_)