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)