Tighten warmup selection to dynamic prep only

2026-02-23 11:25:11 -06:00
parent 03681c532d
commit 909c75d8ee
2 changed files with 106 additions and 13 deletions
--- a/generator/tests/test_warmup_selector.py
+++ b/generator/tests/test_warmup_selector.py
@@ -0,0 +1,86 @@
+from django.contrib.auth.models import User
+from django.test import TestCase
+
+from exercise.models import Exercise
+from generator.models import UserPreference
+from generator.services.exercise_selector import ExerciseSelector
+from registered_user.models import RegisteredUser
+
+
+class TestWarmupSelector(TestCase):
+    def setUp(self):
+        django_user = User.objects.create_user(
+            username='warmup_selector_user',
+            password='testpass123',
+        )
+        registered_user = RegisteredUser.objects.create(
+            user=django_user,
+            first_name='Warmup',
+            last_name='Tester',
+        )
+        self.preference = UserPreference.objects.create(
+            registered_user=registered_user,
+            days_per_week=4,
+            fitness_level=2,
+        )
+
+    def test_warmup_excludes_working_set_movements(self):
+        dynamic_1 = Exercise.objects.create(
+            name='Dynamic Warmup A',
+            movement_patterns='dynamic stretch, mobility - dynamic, activation, warm up',
+            is_duration=True,
+            is_reps=False,
+            is_weight=False,
+            is_compound=False,
+            exercise_tier='accessory',
+            hr_elevation_rating=2,
+            complexity_rating=2,
+            difficulty_level='beginner',
+        )
+        dynamic_2 = Exercise.objects.create(
+            name='Dynamic Warmup B',
+            movement_patterns='mobility - dynamic, cardio/locomotion, balance',
+            is_duration=True,
+            is_reps=False,
+            is_weight=False,
+            is_compound=False,
+            exercise_tier='accessory',
+            hr_elevation_rating=3,
+            complexity_rating=2,
+            difficulty_level='beginner',
+        )
+
+        weighted_press = Exercise.objects.create(
+            name='Lying Dumbbell Tricep Extension',
+            movement_patterns='upper push - horizontal, upper push, arms',
+            is_duration=True,
+            is_reps=False,
+            is_weight=True,
+            is_compound=False,
+            exercise_tier='secondary',
+            hr_elevation_rating=2,
+            complexity_rating=2,
+            difficulty_level='intermediate',
+        )
+        duration_push = Exercise.objects.create(
+            name='Floor Press Hold',
+            movement_patterns='upper push - horizontal, upper push',
+            is_duration=True,
+            is_reps=False,
+            is_weight=False,
+            is_compound=False,
+            exercise_tier='secondary',
+            hr_elevation_rating=2,
+            complexity_rating=2,
+            difficulty_level='intermediate',
+        )
+
+        selector = ExerciseSelector(self.preference)
+        selected = selector.select_warmup_exercises(target_muscles=[], count=4)
+
+        selected_ids = {ex.pk for ex in selected}
+
+        self.assertIn(dynamic_1.pk, selected_ids)
+        self.assertIn(dynamic_2.pk, selected_ids)
+        self.assertNotIn(weighted_press.pk, selected_ids)
+        self.assertNotIn(duration_push.pk, selected_ids)