Lines Matching refs:sampler
48 KllSampler sampler(&compactor_stack); in TEST_F() local
49 EXPECT_THAT(sampler.sampled_item_and_weight(), Eq(std::nullopt)); in TEST_F()
51 sampler.Add(4); in TEST_F()
52 EXPECT_THAT(sampler.sampled_item_and_weight(), Optional(Pair(Eq(4), Eq(1)))); in TEST_F()
53 sampler.Add(10); in TEST_F()
54 EXPECT_THAT(sampler.sampled_item_and_weight(), Eq(std::nullopt)); in TEST_F()
57 sampler.Reset(); in TEST_F()
58 sampler.DoubleCapacity(); in TEST_F()
59 sampler.Add(4); in TEST_F()
60 EXPECT_THAT(sampler.sampled_item_and_weight(), Optional(Pair(Eq(4), Eq(1)))); in TEST_F()
61 sampler.Add(10); in TEST_F()
62 EXPECT_THAT(sampler.sampled_item_and_weight(), Optional(Pair(AnyOf(Eq(4), Eq(10)), Eq(2)))); in TEST_F()
63 sampler.Add(14); in TEST_F()
64 EXPECT_THAT(sampler.sampled_item_and_weight(), Optional(Pair(_, Eq(3)))); in TEST_F()
65 sampler.Add(24); in TEST_F()
66 EXPECT_THAT(sampler.sampled_item_and_weight(), Eq(std::nullopt)); in TEST_F()
72 KllSampler sampler(&compactor_stack); in TEST_F() local
73 EXPECT_THAT(sampler.sampled_item_and_weight(), Eq(std::nullopt)); in TEST_F()
78 KllSampler sampler(&compactor_stack); in TEST_F() local
79 sampler.Add(4); in TEST_F()
80 EXPECT_THAT(sampler.sampled_item_and_weight(), Optional(Pair(Eq(4), Eq(1)))); in TEST_F()
85 KllSampler sampler(&compactor_stack); in TEST_F() local
86 EXPECT_EQ(sampler.capacity(), 2); in TEST_F()
87 sampler.Add(1); in TEST_F()
88 sampler.Add(2); in TEST_F()
91 compactor_stack.compactors()[sampler.num_replaced_levels()]; in TEST_F()
95 EXPECT_THAT(sampler.sampled_item_and_weight(), Eq(std::nullopt)); in TEST_F()
100 KllSampler sampler(&compactor_stack); in TEST_F() local
101 sampler.DoubleCapacity(); in TEST_F()
102 EXPECT_EQ(sampler.capacity(), 4); in TEST_F()
103 EXPECT_EQ(sampler.num_replaced_levels(), 2); in TEST_F()
105 sampler.Add(1); in TEST_F()
106 sampler.Add(2); in TEST_F()
109 EXPECT_THAT(sampler.sampled_item_and_weight(), Optional(Pair(AnyOf(Eq(1), Eq(2)), Eq(2)))); in TEST_F()
114 KllSampler sampler(&compactor_stack); in TEST_F() local
115 EXPECT_EQ(sampler.capacity(), 2); in TEST_F()
116 sampler.Add(1); in TEST_F()
117 sampler.Add(2); in TEST_F()
118 sampler.Add(3); in TEST_F()
119 sampler.Add(4); in TEST_F()
122 compactor_stack.compactors()[sampler.num_replaced_levels()]; in TEST_F()
127 EXPECT_THAT(sampler.sampled_item_and_weight(), Eq(std::nullopt)); in TEST_F()
132 KllSampler sampler(&compactor_stack); in TEST_F() local
133 EXPECT_EQ(sampler.capacity(), 2); in TEST_F()
135 sampler.Add(1); in TEST_F()
136 sampler.Add(2); in TEST_F()
137 sampler.Add(3); in TEST_F()
140 compactor_stack.compactors()[sampler.num_replaced_levels()]; in TEST_F()
143 EXPECT_THAT(sampler.sampled_item_and_weight(), Optional(Pair(Eq(3), Eq(1)))); in TEST_F()
148 KllSampler sampler(&compactor_stack); in TEST_F() local
149 sampler.AddWithWeight(5, 0); in TEST_F()
150 EXPECT_THAT(sampler.sampled_item_and_weight(), Eq(std::nullopt)); in TEST_F()
155 KllSampler sampler(&compactor_stack); in TEST_F() local
156 sampler.AddWithWeight(5, 1); in TEST_F()
157 EXPECT_THAT(sampler.sampled_item_and_weight(), Optional(Pair(Eq(5), Eq(1)))); in TEST_F()
162 KllSampler sampler(&compactor_stack); in TEST_F() local
163 sampler.AddWithWeight(1, 2); in TEST_F()
166 compactor_stack.compactors()[sampler.num_replaced_levels()]; in TEST_F()
169 EXPECT_THAT(sampler.sampled_item_and_weight(), Eq(std::nullopt)); in TEST_F()
174 KllSampler sampler(&compactor_stack); in TEST_F() local
175 sampler.AddWithWeight(3, 3); in TEST_F()
178 compactor_stack.compactors()[sampler.num_replaced_levels()]; in TEST_F()
181 EXPECT_THAT(sampler.sampled_item_and_weight(), Optional(Pair(Eq(3), Eq(1)))); in TEST_F()
186 KllSampler sampler(&compactor_stack); in TEST_F() local
187 sampler.Add(1); in TEST_F()
188 sampler.DoubleCapacity(); in TEST_F()
189 sampler.DoubleCapacity(); in TEST_F()
190 sampler.AddWithWeight(2, 3); in TEST_F()
192 EXPECT_THAT(sampler.sampled_item_and_weight(), Optional(Pair(AnyOf(Eq(1), Eq(2)), Eq(4)))); in TEST_F()
197 KllSampler sampler(&compactor_stack); in TEST_F() local
198 sampler.Add(1); in TEST_F()
199 sampler.DoubleCapacity(); in TEST_F()
200 sampler.Add(2); in TEST_F()
201 sampler.AddWithWeight(3, 5); in TEST_F()
204 compactor_stack.compactors()[sampler.num_replaced_levels()]; in TEST_F()
207 EXPECT_THAT(sampler.sampled_item_and_weight(), Optional(Pair(Eq(3), Eq(3)))); in TEST_F()
212 KllSampler sampler(&compactor_stack); in TEST_F() local
213 sampler.Add(1); in TEST_F()
214 sampler.DoubleCapacity(); in TEST_F()
215 EXPECT_EQ(sampler.capacity(), 4); in TEST_F()
216 EXPECT_EQ(sampler.num_replaced_levels(), 2); in TEST_F()
217 sampler.Add(2); in TEST_F()
218 sampler.Add(3); in TEST_F()
219 EXPECT_THAT(sampler.sampled_item_and_weight(), in TEST_F()
221 sampler.DoubleCapacity(); in TEST_F()
222 EXPECT_EQ(sampler.capacity(), 8); in TEST_F()
223 EXPECT_EQ(sampler.num_replaced_levels(), 3); in TEST_F()
224 sampler.Add(4); in TEST_F()
225 sampler.Add(5); in TEST_F()
226 EXPECT_THAT(sampler.sampled_item_and_weight(), in TEST_F()
232 KllSampler sampler(&compactor_stack); in TEST_F() local
233 sampler.Reset(); in TEST_F()
234 EXPECT_THAT(sampler.sampled_item_and_weight(), Eq(std::nullopt)); in TEST_F()
239 KllSampler sampler(&compactor_stack); in TEST_F() local
240 sampler.Add(1); in TEST_F()
241 sampler.Reset(); in TEST_F()
242 EXPECT_THAT(sampler.sampled_item_and_weight(), Eq(std::nullopt)); in TEST_F()
243 sampler.Add(2); in TEST_F()
244 EXPECT_THAT(sampler.sampled_item_and_weight(), Optional(Pair(Eq(2), Eq(1)))); in TEST_F()
249 KllSampler sampler(&compactor_stack); in TEST_F() local
250 sampler.DoubleCapacity(); in TEST_F()
251 EXPECT_EQ(sampler.capacity(), 4); in TEST_F()
252 EXPECT_EQ(sampler.num_replaced_levels(), 2); in TEST_F()
255 sampler.Add(i); in TEST_F()
257 EXPECT_THAT(sampler.sampled_item_and_weight(), Optional(Pair(_, Eq(2)))); in TEST_F()
259 sampler.Reset(); in TEST_F()
260 EXPECT_THAT(sampler.sampled_item_and_weight(), Eq(std::nullopt)); in TEST_F()
261 EXPECT_EQ(sampler.capacity(), 2); in TEST_F()
262 EXPECT_EQ(sampler.num_replaced_levels(), 1); in TEST_F()
263 sampler.DoubleCapacity(); in TEST_F()
264 EXPECT_EQ(sampler.capacity(), 4); in TEST_F()
265 EXPECT_EQ(sampler.num_replaced_levels(), 2); in TEST_F()
267 sampler.Add(i); in TEST_F()
270 EXPECT_THAT(sampler.sampled_item_and_weight(), Optional(Pair(AnyOf(Eq(8), Eq(9)), Eq(2)))); in TEST_F()
296 KllSampler sampler(&compactor_stack); in TEST_P() local
297 while (sampler.capacity() <= num_items) { in TEST_P()
298 sampler.DoubleCapacity(); in TEST_P()
301 sampler.Add(j); in TEST_P()
303 seen_items[sampler.sampled_item_and_weight().value().first] = true; in TEST_P()
304 EXPECT_EQ(sampler.sampled_item_and_weight().value().second, num_items); in TEST_P()