Merge pull request #371 from killer0071234/forecast-symbols-v1

Forecast symbols v1

Author: killer0071234

package.json

@@ -1,6 +1,6 @@
 {
   "name": "zamg",
-  "version": "0.4.0",
+  "version": "0.4.1",
   "private": true,
   "description": "Asynchronous Python client for GeoSphere Austria (ZAMG) weather data.",
   "scripts": {

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "zamg"
-version = "0.4.0"
+version = "0.4.1"
 description = "Asynchronous Python client for GeoSphere Austria (ZAMG) weather data."
 authors = ["Daniel Gangl <killer007@gmx.at>"]
 maintainers = ["Daniel Gangl <killer007@gmx.at>"]

src/zamg/__init__.py

@@ -1,6 +1,6 @@
 """Asynchronous Python client for GeoSphere Austria weather data."""
 
-__version__ = "0.3.6"
+__version__ = "0.4.1"
 
 from .exceptions import (
     ZamgApiError,

src/zamg/zamg.py

@@ -77,7 +77,15 @@ def __init__(
     def get_forecast_current(
         self,
         forecast_data: dict | None = None,
-        parameters: tuple[str, ...] = ("t2m", "rh2m", "u10m", "v10m", "tcc", "rr_acc"),
+        parameters: tuple[str, ...] = (
+            "t2m",
+            "rh2m",
+            "u10m",
+            "v10m",
+            "tcc",
+            "sy",
+            "rr_acc",
+        ),
     ) -> dict:
         """Return selected forecast parameters for the current/next timestamp.
 
@@ -472,7 +480,7 @@ async def get_forecast(
 
             async with async_timeout.timeout(self.request_timeout):
                 forecast_params = (
-                    self.forecast_parameters or "t2m,rr_acc,u10m,v10m,tcc,rh2m"
+                    self.forecast_parameters or "t2m,rr_acc,u10m,v10m,tcc,sy,rh2m"
                 )
                 if lat_lon is None:
                     station_lat, station_lon = self.get_station_location

tests/test_zamg.py

@@ -359,6 +359,7 @@ async def test_get_forecast_uses_station_location(aresponses) -> None:
                                 "u10m": {"data": [0.0]},
                                 "v10m": {"data": [0.0]},
                                 "tcc": {"data": [0.0]},
+                                "sy": {"data": [0.0]},
                                 "rr_acc": {"data": [0.0]},
                             }
                         }
@@ -424,6 +425,7 @@ def test_get_forecast_current() -> None:
                         "u10m": {"data": [1.0, 2.0, 3.0]},
                         "v10m": {"data": [4.0, 5.0, 6.0]},
                         "tcc": {"data": [0.1, 0.2, 0.3]},
+                        "sy": {"data": [1.0, 2.0, 3.0]},
                         "rr_acc": {"data": [0.5, 0.9, 1.4]},
                     }
                 }
@@ -439,6 +441,8 @@ def test_get_forecast_current() -> None:
     assert result["u10m"] == 2.0
     assert result["v10m"] == 5.0
     assert result["rain"] == 0.4
+    assert result["tcc"] == 0.2
+    assert result["sy"] == 2.0
 
 
 @pytest.mark.asyncio
@@ -467,6 +471,7 @@ async def test_get_forecast_trims_past_data() -> None:
                         "u10m": {"data": [1.0, 2.0, 3.0]},
                         "v10m": {"data": [4.0, 5.0, 6.0]},
                         "tcc": {"data": [0.1, 0.2, 0.3]},
+                        "sy": {"data": [1.0, 2.0, 3.0]},
                         "rr_acc": {"data": [0.5, 0.9, 1.4]},
                     }
                 }
@@ -490,6 +495,11 @@ async def test_get_forecast_trims_past_data() -> None:
         19.4,
         24.1,
     ]
+    assert result["features"][0]["properties"]["parameters"]["tcc"]["data"] == [
+        0.2,
+        0.3,
+    ]
+    assert result["features"][0]["properties"]["parameters"]["sy"]["data"] == [2.0, 3.0]
 
 
 @pytest.fixture