add init test

Author: ValentinGebhart

climada/util/forecast.py

@@ -23,15 +23,18 @@
 
 
 class Forecast:
-    def __init__(self, lead_time=None, member=None, *args, **kwargs):
-        if lead_time is None:
-            self.lead_time = np.array([])
-        else:
-            self.lead_time = np.array(lead_time)
-
-        if member is None:
-            self.member = np.array([])
-        else:
-            self.member = member
+    def __init__(
+        self,
+        lead_time: np.ndarray | None = None,
+        member: np.ndarray | None = None,
+        *args,
+        **kwargs,
+    ):
+
+        self.lead_time = (
+            np.asarray(lead_time) if lead_time is not None else np.array([])
+        )
+
+        self.member = np.asarray(member) if member is not None else np.array([])
 
         super().__init__(*args, **kwargs)

climada/util/test/test_forecast.py

@@ -18,3 +18,27 @@
 
 Tests for Forecast base class.
 """
+
+import numpy as np
+import numpy.testing as npt
+import pytest
+
+from climada.util.forecast import Forecast
+
+
+def test_forecast_init():
+    """Test initialization of Forecast class."""
+    forecast = Forecast()
+    npt.assert_array_equal(forecast.lead_time, np.array([]))
+    npt.assert_array_equal(forecast.member, np.array([]))
+
+    forecast = Forecast(member=np.array([1, 2]))
+    npt.assert_array_equal(forecast.member, np.array([1, 2]), strict=True)
+
+    forecast = Forecast(lead_time=np.array([1, 2]))
+    npt.assert_array_equal(forecast.lead_time, np.array([1, 2]), strict=True)
+
+    forecast = Forecast(lead_time=np.array([1, 2]), member=[3, 4])
+    npt.assert_array_equal(forecast.lead_time, np.array([1, 2]), strict=True)
+    npt.assert_array_equal(forecast.member, np.array([3, 4]), strict=True)
+    assert isinstance(forecast.member, np.ndarray)