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")