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
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
|
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#include <vespa/log/log.h>
LOG_SETUP("summaryfeatures_test");
#include <vespa/vespalib/testkit/testapp.h>
#include <vespa/vespalib/util/featureset.h>
using vespalib::FeatureSet;
using vespalib::Memory;
TEST_SETUP(Test);
int
Test::Main()
{
TEST_INIT("summaryfeatures_test");
{
FeatureSet sf;
EXPECT_EQUAL(sf.getNames().size(), 0u);
EXPECT_EQUAL(sf.numFeatures(), 0u);
EXPECT_EQUAL(sf.numDocs(), 0u);
EXPECT_TRUE(sf.getFeaturesByIndex(0) == 0);
EXPECT_TRUE(sf.getFeaturesByDocId(0) == 0);
std::vector<uint32_t> docs;
EXPECT_TRUE(sf.contains(docs));
docs.push_back(1);
EXPECT_TRUE(!sf.contains(docs));
}
{
FeatureSet::StringVector n;
n.push_back("f1");
n.push_back("f2");
n.push_back("f3");
FeatureSet sf(n, 5);
EXPECT_EQUAL(sf.getNames().size(), 3u);
EXPECT_EQUAL(sf.getNames()[0], "f1");
EXPECT_EQUAL(sf.getNames()[1], "f2");
EXPECT_EQUAL(sf.getNames()[2], "f3");
EXPECT_EQUAL(sf.numFeatures(), 3u);
EXPECT_EQUAL(sf.numDocs(), 0u);
EXPECT_EQUAL(sf.addDocId(10), 0u);
EXPECT_EQUAL(sf.addDocId(20), 1u);
EXPECT_EQUAL(sf.addDocId(30), 2u);
EXPECT_EQUAL(sf.addDocId(40), 3u);
EXPECT_EQUAL(sf.addDocId(50), 4u);
EXPECT_EQUAL(sf.numDocs(), 5u);
FeatureSet::Value *f;
const FeatureSet::Value *cf;
f = sf.getFeaturesByIndex(0);
ASSERT_TRUE(f != 0);
f[0].set_double(11.0);
f[1].set_double(12.0);
f[2].set_double(13.0);
f = sf.getFeaturesByIndex(1);
ASSERT_TRUE(f != 0);
f[0].set_double(21.0);
f[1].set_double(22.0);
f[2].set_double(23.0);
f = sf.getFeaturesByIndex(2);
ASSERT_TRUE(f != 0);
f[0].set_double(31.0);
f[1].set_double(32.0);
f[2].set_double(33.0);
f = sf.getFeaturesByIndex(3);
ASSERT_TRUE(f != 0);
f[0].set_double(41.0);
f[1].set_data(Memory("test", 4));
f[2].set_double(43.0);
f = sf.getFeaturesByIndex(4);
ASSERT_TRUE(f != 0);
f[0].set_double(51.0);
f[1].set_double(52.0);
f[2].set_double(53.0);
EXPECT_TRUE(sf.getFeaturesByIndex(5) == nullptr);
{
std::vector<uint32_t> docs;
EXPECT_TRUE(sf.contains(docs));
}
{
std::vector<uint32_t> docs;
docs.push_back(1);
EXPECT_TRUE(!sf.contains(docs));
}
{
std::vector<uint32_t> docs;
docs.push_back(31);
EXPECT_TRUE(!sf.contains(docs));
}
{
std::vector<uint32_t> docs;
docs.push_back(51);
EXPECT_TRUE(!sf.contains(docs));
}
{
std::vector<uint32_t> docs;
docs.push_back(20);
docs.push_back(40);
EXPECT_TRUE(sf.contains(docs));
}
{
std::vector<uint32_t> docs;
docs.push_back(10);
docs.push_back(20);
docs.push_back(30);
docs.push_back(40);
docs.push_back(50);
EXPECT_TRUE(sf.contains(docs));
}
{
cf = sf.getFeaturesByDocId(10);
ASSERT_TRUE(cf != nullptr);
EXPECT_APPROX(cf[0].as_double(), 11.0, 10e-6);
EXPECT_APPROX(cf[1].as_double(), 12.0, 10e-6);
EXPECT_APPROX(cf[2].as_double(), 13.0, 10e-6);
}
{
cf = sf.getFeaturesByDocId(20);
ASSERT_TRUE(cf != nullptr);
EXPECT_APPROX(cf[0].as_double(), 21.0, 10e-6);
EXPECT_APPROX(cf[1].as_double(), 22.0, 10e-6);
EXPECT_APPROX(cf[2].as_double(), 23.0, 10e-6);
}
{
cf = sf.getFeaturesByDocId(30);
ASSERT_TRUE(cf != nullptr);
EXPECT_APPROX(cf[0].as_double(), 31.0, 10e-6);
EXPECT_APPROX(cf[1].as_double(), 32.0, 10e-6);
EXPECT_APPROX(cf[2].as_double(), 33.0, 10e-6);
}
{
cf = sf.getFeaturesByDocId(40);
ASSERT_TRUE(cf != nullptr);
EXPECT_TRUE(cf[0].is_double());
EXPECT_TRUE(!cf[0].is_data());
EXPECT_EQUAL(cf[0].as_double(), 41.0);
EXPECT_TRUE(!cf[1].is_double());
EXPECT_TRUE(cf[1].is_data());
EXPECT_EQUAL(cf[1].as_data(), Memory("test", 4));
EXPECT_EQUAL(cf[2].as_double(), 43.0);
}
{
cf = sf.getFeaturesByDocId(50);
ASSERT_TRUE(cf != nullptr);
EXPECT_APPROX(cf[0].as_double(), 51.0, 10e-6);
EXPECT_APPROX(cf[1].as_double(), 52.0, 10e-6);
EXPECT_APPROX(cf[2].as_double(), 53.0, 10e-6);
}
EXPECT_TRUE(sf.getFeaturesByDocId(5) == nullptr);
EXPECT_TRUE(sf.getFeaturesByDocId(15) == nullptr);
EXPECT_TRUE(sf.getFeaturesByDocId(25) == nullptr);
EXPECT_TRUE(sf.getFeaturesByDocId(35) == nullptr);
EXPECT_TRUE(sf.getFeaturesByDocId(45) == nullptr);
EXPECT_TRUE(sf.getFeaturesByDocId(55) == nullptr);
}
TEST_DONE();
}
|
