def test_get_workout_session(self):
'''
Test the get_workout_session method
'''
user1 = User.objects.get(pk=1)
user2 = User.objects.get(pk=2)
workout1 = Workout.objects.get(pk=2)
workout2 = Workout.objects.get(pk=2)
WorkoutLog.objects.all().delete()
l = WorkoutLog()
l.user = user1
l.date = datetime.date(2014, 1, 5)
l.exercise = Exercise.objects.get(pk=1)
l.workout = workout1
l.weight = 10
l.reps = 10
l.save()
session1 = WorkoutSession()
session1.user = user1
session1.workout = workout1
session1.notes = 'Something here'
session1.impression = '3'
session1.date = datetime.date(2014, 1, 5)
session1.save()
session2 = WorkoutSession()
session2.user = user1
session2.workout = workout1
session2.notes = 'Something else here'
session2.impression = '1'
session2.date = datetime.date(2014, 1, 1)
session2.save()
session3 = WorkoutSession()
session3.user = user2
session3.workout = workout2
session3.notes = 'The notes here'
session3.impression = '2'
session3.date = datetime.date(2014, 1, 5)
session3.save()
self.assertEqual(l.get_workout_session(), session1)
print("** Generating {0} nutrition plan(s) per user".format(
args.number_nutrition_logs))
if args.add_to_user:
userlist = [User.objects.get(pk=args.add_to_user)]
else:
userlist = [i for i in User.objects.all()]
# Load all ingredients to a list
ingredient_list = [i for i in Ingredient.objects.order_by('?').all()[:100]]
for user in userlist:
plan_list = NutritionPlan.objects.order_by('?').filter(user=user)
print(' - generating for {0}'.format(user.username))
# Add diary entries
for plan in NutritionPlan.objects.filter(user=user):
for i in range(0, args.number_diary_dates):
date = timezone.now() - datetime.timedelta(
days=random.randint(0, 100),
hours=random.randint(0, 12),
minutes=random.randint(0, 59))
for j in range(0, args.number_nutrition_logs):
ingredient = random.choice(ingredient_list)
log = LogItem(plan=plan,
datetime=date,
ingredient=ingredient,
weight_unit=None,
amount=random.randint(10, 300))
log.save()
