def test_save_load(self): before_save = self.symSpell.lookup("tke", Verbosity.CLOSEST)[0].term before_max_length = self.symSpell.max_length() os.makedirs("temp", exist_ok=True) self.symSpell.save_pickle("temp/temp.bin") load_sym_spell = SymSpell() load_sym_spell.load_pickle("temp/temp.bin") after_load = load_sym_spell.lookup("tke", Verbosity.CLOSEST)[0].term after_max_length = load_sym_spell.max_length() os.remove("temp/temp.bin") os.rmdir("temp") assert (before_save == after_load) assert (before_max_length == after_max_length)
def test_pickle_compressed(self): pickle_path = os.path.join(self.fortests_path, "dictionary.pickle") edit_distance_max = 2 prefix_length = 7 sym_spell = SymSpell(edit_distance_max, prefix_length) sym_spell.load_dictionary(self.dictionary_path, 0, 1) sym_spell.save_pickle(pickle_path) sym_spell_2 = SymSpell(edit_distance_max, prefix_length) sym_spell_2.load_pickle(pickle_path) self.assertEqual(sym_spell.max_length(), sym_spell_2.max_length()) self.assertEqual( sym_spell.lookup("flam", Verbosity.TOP, 0, True)[0].term, sym_spell_2.lookup("flam", Verbosity.TOP, 0, True)[0].term) os.remove(pickle_path)