From c93d42d59b477c807ec6a21867d66d63bc59b157 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Mon, 3 Jun 2024 10:12:58 +0200 Subject: [PATCH] Add type hints for FrozenDateTimeFactory in test fixtures (#118690) --- tests/components/energy/test_sensor.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/components/energy/test_sensor.py b/tests/components/energy/test_sensor.py index 4128a80c587..b9aca285829 100644 --- a/tests/components/energy/test_sensor.py +++ b/tests/components/energy/test_sensor.py @@ -4,6 +4,7 @@ import copy from datetime import timedelta from typing import Any +from freezegun.api import FrozenDateTimeFactory import pytest from homeassistant.components.energy import data @@ -47,7 +48,7 @@ async def setup_integration(recorder_mock: Recorder): @pytest.fixture(autouse=True) -def frozen_time(freezer): +def frozen_time(freezer: FrozenDateTimeFactory) -> FrozenDateTimeFactory: """Freeze clock for tests.""" freezer.move_to("2022-04-19 07:53:05") return freezer