Test that the AnnoyIndexer is disabled in most_similar when topn is None

piskvorky · Apr 29, 2019 · 2f16efe · 2f16efe
1 parent fbfe2d4
commit 2f16efe
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/gensim/test/test_similarities.py b/gensim/test/test_similarities.py
@@ -608,6 +608,12 @@ def assertApproxNeighborsMatchExact(self, model, wv, index):
 
         self.assertEqual(approx_words, exact_words)
 
+    def assertAllSimilaritiesDisableIndexer(self, model, wv, index):
+        vector = wv.vectors_norm[0]
+        approx_neighbors = model.wv.most_similar([vector], topn=None, indexer=index)
+
+        self.assertEqual(len(predicted), len(wv.vectors.vocab))
+
     def assertIndexSaved(self, index):
         fname = get_tmpfile('gensim_similarities.tst.pkl')
         index.save(fname)