def test_answer_cache(self):
""" Tests that the AnswerCache returns the actual ratings"""
cache = Netflix.AnswerCache(answers())
self.assertEqual(cache.predict(1, 1), 4)
self.assertEqual(cache.predict(1, 2), 5)
self.assertEqual(cache.predict(2, 2), 4)
def test_answer_cache_invalid_user(self):
""" Tests handling invalid user """
cache = Netflix.AnswerCache(answers())
with self.assertRaises(KeyError):
cache.predict(5, 1)
