chromium/components/assist_ranker/classifier_predictor_unittest.cc

// Copyright 2018 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/assist_ranker/classifier_predictor.h"

#include <memory>
#include <string>
#include <utility>
#include <vector>

#include "base/feature_list.h"
#include "base/functional/bind.h"
#include "base/functional/callback_helpers.h"
#include "base/metrics/field_trial_params.h"
#include "base/test/scoped_feature_list.h"
#include "components/assist_ranker/example_preprocessing.h"
#include "components/assist_ranker/fake_ranker_model_loader.h"
#include "components/assist_ranker/nn_classifier_test_util.h"
#include "components/assist_ranker/proto/ranker_model.pb.h"
#include "components/assist_ranker/ranker_model.h"
#include "testing/gmock/include/gmock/gmock-matchers.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace assist_ranker {

FakeRankerModelLoader;
FloatEq;

// Preprocessor feature names.
const char kFeatureName0[] = …;
const char kFeatureName1[] = …;
const char kFeatureExtra[] = …;

class ClassifierPredictorTest : public ::testing::Test { … };

void ClassifierPredictorTest::SetUp() { … }

std::unique_ptr<ClassifierPredictor> ClassifierPredictorTest::InitPredictor(
    std::unique_ptr<RankerModel> ranker_model,
    const PredictorConfig& config) { … }

BASE_FEATURE(…);

const base::FeatureParam<std::string> kTestRankerUrl{ … };

PredictorConfig ClassifierPredictorTest::GetConfig() { … }

TEST_F(ClassifierPredictorTest, EmptyRankerModel) { … }

TEST_F(ClassifierPredictorTest, NoInferenceModuleForModel) { … }

TEST_F(ClassifierPredictorTest, PredictFeatureVector) { … }

TEST_F(ClassifierPredictorTest, PredictRankerExampleNoPreprocessor) { … }

TEST_F(ClassifierPredictorTest, PredictRankerExampleWithPreprocessor) { … }

TEST_F(ClassifierPredictorTest, PredictRankerExamplePreprocessorError) { … }
}  // namespace assist_ranker