++

backend/tests/test_census.py

@@ -1,116 +1,116 @@
-"""Tests for CensusService FIPS resolution and DemographicProfile model."""
-from __future__ import annotations
-
-import pytest
-
-from app.models.demographics import (
-    AgeDistribution,
-    DemographicProfile,
-    EthnicityDistribution,
-    IncomeDistribution,
-    OccupationDistribution,
-)
-from app.services.census import CensusService
-
-
-@pytest.fixture
-def service():
-    return CensusService(llm=None)
-
-
-# ── FIPS resolution ──────────────────────────────────────────────────────────
-
-class TestResolveFips:
-    def test_known_city_san_francisco(self, service):
-        result = service._resolve_fips("san francisco", "ca")
-        assert result == ("06", "67000")
-
-    def test_known_city_new_york(self, service):
-        result = service._resolve_fips("new york", "ny")
-        assert result == ("36", "51000")
-
-    def test_known_city_chicago(self, service):
-        result = service._resolve_fips("chicago", "il")
-        assert result == ("17", "14000")
-
-    def test_unknown_city_returns_none(self, service):
-        result = service._resolve_fips("nonexistent city", "zz")
-        assert result is None
-
-    def test_unknown_city_no_state_returns_none(self, service):
-        result = service._resolve_fips("timbuktu")
-        assert result is None
-
-    def test_case_insensitive_upper(self, service):
-        result = service._resolve_fips("San Francisco", "CA")
-        assert result == ("06", "67000")
-
-    def test_case_insensitive_mixed(self, service):
-        result = service._resolve_fips("NEW YORK", "NY")
-        assert result == ("36", "51000")
-
-    def test_case_insensitive_no_state(self, service):
-        result = service._resolve_fips("Seattle")
-        assert result == ("53", "63000")
-
-    def test_portland_disambiguated_oregon(self, service):
-        result = service._resolve_fips("portland", "or")
-        assert result == ("41", "59000")
-
-    def test_portland_disambiguated_maine(self, service):
-        result = service._resolve_fips("portland", "me")
-        assert result == ("23", "60545")
-
-    def test_whitespace_stripped(self, service):
-        result = service._resolve_fips("  boston  ", "  ma  ")
-        assert result == ("25", "07000")
-
-
-# ── DemographicProfile model ────────────────────────────────────────────────
-
-class TestDemographicProfile:
-    def test_full_construction(self):
-        profile = DemographicProfile(
-            city_name="Test City",
-            state="TX",
-            population=500000,
-            age=[AgeDistribution(bracket="18_24", percentage=12.5)],
-            income=[IncomeDistribution(bracket="50k_75k", percentage=22.0)],
-            occupations=[OccupationDistribution(category="service", percentage=18.3)],
-            ethnicity=[EthnicityDistribution(group="white", percentage=55.0)],
-            median_household_income=65000,
-            poverty_rate=14.2,
-            unemployment_rate=5.1,
-            homeownership_rate=58.0,
-            median_rent=1200,
-            rent_burden_rate=42.0,
-            college_education_rate=33.5,
-            median_age=34.7,
-            city_character="A vibrant test city",
-        )
-        assert profile.city_name == "Test City"
-        assert profile.population == 500000
-        assert len(profile.age) == 1
-        assert profile.age[0].bracket == "18_24"
-        assert profile.median_household_income == 65000
-        assert profile.city_character == "A vibrant test city"
-
-    def test_defaults(self):
-        profile = DemographicProfile(
-            city_name="Minimal",
-            state="CA",
-            population=1000,
-        )
-        assert profile.age == []
-        assert profile.income == []
-        assert profile.occupations == []
-        assert profile.ethnicity == []
-        assert profile.median_household_income == 0
-        assert profile.poverty_rate == 0.0
-        assert profile.unemployment_rate == 0.0
-        assert profile.homeownership_rate == 0.0
-        assert profile.median_rent == 0
-        assert profile.rent_burden_rate == 0.0
-        assert profile.college_education_rate == 0.0
-        assert profile.median_age == 0.0
-        assert profile.city_character == ""
+"""Tests for CensusService FIPS resolution and DemographicProfile model."""
+from __future__ import annotations
+
+import pytest
+
+from app.models.demographics import (
+    AgeDistribution,
+    DemographicProfile,
+    EthnicityDistribution,
+    IncomeDistribution,
+    OccupationDistribution,
+)
+from app.services.census import CensusService
+
+
+@pytest.fixture
+def service():
+    return CensusService(llm=None)
+
+
+# ── FIPS resolution ──────────────────────────────────────────────────────────
+
+class TestResolveFips:
+    def test_known_city_san_francisco(self, service):
+        result = service._resolve_fips("san francisco", "ca")
+        assert result == ("06", "67000")
+
+    def test_known_city_new_york(self, service):
+        result = service._resolve_fips("new york", "ny")
+        assert result == ("36", "51000")
+
+    def test_known_city_chicago(self, service):
+        result = service._resolve_fips("chicago", "il")
+        assert result == ("17", "14000")
+
+    def test_unknown_city_returns_none(self, service):
+        result = service._resolve_fips("nonexistent city", "zz")
+        assert result is None
+
+    def test_unknown_city_no_state_returns_none(self, service):
+        result = service._resolve_fips("timbuktu")
+        assert result is None
+
+    def test_case_insensitive_upper(self, service):
+        result = service._resolve_fips("San Francisco", "CA")
+        assert result == ("06", "67000")
+
+    def test_case_insensitive_mixed(self, service):
+        result = service._resolve_fips("NEW YORK", "NY")
+        assert result == ("36", "51000")
+
+    def test_case_insensitive_no_state(self, service):
+        result = service._resolve_fips("Seattle")
+        assert result == ("53", "63000")
+
+    def test_portland_disambiguated_oregon(self, service):
+        result = service._resolve_fips("portland", "or")
+        assert result == ("41", "59000")
+
+    def test_portland_disambiguated_maine(self, service):
+        result = service._resolve_fips("portland", "me")
+        assert result == ("23", "60545")
+
+    def test_whitespace_stripped(self, service):
+        result = service._resolve_fips("  boston  ", "  ma  ")
+        assert result == ("25", "07000")
+
+
+# ── DemographicProfile model ────────────────────────────────────────────────
+
+class TestDemographicProfile:
+    def test_full_construction(self):
+        profile = DemographicProfile(
+            city_name="Test City",
+            state="TX",
+            population=500000,
+            age=[AgeDistribution(bracket="18_24", percentage=12.5)],
+            income=[IncomeDistribution(bracket="50k_75k", percentage=22.0)],
+            occupations=[OccupationDistribution(category="service", percentage=18.3)],
+            ethnicity=[EthnicityDistribution(group="white", percentage=55.0)],
+            median_household_income=65000,
+            poverty_rate=14.2,
+            unemployment_rate=5.1,
+            homeownership_rate=58.0,
+            median_rent=1200,
+            rent_burden_rate=42.0,
+            college_education_rate=33.5,
+            median_age=34.7,
+            city_character="A vibrant test city",
+        )
+        assert profile.city_name == "Test City"
+        assert profile.population == 500000
+        assert len(profile.age) == 1
+        assert profile.age[0].bracket == "18_24"
+        assert profile.median_household_income == 65000
+        assert profile.city_character == "A vibrant test city"
+
+    def test_defaults(self):
+        profile = DemographicProfile(
+            city_name="Minimal",
+            state="CA",
+            population=1000,
+        )
+        assert profile.age == []
+        assert profile.income == []
+        assert profile.occupations == []
+        assert profile.ethnicity == []
+        assert profile.median_household_income == 0
+        assert profile.poverty_rate == 0.0
+        assert profile.unemployment_rate == 0.0
+        assert profile.homeownership_rate == 0.0
+        assert profile.median_rent == 0
+        assert profile.rent_burden_rate == 0.0
+        assert profile.college_education_rate == 0.0
+        assert profile.median_age == 0.0
+        assert profile.city_character == ""