def test_retrieval_based_on_insertion_order(self): trie = Trie() trie.insert("rest") trie.insert("restaurant") results = list(trie.lookup("rest")) assert len(results) == 1 results = list(trie.lookup("restaurant")) assert len(results) == 1
def test_retrieval_based_on_insertion_order(self): trie = Trie() trie.insert("rest") trie.insert("restaurant") results = list(trie.lookup("rest")) assert len(results) == 1 results = list(trie.lookup("restaurant")) assert len(results) == 1
def test_edit_distance(self): trie = Trie(max_edit_distance=1) trie.insert("restaurant") results = list(trie.lookup("restauran")) assert len(results) == 1 results = list(trie.lookup("estaurant")) assert len(results) == 1 results = list(trie.lookup("estauran")) assert len(results) == 0
def test_edit_distance(self): trie = Trie(max_edit_distance=1) trie.insert("restaurant") results = list(trie.lookup("restauran")) assert len(results) == 1 results = list(trie.lookup("estaurant")) assert len(results) == 1 results = list(trie.lookup("estauran")) assert len(results) == 0
def test_simple_remove(self): trie = Trie() trie.insert("1", "Number") results = list(trie.lookup("1")) assert len(results) == 1 assert len(results[0].get('data')) == 1 assert trie.remove("1") results = list(trie.lookup("1")) assert len(results) == 0
def test_simple_remove(self): trie = Trie() trie.insert("1", "Number") results = list(trie.lookup("1")) assert len(results) == 1 assert len(results[0].get('data')) == 1 assert trie.remove("1") results = list(trie.lookup("1")) assert len(results) == 0
def test_named_remove(self): trie = Trie() trie.insert("1", "Number") trie.insert("1", "The Loneliest") results = list(trie.lookup("1")) assert len(results) == 1 assert len(results[0].get('data')) == 2 assert trie.remove("1", "Number") results = list(trie.lookup("1")) assert len(results) == 1 assert len(results[0].get('data')) == 1
def test_remove_multi_first(self): trie = Trie(max_edit_distance=2) trie.insert("Kermit", "Muppets") trie.insert("Kermit", "Frogs") kermit_lookup = list(trie.lookup("Kermit"))[0] assert 'Frogs' in kermit_lookup['data'] assert 'Muppets' in kermit_lookup['data'] trie.remove("Kermit", "Muppets") kermit_lookup = list(trie.lookup("Kermit"))[0] assert kermit_lookup['data'] == {"Frogs"} # Right data remains
def test_named_remove(self): trie = Trie() trie.insert("1", "Number") trie.insert("1", "The Loneliest") results = list(trie.lookup("1")) assert len(results) == 1 assert len(results[0].get('data')) == 2 assert trie.remove("1", "Number") results = list(trie.lookup("1")) assert len(results) == 1 assert len(results[0].get('data')) == 1
def test_data_is_correct_on_insert(self): trie = Trie() trie.insert("restaurant", "Concept") results = list(trie.lookup("restaurant")) assert len(results) == 1 assert len(results[0].get('data')) == 1 data = list(results[0].get('data')) assert data[0] == 'Concept'
def test_data_is_correct_on_insert(self): trie = Trie() trie.insert("restaurant", "Concept") results = list(trie.lookup("restaurant")) assert len(results) == 1 assert len(results[0].get('data')) == 1 data = list(results[0].get('data')) assert data[0] == 'Concept'
def test_basic_retrieval(self): trie = Trie() trie.insert("restaurant") results = list(trie.lookup("restaurant")) assert len(results) == 1
def test_basic_retrieval(self): trie = Trie() trie.insert("restaurant") results = list(trie.lookup("restaurant")) assert len(results) == 1