def test_print_rmse_3(self):
"""
test print for rounded up rmse to 2 decimal places
"""
writer = StringIO()
rmse = 0.8001
print_rmse(rmse, writer)
self.assertEqual(writer.getvalue(), "RMSE: 0.80\n")
def test_print_rmse_2(self):
w = StringIO()
movie_id = 1
customer_id = 1
calculated_rating = 1.0
actual_rating = 1.5
ratings = [(movie_id, customer_id, calculated_rating)]
answers = {movie_id: {customer_id: actual_rating}}
print_rmse(w, ratings, answers)
self.assertEqual(w.getvalue(), "RMSE: 0.50\n")