def test_calculate_total_calories(self):
"""Test if correct total calories number is returned"""
training1 = self._create_sample_training(
user=self.user, name="test training1", distance=10
)
training2 = self._create_sample_training(
user=self.user, name="test training2", distance=7
)
training1.calculate_calories_burnt()
training2.calculate_calories_burnt()
expected_number = training1.calories + training2.calories
total_calories = TrainingModel.calculate_total_calories()
self.assertEqual(total_calories, expected_number)
def __then_expected_calories_data_will_be_returned(self):
expected_calories = TrainingModel.calculate_total_calories()
self.assertEqual(
expected_calories, self.response.json["calories_number"]
)
def get(cls):
calories_number = TrainingModel.calculate_total_calories()
return {"calories_number": calories_number}, 200
