Fix missing type[..] annotation in tests (#122217)

This commit is contained in:
Marc Mueller 2024-07-20 11:11:16 +02:00 committed by GitHub
parent f8c4ffc060
commit f0b9a806d1
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 5 additions and 5 deletions

View file

@ -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

View file

@ -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]

View file

@ -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]

View file

@ -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:

View file

@ -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]