Add type hints for FrozenDateTimeFactory in test fixtures (#118690)

This commit is contained in:
epenet 2024-06-03 10:12:58 +02:00 committed by GitHub
parent 134088e1f6
commit c93d42d59b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -4,6 +4,7 @@ import copy
from datetime import timedelta from datetime import timedelta
from typing import Any from typing import Any
from freezegun.api import FrozenDateTimeFactory
import pytest import pytest
from homeassistant.components.energy import data from homeassistant.components.energy import data
@ -47,7 +48,7 @@ async def setup_integration(recorder_mock: Recorder):
@pytest.fixture(autouse=True) @pytest.fixture(autouse=True)
def frozen_time(freezer): def frozen_time(freezer: FrozenDateTimeFactory) -> FrozenDateTimeFactory:
"""Freeze clock for tests.""" """Freeze clock for tests."""
freezer.move_to("2022-04-19 07:53:05") freezer.move_to("2022-04-19 07:53:05")
return freezer return freezer