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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
// 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/embedder/default_model/ios_module_ranker.h"
#include "components/segmentation_platform/embedder/default_model/default_model_test_base.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace segmentation_platform {
class IosModuleRankerTest : public DefaultModelTestBase {
public:
IosModuleRankerTest()
: DefaultModelTestBase(std::make_unique<IosModuleRanker>()) {}
~IosModuleRankerTest() override = default;
void SetUp() override { DefaultModelTestBase::SetUp(); }
void TearDown() override { DefaultModelTestBase::TearDown(); }
};
TEST_F(IosModuleRankerTest, InitAndFetchModel) {
ExpectInitAndFetchModel();
}
TEST_F(IosModuleRankerTest, ExecuteModelWithInputForDefaultOrder) {
ExpectInitAndFetchModel();
ASSERT_TRUE(fetched_metadata_);
EXPECT_FALSE(ExecuteWithInput(/*inputs=*/{}));
std::vector<float> input(40, 0);
input[34] = -1; // mvt_freshness
input[35] = -1; // shortcuts_freshness
input[36] = -1; // safety_check_freshness
input[37] = -1; // tab_resumption_freshness
input[38] = -1; // parcel_tracking_freshness
input[39] = -1; // shop_card_freshness
ExpectClassifierResults(input, {kMostVisitedTiles, kShortcuts, kSafetyCheck,
kTabResumption, kParcelTracking, kShopCard});
}
TEST_F(IosModuleRankerTest, ExecuteModelWithInputForAllModules) {
ExpectInitAndFetchModel();
ASSERT_TRUE(fetched_metadata_);
EXPECT_FALSE(ExecuteWithInput(/*inputs=*/{}));
std::vector<float> input(40, 0);
input[6] = 3.0; // mvt_engagement
input[7] = 11.0; // mvt_impression
input[8] = 4.0; // shortcuts_engagement
input[9] = 2.0; // shortcuts_impression
input[10] = 1.0; // safety_check_engagement
input[11] = 1.0; // safety_check_impression
input[24] = 3.0; // tab_resumption_engagement
input[25] = 11.0; // tab_resumption_impression
input[28] = 3.0; // parcel_tracking_engagement
input[29] = 11.0; // parcel_tracking_impression
input[32] = 3.0; // shop_card_engagement
input[33] = 11.0; // shop_card_impression
input[34] = -1; // mvt_freshness
input[35] = -1; // shortcuts_freshness
input[36] = -1; // safety_check_freshness
input[37] = -1; // tab_resumption_freshness
input[38] = -1; // parcel_tracking_freshness
input[39] = -1; // shop_card_freshness
ExpectClassifierResults(input, {kMostVisitedTiles, kShortcuts, kTabResumption,
kSafetyCheck, kShopCard, kParcelTracking});
}
TEST_F(IosModuleRankerTest, ExecuteModelWithFreshnessInputOnly) {
ExpectInitAndFetchModel();
ASSERT_TRUE(fetched_metadata_);
EXPECT_FALSE(ExecuteWithInput(/*inputs=*/{}));
std::vector<float> input(40, 0);
input[34] = 0; // mvt_freshness
input[35] = 0; // shortcuts_freshness
input[36] = 0; // safety_check_freshness
input[37] = 0; // tab_resumption_freshness
input[38] = 0; // parcel_tracking_freshness
input[39] = 0; // shop_card_freshness
ExpectClassifierResults(input,
{kParcelTracking, kSafetyCheck, kShopCard, kShortcuts,
kMostVisitedTiles, kTabResumption});
input[34] = 1; // mvt_freshness
input[35] = 1; // shortcuts_freshness
input[36] = 2; // safety_check_freshness
input[37] = 2; // tab_resumption_freshness
input[38] = 1; // parcel_tracking_freshness
input[39] = 1; // shop_card_freshness
ExpectClassifierResults(input,
{kParcelTracking, kSafetyCheck, kShopCard, kShortcuts,
kMostVisitedTiles, kTabResumption});
}
} // namespace segmentation_platform
|
