add unrated test

lenskit · Jul 29, 2018 · a896398 · a896398
1 parent 453c1be
commit a896398
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/tests/test_topn_utils.py b/tests/test_topn_utils.py
@@ -0,0 +1,16 @@
+from lenskit import topn
+
+import numpy as np
+
+import lk_test_utils as lktu
+
+
+def test_unrated():
+    ratings = lktu.ml_pandas.renamed.ratings
+    unrate = topn.UnratedCandidates(ratings)
+
+    cs = unrate(100)
+    items = ratings.item.unique()
+    rated = ratings[ratings.user == 100].item.unique()
+    assert len(cs) == len(items) - len(rated)
+    assert len(np.intersect1d(cs, rated)) == 0