Merge pull request #2 from jdejaegh/heat

Add support for heat warning

irm_kmi_api/const.py

@@ -27,6 +27,7 @@ MAP_WARNING_ID_TO_SLUG: Final = {
     3: WarningType.THUNDER,
     7: WarningType.FOG,
     9: WarningType.COLD,
+    10: WarningType.HEAT,
     12: WarningType.THUNDER_WIND_RAIN,
     13: WarningType.THUNDERSTORM_STRONG_GUSTS,
     14: WarningType.THUNDERSTORM_LARGE_RAINFALL,

irm_kmi_api/data.py

@@ -80,6 +80,7 @@ class WarningType(StrEnum):
     COLD = 'cold'
     COLDSPELL = 'coldspell'
     FOG = 'fog'
+    HEAT = 'heat'
     ICE_OR_SNOW = 'ice_or_snow'
     RAIN = 'rain'
     STORM_SURGE = 'storm_surge'

tests/test_warning.py

@@ -24,6 +24,21 @@ async def test_warning_data() -> None:
     assert first.get('id') == 7
     assert first.get('level') == 1
 
+async def test_warning_heat() -> None:
+    api = get_api_with_data("antwerp_with_heat_warning.json")
+
+    result = api.get_warnings(lang='en')
+
+    assert isinstance(result, list)
+    assert len(result) == 1
+
+    first = result[0]
+
+    assert first.get('slug') == WarningType.HEAT
+    assert first.get('friendly_name') == 'Heat'
+    assert first.get('id') == 10
+    assert first.get('level') == 1
+
 
 async def test_warning_data_is_serializable() -> None:
     api = get_api_with_data("be_forecast_warning.json")