summaryrefslogtreecommitdiff
path: root/chromium/components/assist_ranker/binary_classifier_predictor_unittest.cc
diff options
context:
space:
mode:
Diffstat (limited to 'chromium/components/assist_ranker/binary_classifier_predictor_unittest.cc')
-rw-r--r--chromium/components/assist_ranker/binary_classifier_predictor_unittest.cc9
1 files changed, 9 insertions, 0 deletions
diff --git a/chromium/components/assist_ranker/binary_classifier_predictor_unittest.cc b/chromium/components/assist_ranker/binary_classifier_predictor_unittest.cc
index dcde981bf87..672c80fadf0 100644
--- a/chromium/components/assist_ranker/binary_classifier_predictor_unittest.cc
+++ b/chromium/components/assist_ranker/binary_classifier_predictor_unittest.cc
@@ -9,6 +9,7 @@
#include "base/bind.h"
#include "base/bind_helpers.h"
#include "base/feature_list.h"
+#include "base/test/scoped_feature_list.h"
#include "components/assist_ranker/fake_ranker_model_loader.h"
#include "components/assist_ranker/proto/ranker_model.pb.h"
#include "components/assist_ranker/ranker_model.h"
@@ -20,6 +21,8 @@ using ::assist_ranker::testing::FakeRankerModelLoader;
class BinaryClassifierPredictorTest : public ::testing::Test {
public:
+ void SetUp() override;
+
std::unique_ptr<BinaryClassifierPredictor> InitPredictor(
std::unique_ptr<RankerModel> ranker_model,
const PredictorConfig& config);
@@ -33,8 +36,14 @@ class BinaryClassifierPredictorTest : public ::testing::Test {
const std::string feature_ = "feature";
const float weight_ = 1.0;
const float threshold_ = 0.5;
+ base::test::ScopedFeatureList scoped_feature_list_;
};
+void BinaryClassifierPredictorTest::SetUp() {
+ ::testing::Test::SetUp();
+ scoped_feature_list_.Init();
+}
+
std::unique_ptr<BinaryClassifierPredictor>
BinaryClassifierPredictorTest::InitPredictor(
std::unique_ptr<RankerModel> ranker_model,