diff --git a/tests/components/climate/test_init.py b/tests/components/climate/test_init.py index 4756c265aea..ced75ff7ef7 100644 --- a/tests/components/climate/test_init.py +++ b/tests/components/climate/test_init.py @@ -158,7 +158,7 @@ async def test_sync_turn_off(hass: HomeAssistant) -> None: assert climate.turn_off.called -def _create_tuples(enum: Enum, constant_prefix: str) -> list[tuple[Enum, str]]: +def _create_tuples(enum: type[Enum], constant_prefix: str) -> list[tuple[Enum, str]]: return [ (enum_field, constant_prefix) for enum_field in enum diff --git a/tests/components/cover/test_init.py b/tests/components/cover/test_init.py index 7da6c6efe21..37740260c2f 100644 --- a/tests/components/cover/test_init.py +++ b/tests/components/cover/test_init.py @@ -156,7 +156,7 @@ def is_closing(hass, ent): return hass.states.is_state(ent.entity_id, STATE_CLOSING) -def _create_tuples(enum: Enum, constant_prefix: str) -> list[tuple[Enum, str]]: +def _create_tuples(enum: type[Enum], constant_prefix: str) -> list[tuple[Enum, str]]: return [(enum_field, constant_prefix) for enum_field in enum] diff --git a/tests/components/humidifier/test_init.py b/tests/components/humidifier/test_init.py index b90e7084dd1..b31750a3a3b 100644 --- a/tests/components/humidifier/test_init.py +++ b/tests/components/humidifier/test_init.py @@ -48,7 +48,7 @@ async def test_sync_turn_off(hass: HomeAssistant) -> None: assert humidifier.turn_off.called -def _create_tuples(enum: Enum, constant_prefix: str) -> list[tuple[Enum, str]]: +def _create_tuples(enum: type[Enum], constant_prefix: str) -> list[tuple[Enum, str]]: return [(enum_field, constant_prefix) for enum_field in enum] diff --git a/tests/components/weather/__init__.py b/tests/components/weather/__init__.py index c24baad5237..2dbffbbd617 100644 --- a/tests/components/weather/__init__.py +++ b/tests/components/weather/__init__.py @@ -61,7 +61,7 @@ class MockWeatherTest(WeatherPlatform.MockWeather): async def create_entity( hass: HomeAssistant, - mock_weather: WeatherPlatform.MockWeather, + mock_weather: type[WeatherPlatform.MockWeather], manifest_extra: dict[str, Any] | None, **kwargs, ) -> WeatherPlatform.MockWeather: diff --git a/tests/test_const.py b/tests/test_const.py index a6a2387b091..64ccb875cf5 100644 --- a/tests/test_const.py +++ b/tests/test_const.py @@ -15,7 +15,7 @@ from .common import ( def _create_tuples( - value: Enum | list[Enum], constant_prefix: str + value: type[Enum] | list[Enum], constant_prefix: str ) -> list[tuple[Enum, str]]: return [(enum, constant_prefix) for enum in value]