Fix two broken tests (were previously skipped)

piskvorky · Nov 7, 2024 · b147918 · b147918
1 parent 3915c00
commit b147918
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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