diff --git a/tests/unit/test_async_connector.py b/tests/unit/test_async_connector.py index 865aff8..d97aa0b 100644 --- a/tests/unit/test_async_connector.py +++ b/tests/unit/test_async_connector.py @@ -23,9 +23,9 @@ from mocks import FakeInstance import pytest -from google.cloud.alloydb.connector.client import AlloyDBClient from google.cloud.alloydb.connector import AsyncConnector from google.cloud.alloydb.connector import IPTypes +from google.cloud.alloydb.connector.client import AlloyDBClient from google.cloud.alloydb.connector.exceptions import IPTypeNotFoundError from google.cloud.alloydb.connector.instance import RefreshAheadCache diff --git a/tests/unit/test_connector.py b/tests/unit/test_connector.py index 5174b26..6a83f0e 100644 --- a/tests/unit/test_connector.py +++ b/tests/unit/test_connector.py @@ -23,9 +23,9 @@ from mocks import FakeInstance import pytest -from google.cloud.alloydb.connector.client import AlloyDBClient from google.cloud.alloydb.connector import Connector from google.cloud.alloydb.connector import IPTypes +from google.cloud.alloydb.connector.client import AlloyDBClient from google.cloud.alloydb.connector.exceptions import IPTypeNotFoundError from google.cloud.alloydb.connector.instance import RefreshAheadCache from google.cloud.alloydb.connector.utils import generate_keys