コード例 #1
0
 def test_get_descendants_nodes(self, word, expected_results):
     auto_complete = AutoComplete(words=WIKIPEDIA_WORDS, synonyms=SYNONYMS)
     matched_prefix_of_last_word, rest_of_word, node, matched_words = auto_complete._prefix_autofill(
         word)
     size = 2
     found_words_gen = node.get_descendants_nodes(size=size)
     found_words = [_node.word for _node in found_words_gen][:size + 1]
     print(f'word: {word}')
     print(f'expected_results: {expected_results}')
     print(f'found_words: {found_words}')
     assert expected_results == list(found_words)
コード例 #2
0
 def test_prefix_autofill(self, word, expected_matched_prefix_of_last_word,
                          expected_rest_of_word, expected_matched_words,
                          expected_node_path):
     auto_complete = AutoComplete(words=WIKIPEDIA_WORDS, synonyms=SYNONYMS)
     matched_prefix_of_last_word, rest_of_word, node, matched_words = auto_complete._prefix_autofill(
         word)
     print(f'word: {word}')
     print(
         f'expected_matched_prefix_of_last_word: {expected_matched_prefix_of_last_word}'
     )
     print(f'matched_prefix_of_last_word: {matched_prefix_of_last_word}')
     print(f'expected_rest_of_word: {expected_rest_of_word}')
     print(f'rest_of_word: {rest_of_word}')
     print(f'node: {node}')
     print(f'expected_matched_words: {expected_matched_words}')
     print(f'matched_words: {matched_words}')
     expected_node = auto_complete._dwg
     for k in expected_node_path.split(','):
         expected_node = expected_node[k]
     assert expected_node is node
     assert expected_matched_prefix_of_last_word == matched_prefix_of_last_word
     assert expected_rest_of_word == rest_of_word
     assert expected_matched_words == matched_words