diff --git a/gensim/test/test_keyedvectors.py b/gensim/test/test_keyedvectors.py index 13a5b8c5a9..3dc04f5597 100644 --- 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 = [