Update test_recommender.py

PreferredAI · May 22, 2024 · 56a0ae7 · 56a0ae7
1 parent eb16fcf
commit 56a0ae7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/cornac/models/test_recommender.py b/tests/cornac/models/test_recommender.py
@@ -88,9 +88,9 @@ def test_fit(self):
         model.score(0, [])
         result = model.rank(0, history_items=[])
         self.assertTrue((result[0][0:3] == [3, 2, 4]).all())
-        self.assertTrue((np.sort(result[0][3:5]) == [0, 1]).any()) # identical scores, sorting may affect the ordering
+        self.assertTrue((np.sort(result[0][3:5]) == [0, 1]).all()) # identical scores, sorting may affect the ordering
         self.assertTrue((result[0][5:6] == [5]).all())
-        self.assertTrue((np.sort(result[0][6:9]) == [6, 7, 8]).any()) # identical scores, sorting may affect the ordering
+        self.assertTrue((np.sort(result[0][6:9]) == [6, 7, 8]).all()) # identical scores, sorting may affect the ordering
 
 
 if __name__ == "__main__":