test(wave-2): add pure-function tests for models and prompts (#17)

Covers haversine_miles (5 cases: zero, known distance, symmetry, short,
return type) and PatientProfile.summary (14 cases: all fields, phases,
study types, edge cases). Covers lookup_disease_profile (15 cases: exact
match, case-insensitive, synonyms, multi-disease registry, partial
containment, unknown, structure validation).

All 35 tests pass with no mocks — confirms pytest wiring and that the
data/diseases/ JSON registry is structurally intact.

Co-authored-by: Claude Sonnet 4.6 <noreply@anthropic.com>

tests/test_models.py

@@ -0,0 +1,120 @@
+"""Tests for models.py — pure deterministic functions only."""
+from __future__ import annotations
+
+import math
+
+import pytest
+
+from models import PatientProfile, haversine_miles
+
+
+# ---------------------------------------------------------------------------
+# haversine_miles
+# ---------------------------------------------------------------------------
+
+class TestHaversineMiles:
+    def test_same_point_is_zero(self):
+        assert haversine_miles(42.0, -71.0, 42.0, -71.0) == 0.0
+
+    def test_known_distance_boston_nyc(self):
+        # Boston (42.3601, -71.0589) to NYC (40.7128, -74.0060) ≈ 190 miles
+        dist = haversine_miles(42.3601, -71.0589, 40.7128, -74.0060)
+        assert 185 < dist < 195
+
+    def test_symmetry(self):
+        d1 = haversine_miles(42.0, -71.0, 34.0, -118.0)
+        d2 = haversine_miles(34.0, -118.0, 42.0, -71.0)
+        assert math.isclose(d1, d2, rel_tol=1e-9)
+
+    def test_short_distance(self):
+        # Two points ~1 mile apart
+        dist = haversine_miles(42.3370, -71.1061, 42.3514, -71.1061)
+        assert 0.8 < dist < 1.2
+
+    def test_returns_float(self):
+        result = haversine_miles(0.0, 0.0, 0.0, 1.0)
+        assert isinstance(result, float)
+
+
+# ---------------------------------------------------------------------------
+# PatientProfile.summary
+# ---------------------------------------------------------------------------
+
+class TestPatientProfileSummary:
+    def test_summary_contains_disease(self, als_patient):
+        assert "Amyotrophic Lateral Sclerosis" in als_patient.summary()
+
+    def test_summary_contains_age(self, als_patient):
+        assert "52" in als_patient.summary()
+
+    def test_summary_contains_onset_months(self, als_patient):
+        assert "18" in als_patient.summary()
+
+    def test_summary_contains_diagnosis_months(self, als_patient):
+        assert "12" in als_patient.summary()
+
+    def test_summary_contains_zip(self, als_patient):
+        assert "02115" in als_patient.summary()
+
+    def test_summary_contains_radius(self, als_patient):
+        assert "100" in als_patient.summary()
+
+    def test_summary_contains_benchmarks(self, als_patient):
+        text = als_patient.summary()
+        assert "alsfrs_r" in text
+        assert "38" in text
+
+    def test_summary_no_benchmarks(self):
+        patient = PatientProfile(
+            disease="ALS", age=45, onset_months=6,
+            zip_code="10001", lat=40.75, lon=-73.99,
+        )
+        assert "Benchmarks" not in patient.summary()
+
+    def test_summary_phase_labels(self):
+        patient = PatientProfile(
+            disease="ALS", age=45, onset_months=6,
+            zip_code="10001", lat=40.75, lon=-73.99,
+            phases=["2", "3", "0", "na"],
+        )
+        text = patient.summary()
+        assert "Phase 2" in text
+        assert "Phase 3" in text
+        assert "Early Phase 1" in text
+        assert "Not Applicable" in text
+
+    def test_summary_study_type_defaults(self):
+        patient = PatientProfile(
+            disease="ALS", age=45, onset_months=6,
+            zip_code="10001", lat=40.75, lon=-73.99,
+        )
+        text = patient.summary()
+        assert "Clinical trials" in text
+        assert "Observational" not in text
+        assert "EAP" not in text
+
+    def test_summary_include_observational(self):
+        patient = PatientProfile(
+            disease="ALS", age=45, onset_months=6,
+            zip_code="10001", lat=40.75, lon=-73.99,
+            include_observational=True,
+        )
+        assert "Observational" in patient.summary()
+
+    def test_summary_include_eap(self):
+        patient = PatientProfile(
+            disease="ALS", age=45, onset_months=6,
+            zip_code="10001", lat=40.75, lon=-73.99,
+            include_eap=True,
+        )
+        assert "EAP" in patient.summary()
+
+    def test_summary_returns_string(self, als_patient):
+        assert isinstance(als_patient.summary(), str)
+
+    def test_summary_no_phases(self):
+        patient = PatientProfile(
+            disease="ALS", age=45, onset_months=6,
+            zip_code="10001", lat=40.75, lon=-73.99,
+        )
+        assert "Phases:" not in patient.summary()

tests/test_prompts.py

@@ -0,0 +1,98 @@
+"""Tests for prompts.py — lookup_disease_profile function."""
+from __future__ import annotations
+
+import pytest
+
+from prompts import lookup_disease_profile
+
+
+class TestLookupDiseaseProfile:
+    # --- exact full_name matches ---
+
+    def test_exact_full_name(self):
+        result = lookup_disease_profile("Amyotrophic Lateral Sclerosis")
+        assert result is not None
+        assert result["id"] == "als"
+
+    def test_case_insensitive_full_name(self):
+        result = lookup_disease_profile("amyotrophic lateral sclerosis")
+        assert result is not None
+        assert result["id"] == "als"
+
+    def test_mixed_case_full_name(self):
+        result = lookup_disease_profile("AMYOTROPHIC LATERAL SCLEROSIS")
+        assert result is not None
+        assert result["id"] == "als"
+
+    # --- synonym matches ---
+
+    def test_synonym_als(self):
+        result = lookup_disease_profile("ALS")
+        assert result is not None
+        assert result["id"] == "als"
+
+    def test_synonym_lou_gehrig(self):
+        result = lookup_disease_profile("Lou Gehrig's disease")
+        assert result is not None
+        assert result["id"] == "als"
+
+    def test_synonym_mnd(self):
+        result = lookup_disease_profile("motor neuron disease")
+        assert result is not None
+        assert result["id"] == "als"
+
+    # --- other diseases in registry ---
+
+    def test_huntingtons_exact(self):
+        result = lookup_disease_profile("Huntington's Disease")
+        assert result is not None
+        assert result["id"] == "huntingtons"
+
+    def test_parkinsons_exact(self):
+        result = lookup_disease_profile("Parkinson's Disease")
+        assert result is not None
+        assert result["id"] == "parkinsons"
+
+    def test_sma_synonym(self):
+        result = lookup_disease_profile("SMA")
+        assert result is not None
+        assert result["id"] == "sma"
+
+    # --- partial containment fallback ---
+
+    def test_partial_containment_als_in_longer_string(self):
+        result = lookup_disease_profile("ALS (Amyotrophic Lateral Sclerosis)")
+        assert result is not None
+        assert result["id"] == "als"
+
+    # --- unknown disease ---
+
+    def test_unknown_disease_returns_none(self):
+        assert lookup_disease_profile("Totally Unknown Rare Disease XYZ") is None
+
+    def test_truly_unknown_disease_returns_none(self):
+        # A name that shares no substring with any registered disease
+        assert lookup_disease_profile("xyzzy-9999-zqj-unknown") is None
+
+    # --- profile structure ---
+
+    def test_profile_has_required_keys(self):
+        result = lookup_disease_profile("Amyotrophic Lateral Sclerosis")
+        assert result is not None
+        assert "id" in result
+        assert "full_name" in result
+        assert "synonyms" in result
+        assert "benchmarks" in result
+
+    def test_benchmarks_is_list(self):
+        result = lookup_disease_profile("Amyotrophic Lateral Sclerosis")
+        assert isinstance(result["benchmarks"], list)
+
+    def test_synonyms_is_list(self):
+        result = lookup_disease_profile("Amyotrophic Lateral Sclerosis")
+        assert isinstance(result["synonyms"], list)
+
+    def test_whitespace_stripped(self):
+        result = lookup_disease_profile("  Amyotrophic Lateral Sclerosis  ")
+        assert result is not None
+        assert result["id"] == "als"