diff --git a/gensim/test/test_similarities.py b/gensim/test/test_similarities.py index 314336dadc..153a083762 100644 --- a/gensim/test/test_similarities.py +++ b/gensim/test/test_similarities.py @@ -625,7 +625,7 @@ def assertIndexSaved(self, index): fname = get_tmpfile('gensim_similarities.tst.pkl') index.save(fname) self.assertTrue(os.path.exists(fname)) - self.assertTrue(os.path.exists(fname + '.d')) + self.assertTrue(os.path.exists(fname + ".dict")) def assertLoadedIndexEqual(self, index, model): from gensim.similarities.annoy import AnnoyIndexer @@ -676,7 +676,7 @@ def test_save(self): fname = get_tmpfile('gensim_similarities.tst.pkl') self.index.save(fname) self.assertTrue(os.path.exists(fname)) - self.assertTrue(os.path.exists(fname + '.d')) + self.assertTrue(os.path.exists(fname + ".dict")) def test_load_not_exist(self): from gensim.similarities.annoy import AnnoyIndexer