Test the difference between topn=0 and topn=None in most_similar.

piskvorky · Apr 29, 2019 · fbfe2d4 · fbfe2d4
1 parent cc14e72
commit fbfe2d4
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/gensim/test/test_keyedvectors.py b/gensim/test/test_keyedvectors.py
@@ -106,6 +106,9 @@ def test_most_similar_topn(self):
         predicted = self.vectors.most_similar('war', topn=None)
         self.assertEqual(len(predicted), len(self.vectors.vocab))
 
+        predicted = self.vectors.most_similar('war', topn=0)
+        self.assertEqual(len(predicted), 0)
+
     def test_relative_cosine_similarity(self):
         """Test relative_cosine_similarity returns expected results with an input of a word pair and topn"""
         wordnet_syn = [