}

TEST_F(LiftAsFunctionCallTest, GetQuantizationMethodSucceeds) {
  // Function containing a simple `TF::XlaCallModuleOp` with a valid string
  // attribute `_quantization_method` set to `"no_quantization {}"`.
  constexpr absl::string_view kXlaCallModuleOpWithQuantizationMethodAttr =
      R"mlir(
    func.func @main(%arg0: tensor<1x1x3xf32>, %arg1: tensor<3x4xf32>) -> tensor<1x1x4xf32> {

  EXPECT_TRUE(IsOpQuantizableStableHlo(xla_call_module_op));
}

TEST_F(IsOpQuantizableStableHloTest, DenylistedXlaCallModuleOpNotQuantizable) {
  // A `TF::XlaCallModuleOp` with `_quantization_method = "no_quantization {}"`,
  // indicating it has been explicitly denylisted by the user.
  constexpr absl::string_view kModuleDenylistedXlaCallModuleOp = R"mlir(

  //   matcher {function_name {regex: "composite_conv.*"}}
  //   method {static_range_ptq {...}}}
  // }
  // specs {
  //   matcher {function_name {regex: "composite_dot_general_fn_1"}}
  //   method {no_quantization {}}
  // }
  const QuantizationConfig new_config = ExpandPresets(config);
  ASSERT_THAT(new_config.specs().specs(), SizeIs(3));

  const QuantizationSpec& first_spec = new_config.specs().specs(0);

  // Matches lifted functions by their names.
  FunctionNameMatcherSpec function_name = 1;
}

// Specifies how to quantize matched quantizable units.
message Method {
  oneof method {
    NoQuantization no_quantization = 1;
    StaticRangePtq static_range_ptq = 2;
    WeightOnlyPtq weight_only_ptq = 3;
  }
}

// A QuantizationSpec is essentially a (matcher spec, quantization method) pair,

  ASSERT_THAT(results.results(), SizeIs(1));

  // Test that the unquantized `QuantizableUnit` corresponding to
  // `composite_dot_general_fn` is captured. The `Method` contains
  // `NoQuantization`.
  const QuantizationResult& result = results.results(0);
  EXPECT_THAT(result.quantizable_unit().name(),
              StrEq("composite_dot_general_fn"));
  EXPECT_TRUE(result.method().has_no_quantization());
}