1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
// Copyright 2022 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/embedder/default_model/search_user_model.h"
#include "components/segmentation_platform/embedder/default_model/default_model_test_base.h"
namespace segmentation_platform {
class SearchUserModelTest : public DefaultModelTestBase {
public:
SearchUserModelTest()
: DefaultModelTestBase(std::make_unique<SearchUserModel>()) {}
~SearchUserModelTest() override = default;
};
TEST_F(SearchUserModelTest, InitAndFetchModel) {
ExpectInitAndFetchModel();
ASSERT_TRUE(fetched_metadata_);
}
// Segmentation Ukm Engine is disabled on CrOS.
#if !BUILDFLAG(IS_CHROMEOS)
TEST_F(SearchUserModelTest, VerifyMetadata) {
ExpectInitAndFetchModel();
ASSERT_TRUE(fetched_metadata_);
ASSERT_EQ(2, fetched_metadata_.value().input_features_size());
const proto::UMAFeature feature =
fetched_metadata_.value().input_features(0).uma_feature();
EXPECT_EQ(proto::SignalType::HISTOGRAM_ENUM, feature.type());
EXPECT_EQ("Omnibox.SuggestionUsed.ClientSummarizedResultType",
feature.name());
EXPECT_EQ(proto::Aggregation::COUNT, feature.aggregation());
EXPECT_EQ(1u, feature.tensor_length());
ASSERT_EQ(1, feature.enum_ids_size());
// This must match the `Search` entry in `ClientSummaryResultGroup` in
// //tools/metrics/histograms/enums.xml.
EXPECT_EQ(1, feature.enum_ids(0));
const proto::SqlFeature sql_feature =
fetched_metadata_.value().input_features(1).sql_feature();
EXPECT_NE(sql_feature.sql(), "");
}
#endif //! BUILDFLAG(IS_CHROMEOS)
TEST_F(SearchUserModelTest, ExecuteModelWithInput) {
ExpectInitAndFetchModel();
ASSERT_TRUE(fetched_metadata_);
EXPECT_FALSE(ExecuteWithInput(/*inputs=*/{}));
ModelProvider::Request input = {0};
ExpectClassifierResults(/*input=*/{0}, {kSearchUserModelLabelNone});
ExpectClassifierResults(/*input=*/{1}, {kSearchUserModelLabelLow});
ExpectClassifierResults(/*input=*/{5}, {kSearchUserModelLabelMedium});
ExpectClassifierResults(/*input=*/{22}, {kSearchUserModelLabelHigh});
}
} // namespace segmentation_platform
|
