Fixes piskvorky#1828 , Tests Added

sj29-innovate · Jan 9, 2018 · f1f65e2 · f1f65e2
1 parent 77409dd
commit f1f65e2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/gensim/test/test_BM25.py b/gensim/test/test_BM25.py
@@ -27,7 +27,9 @@ def test_max_match_with_itself(self):
     def test_nonnegative_weights(self):
         """ All the weights for a partiular document should be non negative """
         weights = get_bm25_weights(common_texts)
-        self.assertTrue(min(weights) >= 0)
+        for doc_weights in weights:
+        	for weight in doc_weights:
+        		self.assertTrue(weight >=0 )
 
     def test_same_match_with_same_document(self):
         """ A document should always get the same weight when matched with a particular document """