diff --git a/tests/cornac/models/test_recommender.py b/tests/cornac/models/test_recommender.py index d29b05a3..12d5f413 100644 --- a/tests/cornac/models/test_recommender.py +++ b/tests/cornac/models/test_recommender.py @@ -12,7 +12,7 @@ # See the License for the specific language governing permissions and # limitations under the License. # ============================================================================ - +import sys; sys.path.append('./') import unittest from cornac.data import BasketDataset, Dataset, SequentialDataset, Reader @@ -85,7 +85,10 @@ def test_fit(self): model.fit(dataset) model.score(0, []) result = model.rank(0, history_items=[]) - self.assertTrue((result[0] == [3, 2, 4, 1, 0, 5, 8, 7, 6]).all()) + self.assertTrue((result[0][0:3] == [3, 2, 4]).all()) + self.assertTrue((result[0][3:5] == [0, 1]).any()) # identical scores, sorting may affect the ordering + self.assertTrue((result[0][5:6] == [5]).all()) + self.assertTrue((result[0][6:9] == [8, 7, 6]).any()) # identical scores, sorting may affect the ordering if __name__ == "__main__":