Sort imports

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, \
+from .const import MAP_WARNING_ID_TO_SLUG as SLUG_MAP
-    MAP_WARNING_ID_TO_SLUG as SLUG_MAP
+from .const import OPTION_STYLE_SATELLITE, STYLE_TO_PARAM_MAP, WEEKDAYS
-from .data import CurrentWeatherData, IrmKmiForecast, Forecast, \
+from .data import (AnimationFrameData, CurrentWeatherData, Forecast,
-    IrmKmiRadarForecast, RadarAnimationData, AnimationFrameData, WarningData
+                   IrmKmiForecast, IrmKmiRadarForecast, RadarAnimationData,
+                   WarningData)
 from .pollen import PollenParser
 from .utils import next_weekday
 

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):

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__)
 

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

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
+from tests.const import (ATTR_CONDITION_CLOUDY, ATTR_CONDITION_PARTLYCLOUDY,
-
+                         ATTR_CONDITION_RAINY)
-pytest_plugins = ('pytest_asyncio',)
-
 
 
 @freeze_time(datetime.fromisoformat('2024-01-12T07:10:00+00:00'))

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