Lines Matching refs:key_ids
11 FindCallback(std::vector<marisa::UInt32> *key_ids, in FindCallback() argument
13 : key_ids_(key_ids), key_lengths_(key_lengths) {} in FindCallback()
33 PredictCallback(std::vector<marisa::UInt32> *key_ids, in PredictCallback() argument
35 : key_ids_(key_ids), keys_(keys) {} in PredictCallback()
75 std::vector<marisa::UInt32> key_ids; in TestTrie() local
76 trie.build(keys, &key_ids, 1 | MARISA_WITHOUT_TAIL | MARISA_LABEL_ORDER); in TestTrie()
82 ASSERT(key_ids.size() == 5); in TestTrie()
83 ASSERT(key_ids[0] == 3); in TestTrie()
84 ASSERT(key_ids[1] == 1); in TestTrie()
85 ASSERT(key_ids[2] == 0); in TestTrie()
86 ASSERT(key_ids[3] == 3); in TestTrie()
87 ASSERT(key_ids[4] == 2); in TestTrie()
92 key_length = trie.restore(key_ids[i], key_buf, sizeof(key_buf)); in TestTrie()
94 ASSERT(trie[keys[i]] == key_ids[i]); in TestTrie()
95 ASSERT(trie[key_ids[i]] == keys[i]); in TestTrie()
107 trie.build(keys, &key_ids, 1 | MARISA_WITHOUT_TAIL | MARISA_WEIGHT_ORDER); in TestTrie()
113 ASSERT(key_ids.size() == 5); in TestTrie()
114 ASSERT(key_ids[0] == 3); in TestTrie()
115 ASSERT(key_ids[1] == 1); in TestTrie()
116 ASSERT(key_ids[2] == 2); in TestTrie()
117 ASSERT(key_ids[3] == 3); in TestTrie()
118 ASSERT(key_ids[4] == 0); in TestTrie()
121 ASSERT(trie[keys[i]] == key_ids[i]); in TestTrie()
122 ASSERT(trie[key_ids[i]] == keys[i]); in TestTrie()
191 std::vector<marisa::UInt32> key_ids; in TestPrefixTrie() local
192 trie.build(keys, &key_ids, 1 | MARISA_PREFIX_TRIE in TestPrefixTrie()
202 key_length = trie.restore(key_ids[i], key_buf, sizeof(key_buf)); in TestPrefixTrie()
204 ASSERT(trie[keys[i]] == key_ids[i]); in TestPrefixTrie()
205 ASSERT(trie[key_ids[i]] == keys[i]); in TestPrefixTrie()
210 key_length = trie.restore(key_ids[0], NULL, 0); in TestPrefixTrie()
213 EXCEPT(trie.restore(key_ids[0], NULL, 5), MARISA_PARAM_ERROR); in TestPrefixTrie()
215 key_length = trie.restore(key_ids[0], key_buf, 5); in TestPrefixTrie()
219 key_length = trie.restore(key_ids[0], key_buf, 6); in TestPrefixTrie()
223 trie.build(keys, &key_ids, 2 | MARISA_PREFIX_TRIE in TestPrefixTrie()
231 key_length = trie.restore(key_ids[i], key_buf, sizeof(key_buf)); in TestPrefixTrie()
233 ASSERT(trie[keys[i]] == key_ids[i]); in TestPrefixTrie()
234 ASSERT(trie[key_ids[i]] == keys[i]); in TestPrefixTrie()
239 key_length = trie.restore(key_ids[0], NULL, 0); in TestPrefixTrie()
242 EXCEPT(trie.restore(key_ids[0], NULL, 5), MARISA_PARAM_ERROR); in TestPrefixTrie()
244 key_length = trie.restore(key_ids[0], key_buf, 5); in TestPrefixTrie()
248 key_length = trie.restore(key_ids[0], key_buf, 6); in TestPrefixTrie()
252 trie.build(keys, &key_ids, 2 | MARISA_PREFIX_TRIE in TestPrefixTrie()
260 key_length = trie.restore(key_ids[i], key_buf, sizeof(key_buf)); in TestPrefixTrie()
262 ASSERT(trie[keys[i]] == key_ids[i]); in TestPrefixTrie()
263 ASSERT(trie[key_ids[i]] == keys[i]); in TestPrefixTrie()
279 key_length = trie.restore(key_ids[i], key_buf, sizeof(key_buf)); in TestPrefixTrie()
281 ASSERT(trie[keys[i]] == key_ids[i]); in TestPrefixTrie()
282 ASSERT(trie[key_ids[i]] == keys[i]); in TestPrefixTrie()
297 key_length = trie.restore(key_ids[i], key_buf, sizeof(key_buf)); in TestPrefixTrie()
299 ASSERT(trie[keys[i]] == key_ids[i]); in TestPrefixTrie()
300 ASSERT(trie[key_ids[i]] == keys[i]); in TestPrefixTrie()
305 trie.build(keys, &key_ids, 3 | MARISA_PREFIX_TRIE in TestPrefixTrie()
313 key_length = trie.restore(key_ids[i], key_buf, sizeof(key_buf)); in TestPrefixTrie()
315 ASSERT(trie[keys[i]] == key_ids[i]); in TestPrefixTrie()
316 ASSERT(trie[key_ids[i]] == keys[i]); in TestPrefixTrie()
417 std::vector<marisa::UInt32> key_ids; in TestPatriciaTrie() local
418 trie.build(keys, &key_ids, 1); in TestPatriciaTrie()
424 ASSERT(key_ids.size() == 5); in TestPatriciaTrie()
425 ASSERT(key_ids[0] == 2); in TestPatriciaTrie()
426 ASSERT(key_ids[1] == 3); in TestPatriciaTrie()
427 ASSERT(key_ids[2] == 1); in TestPatriciaTrie()
428 ASSERT(key_ids[3] == 0); in TestPatriciaTrie()
429 ASSERT(key_ids[4] == 0); in TestPatriciaTrie()
434 key_length = trie.restore(key_ids[i], key_buf, sizeof(key_buf)); in TestPatriciaTrie()
436 ASSERT(trie[keys[i]] == key_ids[i]); in TestPatriciaTrie()
437 ASSERT(trie[key_ids[i]] == keys[i]); in TestPatriciaTrie()
442 trie.build(keys, &key_ids, 2 | MARISA_WITHOUT_TAIL); in TestPatriciaTrie()
449 key_length = trie.restore(key_ids[i], key_buf, sizeof(key_buf)); in TestPatriciaTrie()
451 ASSERT(trie[keys[i]] == key_ids[i]); in TestPatriciaTrie()
452 ASSERT(trie[key_ids[i]] == keys[i]); in TestPatriciaTrie()
457 trie.build(keys, &key_ids, 2); in TestPatriciaTrie()
464 key_length = trie.restore(key_ids[i], key_buf, sizeof(key_buf)); in TestPatriciaTrie()
466 ASSERT(trie[keys[i]] == key_ids[i]); in TestPatriciaTrie()
467 ASSERT(trie[key_ids[i]] == keys[i]); in TestPatriciaTrie()
472 trie.build(keys, &key_ids, 3 | MARISA_WITHOUT_TAIL); in TestPatriciaTrie()
479 key_length = trie.restore(key_ids[i], key_buf, sizeof(key_buf)); in TestPatriciaTrie()
481 ASSERT(trie[keys[i]] == key_ids[i]); in TestPatriciaTrie()
482 ASSERT(trie[key_ids[i]] == keys[i]); in TestPatriciaTrie()
497 key_length = trie.restore(key_ids[i], key_buf, sizeof(key_buf)); in TestPatriciaTrie()
499 ASSERT(trie[keys[i]] == key_ids[i]); in TestPatriciaTrie()
500 ASSERT(trie[key_ids[i]] == keys[i]); in TestPatriciaTrie()
515 std::vector<marisa::UInt32> key_ids; in TestEmptyString() local
516 trie.build(keys, &key_ids); in TestEmptyString()
522 ASSERT(key_ids.size() == 1); in TestEmptyString()
523 ASSERT(key_ids[0] == 0); in TestEmptyString()
579 std::vector<marisa::UInt32> key_ids; in TestBinaryKey() local
580 trie.build(keys, &key_ids, 1 | MARISA_WITHOUT_TAIL); in TestBinaryKey()
585 ASSERT(key_ids.size() == 1); in TestBinaryKey()
591 ASSERT(trie[keys[0]] == key_ids[0]); in TestBinaryKey()
592 ASSERT(trie[key_ids[0]] == keys[0]); in TestBinaryKey()
595 trie.build(keys, &key_ids, 1 | MARISA_PREFIX_TRIE | MARISA_BINARY_TAIL); in TestBinaryKey()
600 ASSERT(key_ids.size() == 1); in TestBinaryKey()
604 ASSERT(trie[keys[0]] == key_ids[0]); in TestBinaryKey()
605 ASSERT(trie[key_ids[0]] == keys[0]); in TestBinaryKey()
608 trie.build(keys, &key_ids, 1 | MARISA_PREFIX_TRIE | MARISA_TEXT_TAIL); in TestBinaryKey()
613 ASSERT(key_ids.size() == 1); in TestBinaryKey()
617 ASSERT(trie[keys[0]] == key_ids[0]); in TestBinaryKey()
618 ASSERT(trie[key_ids[0]] == keys[0]); in TestBinaryKey()
624 ASSERT(ids[0] == key_ids[0]); in TestBinaryKey()
629 ASSERT(ids[1] == key_ids[0]); in TestBinaryKey()