feat: add state legislative districts support

- Add StateLegislativeDistrict model

- Add state_legislative_districts and state_legislative_districts_next to GeocodioFields

- Add unit and integration tests for state legislative districts

Author: funkatron

tests/e2e/test_api.py

@@ -178,4 +178,47 @@ def test_integration_batch_geocode(client):
     assert components.suffix == "Ave"
     assert components.city == "Denver"
     assert components.state == "CO"
-    assert components.zip == "80203"
+    assert components.zip == "80203"
+
+
+def test_integration_with_state_legislative_districts(client):
+    """Test real API call with state legislative district fields."""
+    # Test address
+    address = "1600 Pennsylvania Ave NW, Washington, DC"
+
+    # Request additional fields
+    response = client.geocode(
+        address,
+        fields=["stateleg", "stateleg-next"]
+    )
+
+    # 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 state legislative districts
+    if fields.state_legislative_districts:
+        district = fields.state_legislative_districts[0]
+        assert district.name is not None
+        assert isinstance(district.district_number, int)
+        assert district.chamber in ["house", "senate"]
+        if district.ocd_id:
+            assert isinstance(district.ocd_id, str)
+        if district.proportion:
+            assert isinstance(district.proportion, float)
+
+    # Check upcoming state legislative districts
+    if fields.state_legislative_districts_next:
+        district = fields.state_legislative_districts_next[0]
+        assert district.name is not None
+        assert isinstance(district.district_number, int)
+        assert district.chamber in ["house", "senate"]
+        if district.ocd_id:
+            assert isinstance(district.ocd_id, str)
+        if district.proportion:
+            assert isinstance(district.proportion, float)

tests/test_models.py

@@ -1,7 +1,7 @@
 import pytest
 from geocodio.models import (
     AddressComponents, Timezone, CongressionalDistrict,
-    GeocodioFields, GeocodingResult, GeocodingResponse, Location
+    GeocodioFields, GeocodingResult, GeocodingResponse, Location, StateLegislativeDistrict
 )
 
 
@@ -88,4 +88,26 @@ def test_geocoding_result_without_fields():
     assert result.fields is None
     assert result.address_components.city == "Arlington"
     assert result.location.lat == 38.886672
-    assert result.location.lng == -77.094735
+    assert result.location.lng == -77.094735
+
+
+def test_state_legislative_district_extras():
+    # Test that extra fields are stored in extras
+    data = {
+        "name": "Virginia House District 8",
+        "district_number": 8,
+        "chamber": "house",
+        "ocd_id": "ocd-division/country:us/state:va/sldl:8",
+        "proportion": 1.0,
+        "extra_field": "extra value"
+    }
+
+    district = StateLegislativeDistrict.from_api(data)
+
+    assert district.name == "Virginia House District 8"
+    assert district.district_number == 8
+    assert district.chamber == "house"
+    assert district.ocd_id == "ocd-division/country:us/state:va/sldl:8"
+    assert district.proportion == 1.0
+    assert district.get_extra("extra_field") == "extra value"
+    assert district.get_extra("nonexistent", "default") == "default"