EXPECT_THAT(*representative_dataset_file_map,
              Contains(Key("serving_default")));
  EXPECT_THAT(representative_dataset_file_map->at("serving_default")
                  .tfrecord_file_path(),
              StrEq("test_path"));
}

TEST(CreateRepresentativeDatasetFileMapTest, ConfigWithExplicitSignatureKey) {
  std::vector<RepresentativeDatasetConfig> representative_dataset_configs;

message RepresentativeDatasetFile {
  // Only TfRecord file is supported at the moment but defining this field
  // as oneof so it can be easily extended to support other types.
  oneof dataset_file {
    string tfrecord_file_path = 1;
  }
}

// Defines various options to specify and control the behavior of the quantizer.
// It consists of
// 1) Model-wise quantization configuration as a default configuration. If it is