Lines Matching refs:keyset
45 marisa::Keyset keyset; in TestEmptyTrie() local
46 trie.build(keyset); in TestEmptyTrie()
65 keyset.push_back(""); in TestEmptyTrie()
66 trie.build(keyset); in TestEmptyTrie()
89 marisa::Keyset keyset; in TestTinyTrie() local
90 keyset.push_back("bach"); in TestTinyTrie()
91 keyset.push_back("bet"); in TestTinyTrie()
92 keyset.push_back("chat"); in TestTinyTrie()
93 keyset.push_back("check"); in TestTinyTrie()
94 keyset.push_back("check"); in TestTinyTrie()
97 trie.build(keyset, 1); in TestTinyTrie()
106 ASSERT(keyset[0].id() == 2); in TestTinyTrie()
107 ASSERT(keyset[1].id() == 3); in TestTinyTrie()
108 ASSERT(keyset[2].id() == 1); in TestTinyTrie()
109 ASSERT(keyset[3].id() == 0); in TestTinyTrie()
110 ASSERT(keyset[4].id() == 0); in TestTinyTrie()
113 for (std::size_t i = 0; i < keyset.size(); ++i) { in TestTinyTrie()
114 agent.set_query(keyset[i].ptr(), keyset[i].length()); in TestTinyTrie()
116 ASSERT(agent.key().id() == keyset[i].id()); in TestTinyTrie()
118 agent.set_query(keyset[i].id()); in TestTinyTrie()
120 ASSERT(agent.key().length() == keyset[i].length()); in TestTinyTrie()
121 ASSERT(std::memcmp(agent.key().ptr(), keyset[i].ptr(), in TestTinyTrie()
166 trie.build(keyset, 1 | MARISA_LABEL_ORDER); in TestTinyTrie()
175 ASSERT(keyset[0].id() == 0); in TestTinyTrie()
176 ASSERT(keyset[1].id() == 1); in TestTinyTrie()
177 ASSERT(keyset[2].id() == 2); in TestTinyTrie()
178 ASSERT(keyset[3].id() == 3); in TestTinyTrie()
179 ASSERT(keyset[4].id() == 3); in TestTinyTrie()
181 for (std::size_t i = 0; i < keyset.size(); ++i) { in TestTinyTrie()
182 agent.set_query(keyset[i].ptr(), keyset[i].length()); in TestTinyTrie()
184 ASSERT(agent.key().id() == keyset[i].id()); in TestTinyTrie()
186 agent.set_query(keyset[i].id()); in TestTinyTrie()
188 ASSERT(agent.key().length() == keyset[i].length()); in TestTinyTrie()
189 ASSERT(std::memcmp(agent.key().ptr(), keyset[i].ptr(), in TestTinyTrie()
204 marisa::Keyset *keyset) { in MakeKeyset() argument
215 keyset->push_back(key_buf, length); in MakeKeyset()
219 void TestLookup(const marisa::Trie &trie, const marisa::Keyset &keyset) { in TestLookup() argument
221 for (std::size_t i = 0; i < keyset.size(); ++i) { in TestLookup()
222 agent.set_query(keyset[i].ptr(), keyset[i].length()); in TestLookup()
224 ASSERT(agent.key().id() == keyset[i].id()); in TestLookup()
226 agent.set_query(keyset[i].id()); in TestLookup()
228 ASSERT(agent.key().length() == keyset[i].length()); in TestLookup()
229 ASSERT(std::memcmp(agent.key().ptr(), keyset[i].ptr(), in TestLookup()
235 const marisa::Keyset &keyset) { in TestCommonPrefixSearch() argument
237 for (std::size_t i = 0; i < keyset.size(); ++i) { in TestCommonPrefixSearch()
238 agent.set_query(keyset[i].ptr(), keyset[i].length()); in TestCommonPrefixSearch()
240 ASSERT(agent.key().id() <= keyset[i].id()); in TestCommonPrefixSearch()
242 ASSERT(agent.key().id() <= keyset[i].id()); in TestCommonPrefixSearch()
244 ASSERT(agent.key().id() == keyset[i].id()); in TestCommonPrefixSearch()
249 const marisa::Keyset &keyset) { in TestPredictiveSearch() argument
251 for (std::size_t i = 0; i < keyset.size(); ++i) { in TestPredictiveSearch()
252 agent.set_query(keyset[i].ptr(), keyset[i].length()); in TestPredictiveSearch()
254 ASSERT(agent.key().id() == keyset[i].id()); in TestPredictiveSearch()
256 ASSERT(agent.key().id() > keyset[i].id()); in TestPredictiveSearch()
262 marisa::NodeOrder node_order, marisa::Keyset &keyset) { in TestTrie() argument
263 for (std::size_t i = 0; i < keyset.size(); ++i) { in TestTrie()
264 keyset[i].set_weight(1.0F); in TestTrie()
268 trie.build(keyset, num_tries | tail_mode | node_order); in TestTrie()
271 ASSERT(trie.num_keys() <= keyset.size()); in TestTrie()
276 TestLookup(trie, keyset); in TestTrie()
277 TestCommonPrefixSearch(trie, keyset); in TestTrie()
278 TestPredictiveSearch(trie, keyset); in TestTrie()
286 ASSERT(trie.num_keys() <= keyset.size()); in TestTrie()
291 TestLookup(trie, keyset); in TestTrie()
315 ASSERT(trie.num_keys() <= keyset.size()); in TestTrie()
320 TestLookup(trie, keyset); in TestTrie()
326 ASSERT(trie.num_keys() <= keyset.size()); in TestTrie()
331 TestLookup(trie, keyset); in TestTrie()
341 ASSERT(trie.num_keys() <= keyset.size()); in TestTrie()
346 TestLookup(trie, keyset); in TestTrie()
350 marisa::Keyset &keyset) { in TestTrie() argument
357 TestTrie(i, tail_mode, node_order, keyset); in TestTrie()
364 marisa::Keyset keyset; in TestTrie() local
365 MakeKeyset(1000, tail_mode, &keyset); in TestTrie()
367 TestTrie(tail_mode, MARISA_WEIGHT_ORDER, keyset); in TestTrie()
368 TestTrie(tail_mode, MARISA_LABEL_ORDER, keyset); in TestTrie()