diff --git a/irm_kmi_api/api.py b/irm_kmi_api/api.py index 416a540..1a082e6 100644 --- a/irm_kmi_api/api.py +++ b/irm_kmi_api/api.py @@ -14,10 +14,11 @@ from zoneinfo import ZoneInfo import aiohttp import async_timeout -from .const import WEEKDAYS, STYLE_TO_PARAM_MAP, OPTION_STYLE_SATELLITE, \ - MAP_WARNING_ID_TO_SLUG as SLUG_MAP -from .data import CurrentWeatherData, IrmKmiForecast, Forecast, \ - IrmKmiRadarForecast, RadarAnimationData, AnimationFrameData, WarningData +from .const import MAP_WARNING_ID_TO_SLUG as SLUG_MAP +from .const import OPTION_STYLE_SATELLITE, STYLE_TO_PARAM_MAP, WEEKDAYS +from .data import (AnimationFrameData, CurrentWeatherData, Forecast, + IrmKmiForecast, IrmKmiRadarForecast, RadarAnimationData, + WarningData) from .pollen import PollenParser from .utils import next_weekday diff --git a/irm_kmi_api/data.py b/irm_kmi_api/data.py index 270bd52..4517b7a 100644 --- a/irm_kmi_api/data.py +++ b/irm_kmi_api/data.py @@ -1,6 +1,6 @@ """Data classes for IRM KMI integration""" from datetime import datetime -from typing import TypedDict, Required, List +from typing import List, Required, TypedDict class Forecast(TypedDict, total=False): diff --git a/irm_kmi_api/pollen.py b/irm_kmi_api/pollen.py index a3f1f09..601bf35 100644 --- a/irm_kmi_api/pollen.py +++ b/irm_kmi_api/pollen.py @@ -3,7 +3,7 @@ import logging import xml.etree.ElementTree as ET from typing import List -from .const import POLLEN_NAMES, POLLEN_LEVEL_TO_COLOR +from .const import POLLEN_LEVEL_TO_COLOR, POLLEN_NAMES _LOGGER = logging.getLogger(__name__) diff --git a/irm_kmi_api/rain_graph.py b/irm_kmi_api/rain_graph.py index 10b31d6..a83bab8 100644 --- a/irm_kmi_api/rain_graph.py +++ b/irm_kmi_api/rain_graph.py @@ -4,7 +4,7 @@ import base64 import copy import datetime import logging -from typing import List, Self, Any +from typing import Any, List, Self import async_timeout from svgwrite import Drawing diff --git a/tests/test_coordinator.py b/tests/test_coordinator.py index 29daaf6..7b3b5c2 100644 --- a/tests/test_coordinator.py +++ b/tests/test_coordinator.py @@ -3,12 +3,11 @@ from zoneinfo import ZoneInfo from freezegun import freeze_time -from irm_kmi_api.data import CurrentWeatherData, IrmKmiRadarForecast, Forecast, IrmKmiForecast +from irm_kmi_api.data import (CurrentWeatherData, Forecast, IrmKmiForecast, + IrmKmiRadarForecast) from tests.conftest import get_api_with_data -from tests.const import ATTR_CONDITION_PARTLYCLOUDY, ATTR_CONDITION_CLOUDY, ATTR_CONDITION_RAINY - -pytest_plugins = ('pytest_asyncio',) - +from tests.const import (ATTR_CONDITION_CLOUDY, ATTR_CONDITION_PARTLYCLOUDY, + ATTR_CONDITION_RAINY) @freeze_time(datetime.fromisoformat('2024-01-12T07:10:00+00:00')) diff --git a/tests/test_rain_graph.py b/tests/test_rain_graph.py index 09510c1..8eb8195 100644 --- a/tests/test_rain_graph.py +++ b/tests/test_rain_graph.py @@ -1,6 +1,7 @@ import base64 import datetime -from datetime import datetime as dt, timedelta +from datetime import datetime as dt +from datetime import timedelta from irm_kmi_api.data import AnimationFrameData, RadarAnimationData from irm_kmi_api.rain_graph import RainGraph