}

struct MatMulTestCase {
  std::string mat_mul_method;
};

using BatchMatMulTest = TestWithParam<MatMulTestCase>;

TEST_P(BatchMatMulTest, BatchMatMul) {
  const MatMulTestCase& test_case = GetParam();
  static constexpr char kMatMulModuleStr[] = R"(
  module attributes {tf.versions = {bad_consumers = [], min_consumer = 0 : i32, producer = 268 : i32}} {
    func.func @main() -> (tensor<1x4x4xf32>) {

// or "pattern", whichever you prefer to think.

TEST_P(FooTest, DoesBlah) {
  // Inside a test, access the test parameter with the GetParam() method
  // of the TestWithParam<T> class:
  EXPECT_TRUE(foo.Blah(GetParam()));
  ...
}

TEST_P(FooTest, HasBlahBlah) {
  ...
}

// Finally, you can use INSTANTIATE_TEST_CASE_P to instantiate the test