diff --git a/gensim/test/test_similarities.py b/gensim/test/test_similarities.py index d4a8829f72..b9353124e6 100644 --- a/gensim/test/test_similarities.py +++ b/gensim/test/test_similarities.py @@ -1032,6 +1032,11 @@ def test_most_similar(self): self.assertLess(1, len(results)) self.assertGreaterEqual(4, len(results)) + # check the order of the results + result = index.most_similar(u"holiday", topn=1) + term, _ = next(result) + self.assertEqual(u"hollingworth", term) + # check that the term itself is not returned index = LevenshteinSimilarityIndex(self.dictionary) terms = [term for term, similarity in index.most_similar(u"holiday", topn=len(self.dictionary))]