def test_remove_doc(self): index = Levenshtein_search.populate_wordset(-1, self.excerpt1) Levenshtein_search.remove_string(index, 'overcoat') results = Levenshtein_search.lookup(index, 'overcoat', 6) assert results == [['went', 6, 0.024390243902439025], ['cold', 6, 0.024390243902439025], ['Versh', 6, 0.04878048780487805], ['overshoes', 4, 0.04878048780487805], ['not', 6, 0.024390243902439025]]
def unindex(self, doc): del self._doc_to_id[doc] Levenshtein_search.remove_string(self.index_key, doc)