chromium/components/segmentation_platform/internal/migration/migration_test_utils.cc

// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "components/segmentation_platform/internal/migration/migration_test_utils.h"

#include "components/segmentation_platform/internal/metadata/metadata_writer.h"
#include "components/segmentation_platform/public/constants.h"
#include "components/segmentation_platform/public/proto/model_metadata.pb.h"

namespace segmentation_platform::migration_test_utils {

constexpr std::array<const char*, 5> kAdaptiveToolbarModelLabels = …;

std::unique_ptr<Config> GetTestConfigForBinaryClassifier(
    const std::string& segmentation_key,
    const std::string& segmentation_uma_name,
    proto::SegmentId segment_id) { … }

std::unique_ptr<Config> GetTestConfigForAdaptiveToolbar() { … }

proto::OutputConfig GetTestOutputConfigForBinaryClassifier(
    proto::SegmentId segment_id) { … }

proto::OutputConfig GetTestOutputConfigForAdaptiveToolbar() { … }

}  // namespace segmentation_platform::migration_test_utils