diff --git a/movies_test.py b/movies_test.py index 07d02807f948dcceecb8ecf0538e7e5838f33e72..8211f1542ef0356b556c7b824299ae5676dcede9 100644 --- a/movies_test.py +++ b/movies_test.py @@ -29,3 +29,11 @@ class LoadTests(unittest.TestCase): self.assertEqual(user_movies[1], [1, 4]) self.assertEqual(user_movies[2], [3, 1]) self.assertEqual(set(user_movies.keys()), {1, 2}) + + def test_collect_user_context_min_rating(self): + """Check the min rating threshold""" + rating = self.toy_ratings() + user_movies = movies.collect_user_context(rating, min_rating=5) + + self.assertEqual(user_movies[1], [4]) + self.assertEqual(set(user_movies.keys()), {1})