diff --git a/tests/test_source_azure_key_vault.py b/tests/test_source_azure_key_vault.py index b8badbb..bd4ddaa 100644 --- a/tests/test_source_azure_key_vault.py +++ b/tests/test_source_azure_key_vault.py @@ -2,12 +2,11 @@ Test pydantic_settings.AzureKeyVaultSettingsSource. """ -from __future__ import annotations - -from typing import TYPE_CHECKING +from typing import Tuple, Type import pytest from pydantic import BaseModel, Field +from pytest_mock import MockerFixture from pydantic_settings import ( AzureKeyVaultSettingsSource, @@ -25,9 +24,6 @@ except ImportError: azure_key_vault = False -if TYPE_CHECKING: - from pytest_mock import MockerFixture - MODULE = 'pydantic_settings.sources' @@ -93,12 +89,12 @@ class AzureKeyVaultSettings(BaseSettings): @classmethod def settings_customise_sources( cls, - settings_cls: type[BaseSettings], + settings_cls: Type[BaseSettings], init_settings: PydanticBaseSettingsSource, env_settings: PydanticBaseSettingsSource, dotenv_settings: PydanticBaseSettingsSource, file_secret_settings: PydanticBaseSettingsSource, - ) -> tuple[PydanticBaseSettingsSource, ...]: + ) -> Tuple[PydanticBaseSettingsSource, ...]: return ( AzureKeyVaultSettingsSource( settings_cls, 'https://my-resource.vault.azure.net/', DefaultAzureCredential() diff --git a/tests/test_source_pyproject_toml.py b/tests/test_source_pyproject_toml.py index 6c7a25a..5468d5b 100644 --- a/tests/test_source_pyproject_toml.py +++ b/tests/test_source_pyproject_toml.py @@ -2,13 +2,13 @@ Test pydantic_settings.PyprojectTomlConfigSettingsSource. """ -from __future__ import annotations - import sys -from typing import TYPE_CHECKING, Optional, Tuple, Type +from pathlib import Path +from typing import Optional, Tuple, Type import pytest from pydantic import BaseModel +from pytest_mock import MockerFixture from pydantic_settings import ( BaseSettings, @@ -22,11 +22,6 @@ except ImportError: tomli = None -if TYPE_CHECKING: - from pathlib import Path - - from pytest_mock import MockerFixture - MODULE = 'pydantic_settings.sources'