docstrings and datetime tests

Author: Evelyn-M

climada/util/forecast.py

@@ -23,13 +23,34 @@
 
 
 class Forecast:
+    """Mixin class for forecast data.
+
+    Attributes
+    ----------
+    lead_time : np.ndarray
+        Array of forecast lead times, given as datetime64 objects.
+        Represents the time points for which forecasts are made.
+    member : np.ndarray
+        Array of ensemble member identifiers, given as integers.
+        Represents different forecast ensemble members.
+    """
+
     def __init__(
         self,
         lead_time: np.ndarray | None = None,
         member: np.ndarray | None = None,
         *args,
         **kwargs,
     ):
+        """Initialize Forecast.
+
+        Parameters
+        ----------
+        lead_time : np.ndarray or None, optional
+            Forecast lead times. Default is empty array.
+        member : np.ndarray or None, optional
+            Ensemble member identifiers. Default is empty array.
+        """
 
         self.lead_time = (
             np.asarray(lead_time) if lead_time is not None else np.array([])

climada/util/test/test_forecast.py

@@ -42,3 +42,12 @@ def test_forecast_init():
     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)
+
+    # Test with datetime64 including seconds
+    lead_times_seconds = np.array(
+        ["2024-01-01T00:00:00", "2024-01-01T00:01:00", "2024-01-01"],
+        dtype="datetime64[s]",
+    )
+    forecast = Forecast(lead_time=lead_times_seconds, member=[1, 2, 3])
+    npt.assert_array_equal(forecast.lead_time, lead_times_seconds, strict=True)
+    assert forecast.lead_time.dtype == np.dtype("datetime64[s]")