def METHOD_NAME(self):
    sample = self.full_loop(prediction_type="v_prediction")
    result_sum = torch.sum(torch.abs(sample))
    result_mean = torch.mean(torch.abs(sample))
    assert abs(result_sum.item() - 52.5302) < 1e-2
    assert abs(result_mean.item() - 0.0684) < 1e-3