def test_benchmark_exc(self):
self.assertRaise(
lambda: list(einsum_benchmark(shape=5, runtime="UNK")), ValueError)
self.assertRaise(
lambda: list(
einsum_benchmark(shape=5, equation="abc,cd->abD", perm=True)),
ValueError)
def test_benchmark1(self):
for rt in ['numpy', 'python', 'onnxruntime']:
with self.subTest(rt=rt):
res = list(einsum_benchmark(shape=5, runtime=rt))
self.assertEqual(len(res), 2)
def test_benchmarkn(self):
for rt in ['numpy']:
with self.subTest(rt=rt):
res = list(einsum_benchmark(shape=5, perm=True, runtime=rt))
self.assertEqual(len(res), 48)