evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Improve type hints in tod tests (#123917)

Browse source
This commit is contained in:
epenet 2024-08-14 15:03:38 +02:00 • committed by GitHub
parent 7f6bf95aa6
commit 1e5762fbf7
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/components/tod/test_binary_sensor.py
View file

@ -1,6 +1,6 @@
"""Test Times of the Day Binary Sensor.""" """Test Times of the Day Binary Sensor."""
from datetime import datetime, timedelta from datetime import datetime, timedelta, tzinfo
from freezegun.api import FrozenDateTimeFactory from freezegun.api import FrozenDateTimeFactory
import pytest import pytest
@ -16,13 +16,13 @@ from tests.common import assert_setup_component, async_fire_time_changed
@pytest.fixture @pytest.fixture
def hass_time_zone(): def hass_time_zone() -> str:
"""Return default hass timezone.""" """Return default hass timezone."""
return "US/Pacific" return "US/Pacific"
@pytest.fixture(autouse=True) @pytest.fixture(autouse=True)
async def setup_fixture(hass, hass_time_zone): async def setup_fixture(hass: HomeAssistant, hass_time_zone: str) -> None:
"""Set up things to be run when tests are started.""" """Set up things to be run when tests are started."""
hass.config.latitude = 50.27583 hass.config.latitude = 50.27583
hass.config.longitude = 18.98583 hass.config.longitude = 18.98583
@ -30,7 +30,7 @@ async def setup_fixture(hass, hass_time_zone):
@pytest.fixture @pytest.fixture
def hass_tz_info(hass): def hass_tz_info(hass: HomeAssistant) -> tzinfo | None:
"""Return timezone info for the hass timezone.""" """Return timezone info for the hass timezone."""
return dt_util.get_time_zone(hass.config.time_zone) return dt_util.get_time_zone(hass.config.time_zone)