Add missing hass type hint in component tests (e) (#124075)

This commit is contained in:
epenet
2024-08-18 15:34:59 +02:00
committed by GitHub
parent 565f271c5c
commit c8797298ea
2 changed files with 11 additions and 3 deletions

View File

@@ -1,5 +1,6 @@
"""Test the Energy sensors."""
from collections.abc import Callable, Coroutine
import copy
from datetime import timedelta
from typing import Any
@@ -37,10 +38,12 @@ TEST_TIME_ADVANCE_INTERVAL = timedelta(milliseconds=10)
@pytest.fixture
async def setup_integration(recorder_mock: Recorder):
async def setup_integration(
recorder_mock: Recorder,
) -> Callable[[HomeAssistant], Coroutine[Any, Any, None]]:
"""Set up the integration."""
async def setup_integration(hass):
async def setup_integration(hass: HomeAssistant) -> None:
assert await async_setup_component(hass, "energy", {})
await hass.async_block_till_done()