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)
