feat: add census2023 support

- Add MSA and CSA codes to CensusData model

- Add census2023 to GeocodioFields

- Add unit and integration tests for census2023

Author: funkatron

src/geocodio/client.py

@@ -213,6 +213,11 @@ def _parse_fields(self, fields_data: dict | None) -> GeocodioFields | None:
             if "census2020" in fields_data else None
         )
 
+        census2023 = (
+            CensusData.from_api(fields_data["census2023"])
+            if "census2023" in fields_data else None
+        )
+
         acs = (
             ACSSurveyData.from_api(fields_data["acs"])
             if "acs" in fields_data else None
@@ -226,5 +231,6 @@ def _parse_fields(self, fields_data: dict | None) -> GeocodioFields | None:
             school_districts=school_districts,
             census2010=census2010,
             census2020=census2020,
+            census2023=census2023,
             acs=acs,
         )

src/geocodio/models.py

@@ -106,6 +106,8 @@ class CensusData(_HasExtras, ApiModelMixin):
     tract: Optional[str] = None
     county_fips: Optional[str] = None
     state_fips: Optional[str] = None
+    msa_code: Optional[str] = None  # Metropolitan Statistical Area
+    csa_code: Optional[str] = None  # Combined Statistical Area
     extras: Dict[str, Any] = field(default_factory=dict, repr=False)
 
 
@@ -146,6 +148,7 @@ class GeocodioFields:
     school_districts: Optional[List[SchoolDistrict]] = None
     census2010: Optional[CensusData] = None
     census2020: Optional[CensusData] = None
+    census2023: Optional[CensusData] = None
     acs: Optional[ACSSurveyData] = None
 
 

tests/e2e/test_api.py

@@ -253,4 +253,36 @@ def test_integration_with_school_districts(client):
         if district.lea_id:
             assert isinstance(district.lea_id, str)
         if district.nces_id:
-            assert isinstance(district.nces_id, str)
+            assert isinstance(district.nces_id, str)
+
+
+def test_integration_with_census2023(client):
+    """Test real API call with census2023 field."""
+    # Test address
+    address = "1600 Pennsylvania Ave NW, Washington, DC"
+
+    # Request additional fields
+    response = client.geocode(
+        address,
+        fields=["census2023"]
+    )
+
+    # Verify response structure
+    assert response is not None
+    assert len(response.results) > 0
+    result = response.results[0]
+
+    # Verify fields data
+    fields = result.fields
+    assert fields is not None
+
+    # Check census2023 data
+    if fields.census2023:
+        assert fields.census2023.tract is not None
+        assert fields.census2023.block is not None
+        assert fields.census2023.county_fips is not None
+        assert fields.census2023.state_fips is not None
+        if fields.census2023.msa_code:
+            assert isinstance(fields.census2023.msa_code, str)
+        if fields.census2023.csa_code:
+            assert isinstance(fields.census2023.csa_code, str)

tests/test_models.py

@@ -1,7 +1,7 @@
 import pytest
 from geocodio.models import (
     AddressComponents, Timezone, CongressionalDistrict,
-    GeocodioFields, GeocodingResult, GeocodingResponse, Location, StateLegislativeDistrict, SchoolDistrict
+    GeocodioFields, GeocodingResult, GeocodingResponse, Location, StateLegislativeDistrict, SchoolDistrict, CensusData
 )
 
 
@@ -130,4 +130,30 @@ def test_school_district_extras():
     assert district.lea_id == "5100000"
     assert district.nces_id == "5100000"
     assert district.get_extra("extra_field") == "extra value"
-    assert district.get_extra("nonexistent", "default") == "default"
+    assert district.get_extra("nonexistent", "default") == "default"
+
+
+def test_census_data_extras():
+    # Test that extra fields are stored in extras
+    data = {
+        "block": "1000",
+        "blockgroup": "1",
+        "tract": "000100",
+        "county_fips": "51013",
+        "state_fips": "51",
+        "msa_code": "47900",
+        "csa_code": "548",
+        "extra_field": "extra value"
+    }
+
+    census = CensusData.from_api(data)
+
+    assert census.block == "1000"
+    assert census.blockgroup == "1"
+    assert census.tract == "000100"
+    assert census.county_fips == "51013"
+    assert census.state_fips == "51"
+    assert census.msa_code == "47900"
+    assert census.csa_code == "548"
+    assert census.get_extra("extra_field") == "extra value"
+    assert census.get_extra("nonexistent", "default") == "default"