test: update tests to use renamed Geocodio class

Update all test files and smoke tests to import and use the new Geocodio class name instead of GeocodioClient.

Author: MiniCodeMonkey

smoke.py

@@ -1,6 +1,6 @@
 #!/usr/bin/env python3
 
-from geocodio import GeocodioClient
+from geocodio import Geocodio
 from dotenv import load_dotenv
 import os
 import json
@@ -10,7 +10,7 @@
 load_dotenv()
 
 # Initialize the client with your API key
-client = GeocodioClient(os.getenv("GEOCODIO_API_KEY"))
+client = Geocodio(os.getenv("GEOCODIO_API_KEY"))
 
 # Single forward geocode
 print("\nSingle forward geocode:")

smoke_lists.py

@@ -3,7 +3,7 @@
 import os
 import time
 import logging
-from geocodio import GeocodioClient
+from geocodio import Geocodio
 from geocodio.models import ListProcessingState
 from dotenv import load_dotenv
 
@@ -51,7 +51,7 @@ def main():
         logger.error("GEOCODIO_API_KEY not set in environment.")
         exit(1)
 
-    client = GeocodioClient(api_key)
+    client = Geocodio(api_key)
 
     # Step 1: Create a list
     logger.info("Creating a new list...")

tests/conftest.py

@@ -6,7 +6,7 @@
 import logging
 from dotenv import load_dotenv
 import pytest
-from geocodio import GeocodioClient
+from geocodio import Geocodio
 
 # Load environment variables from .env file
 load_dotenv()
@@ -18,15 +18,15 @@
 
 @pytest.fixture
 def client(request):
-    """Create a GeocodioClient instance with test configuration"""
+    """Create a Geocodio instance with test configuration"""
     # Use TEST_KEY for all tests except e2e tests
     if "e2e" in request.node.fspath.strpath:
         logger.debug("Running e2e tests - using API key from environment")
         api_key = os.getenv("GEOCODIO_API_KEY")
         if not api_key:
             logger.warning("GEOCODIO_API_KEY not set - skipping e2e test")
             pytest.skip("GEOCODIO_API_KEY environment variable not set")
-        return GeocodioClient(api_key=api_key)
+        return Geocodio(api_key=api_key)
     else:
         logger.debug("Running unit tests - using TEST_KEY with api.test hostname")
-        return GeocodioClient(api_key="TEST_KEY", hostname="api.test")
+        return Geocodio(api_key="TEST_KEY", hostname="api.test")

tests/e2e/test_api.py

@@ -5,7 +5,7 @@
 
 import os
 import pytest
-from geocodio import GeocodioClient
+from geocodio import Geocodio
 from geocodio.exceptions import GeocodioError
 
 
@@ -15,7 +15,7 @@ def client():
     api_key = os.getenv("GEOCODIO_API_KEY")
     if not api_key:
         pytest.skip("GEOCODIO_API_KEY environment variable not set")
-    return GeocodioClient(api_key)
+    return Geocodio(api_key)
 
 
 def test_integration_geocode(client):

tests/e2e/test_batch_reverse.py

@@ -3,7 +3,7 @@
 """
 
 import pytest
-from geocodio import GeocodioClient
+from geocodio import Geocodio
 
 
 def test_batch_reverse_geocoding(client):

tests/e2e/test_lists_api.py

@@ -7,7 +7,7 @@
 import pytest
 import time
 from unittest.mock import patch
-from geocodio import GeocodioClient
+from geocodio import Geocodio
 from geocodio.models import ListResponse, PaginatedResponse, ListProcessingState
 from geocodio.exceptions import GeocodioServerError
 import logging
@@ -22,7 +22,7 @@ def client():
     api_key = os.getenv("GEOCODIO_API_KEY")
     if not api_key:
         pytest.skip("GEOCODIO_API_KEY environment variable not set")
-    return GeocodioClient(api_key)
+    return Geocodio(api_key)
 
 
 @pytest.fixture
@@ -59,8 +59,8 @@ def wait_for_list_processed(client, list_id, timeout=120):
 
 def test_create_list(client):
     """
-    Test creating a list with GeocodioClient.create_list()
-    :param client: GeocodioClient instance
+    Test creating a list with Geocodio.create_list()
+    :param client: Geocodio instance
     """
     new_list = client.create_list(file="Zip\n20003\n20001", filename="test_list.csv")
 
@@ -69,8 +69,8 @@ def test_create_list(client):
 
 def test_get_list_status(client, list_response):
     """
-    Test retrieving the status of a list with GeocodioClient.get_list_status()
-    :param client: GeocodioClient instance
+    Test retrieving the status of a list with Geocodio.get_list_status()
+    :param client: Geocodio instance
     :param list_response: List object created in the fixture
     """
     # Get the status of the test list

tests/unit/test_client.py

@@ -1,30 +1,30 @@
 """
-Tests for the GeocodioClient class
+Tests for the Geocodio class
 """
 
 import pytest
 import httpx
-from geocodio import GeocodioClient
+from geocodio import Geocodio
 from geocodio.exceptions import AuthenticationError
 
 
 @pytest.fixture
 def mock_request(mocker):
     """Mock the _request method."""
-    return mocker.patch('geocodio.client.GeocodioClient._request')
+    return mocker.patch('geocodio.client.Geocodio._request')
 
 
 def test_client_initialization():
     """Test that the client can be initialized with an API key"""
-    client = GeocodioClient(api_key="test-key")
+    client = Geocodio(api_key="test-key")
     assert client.api_key == "test-key"
     assert client.hostname == "api.geocod.io"
 
 
 def test_client_initialization_with_env_var(monkeypatch):
     """Test that the client can be initialized with an environment variable"""
     monkeypatch.setenv("GEOCODIO_API_KEY", "env-key")
-    client = GeocodioClient()
+    client = Geocodio()
     assert client.api_key == "env-key"
 
 
@@ -33,7 +33,7 @@ def test_client_initialization_no_key(monkeypatch):
     # Ensure environment variable is not set
     monkeypatch.delenv("GEOCODIO_API_KEY", raising=False)
     with pytest.raises(AuthenticationError, match="No API key supplied and GEOCODIO_API_KEY is not set"):
-        GeocodioClient()
+        Geocodio()
 
 
 def test_geocode_with_census_data(mock_request):
@@ -64,7 +64,7 @@ def test_geocode_with_census_data(mock_request):
         }]
     })
 
-    client = GeocodioClient("fake-key")
+    client = Geocodio("fake-key")
     response = client.geocode(
         {"street": "1109 N Highland St", "city": "Arlington", "state": "VA"},
         fields=["census2010"]
@@ -102,7 +102,7 @@ def test_geocode_with_acs_data(mock_request):
         }]
     })
 
-    client = GeocodioClient("fake-key")
+    client = Geocodio("fake-key")
     response = client.geocode(
         {"street": "1109 N Highland St", "city": "Arlington", "state": "VA"},
         fields=["acs"]
@@ -154,7 +154,7 @@ def test_geocode_batch_with_custom_keys(mock_request):
         ]
     })
 
-    client = GeocodioClient("fake-key")
+    client = Geocodio("fake-key")
     response = client.geocode({
         "address1": "1109 N Highland St, Arlington, VA",
         "address2": "525 University Ave, Toronto, ON, Canada"
@@ -193,7 +193,7 @@ def test_geocode_with_congressional_districts(mock_request):
         }]
     })
 
-    client = GeocodioClient("fake-key")
+    client = Geocodio("fake-key")
     response = client.geocode(
         {"street": "1109 N Highland St", "city": "Arlington", "state": "VA"},
         fields=["cd"]
@@ -229,7 +229,7 @@ def test_user_agent_header_in_request(mocker):
         }]
     })
 
-    client = GeocodioClient("test-api-key")
+    client = Geocodio("test-api-key")
     client.geocode("1109 N Highland St, Arlington, VA")
 
     # Verify request was made with correct headers
@@ -247,7 +247,7 @@ def test_user_agent_header_format():
     """Test that the User-Agent header has the correct format."""
     from geocodio import __version__
 
-    client = GeocodioClient("test-api-key")
+    client = Geocodio("test-api-key")
     expected_user_agent = f"geocodio-library-python/{__version__}"
     assert client.USER_AGENT == expected_user_agent
 
@@ -262,7 +262,7 @@ def test_user_agent_header_in_batch_request(mocker):
         "results": []
     })
 
-    client = GeocodioClient("test-api-key")
+    client = Geocodio("test-api-key")
     client.geocode(["Address 1", "Address 2"])
 
     # Verify headers in batch request
@@ -295,7 +295,7 @@ def test_user_agent_header_in_reverse_geocode(mocker):
         }]
     })
 
-    client = GeocodioClient("test-api-key")
+    client = Geocodio("test-api-key")
     client.reverse("38.886665,-77.094733")
 
     # Verify headers in reverse geocode request
@@ -321,7 +321,7 @@ def test_user_agent_header_in_list_api(mocker):
         "per_page": 10
     })
 
-    client = GeocodioClient("test-api-key")
+    client = Geocodio("test-api-key")
     client.get_lists()
 
     # Verify headers in list API request

tests/unit/test_geocoding.py

@@ -7,19 +7,19 @@
 import pytest
 from dotenv import load_dotenv
 
-from geocodio import GeocodioClient
+from geocodio import Geocodio
 from geocodio.exceptions import AuthenticationError
 
 # Load environment variables from .env file
 load_dotenv()
 
 @pytest.fixture
-def client() -> GeocodioClient:
-    """Create a GeocodioClient instance for testing."""
+def client() -> Geocodio:
+    """Create a Geocodio instance for testing."""
     api_key = os.getenv("GEOCODIO_API_KEY")
     if not api_key:
         pytest.skip("GEOCODIO_API_KEY environment variable not set")
-    return GeocodioClient(api_key)
+    return Geocodio(api_key)
 
 def test_client_requires_api_key():
     """Test that client raises AuthenticationError when no API key is provided."""
@@ -30,13 +30,13 @@ def test_client_requires_api_key():
 
     try:
         with pytest.raises(AuthenticationError):
-            GeocodioClient("")
+            Geocodio("")
     finally:
         # Restore the environment variable
         if original_key:
             os.environ["GEOCODIO_API_KEY"] = original_key
 
-def test_single_forward_geocode(client: GeocodioClient):
+def test_single_forward_geocode(client: Geocodio):
     """Test forward geocoding of a single address."""
     response = client.geocode("3730 N Clark St, Chicago, IL")
 
@@ -63,7 +63,7 @@ def test_single_forward_geocode(client: GeocodioClient):
     assert components.state == "IL"
     assert components.zip == "60613"
 
-def test_batch_forward_geocode(client: GeocodioClient):
+def test_batch_forward_geocode(client: Geocodio):
     """Test forward geocoding of multiple addresses."""
     addresses: List[str] = [
         "3730 N Clark St, Chicago, IL",
@@ -85,7 +85,7 @@ def test_batch_forward_geocode(client: GeocodioClient):
     assert denver.accuracy > 0.9
     assert denver.accuracy_type == "rooftop"
 
-def test_single_reverse_geocode(client: GeocodioClient):
+def test_single_reverse_geocode(client: Geocodio):
     """Test reverse geocoding of coordinates."""
     response = client.reverse("38.9002898,-76.9990361")
 
@@ -102,7 +102,7 @@ def test_single_reverse_geocode(client: GeocodioClient):
     assert 38.89 < result.location.lat < 38.91
     assert -77.00 < result.location.lng < -76.99
 
-def test_geocode_with_fields(client: GeocodioClient):
+def test_geocode_with_fields(client: Geocodio):
     """Test geocoding with additional data fields."""
     response = client.geocode(
         "3730 N Clark St, Chicago, IL",