diff --git a/custom_components/irm_kmi/coordinator.py b/custom_components/irm_kmi/coordinator.py index a99f97b..6854a14 100644 --- a/custom_components/irm_kmi/coordinator.py +++ b/custom_components/irm_kmi/coordinator.py @@ -24,7 +24,8 @@ from .const import IRM_KMI_TO_HA_CONDITION_MAP as CDT_MAP from .const import MAP_WARNING_ID_TO_SLUG as SLUG_MAP from .const import OPTION_STYLE_SATELLITE, OUT_OF_BENELUX, STYLE_TO_PARAM_MAP from .data import (AnimationFrameData, CurrentWeatherData, IrmKmiForecast, - ProcessedCoordinatorData, RadarAnimationData, WarningData, IrmKmiRadarForecast) + IrmKmiRadarForecast, ProcessedCoordinatorData, + RadarAnimationData, WarningData) from .pollen import PollenParser from .rain_graph import RainGraph from .utils import disable_from_config, get_config_value, preferred_language diff --git a/tests/test_coordinator.py b/tests/test_coordinator.py index c82cd32..df0f746 100644 --- a/tests/test_coordinator.py +++ b/tests/test_coordinator.py @@ -10,8 +10,9 @@ from pytest_homeassistant_custom_component.common import MockConfigEntry from custom_components.irm_kmi.const import CONF_LANGUAGE_OVERRIDE from custom_components.irm_kmi.coordinator import IrmKmiCoordinator from custom_components.irm_kmi.data import (CurrentWeatherData, IrmKmiForecast, + IrmKmiRadarForecast, ProcessedCoordinatorData, - RadarAnimationData, IrmKmiRadarForecast) + RadarAnimationData) from custom_components.irm_kmi.pollen import PollenParser from tests.conftest import get_api_data diff --git a/tests/test_weather.py b/tests/test_weather.py index b4f293e..705efee 100644 --- a/tests/test_weather.py +++ b/tests/test_weather.py @@ -9,7 +9,8 @@ from homeassistant.core import HomeAssistant from pytest_homeassistant_custom_component.common import MockConfigEntry from custom_components.irm_kmi import IrmKmiCoordinator, IrmKmiWeather -from custom_components.irm_kmi.data import ProcessedCoordinatorData, IrmKmiRadarForecast +from custom_components.irm_kmi.data import (IrmKmiRadarForecast, + ProcessedCoordinatorData) from tests.conftest import get_api_data