ensure list of event names is flat

CLIMADA-project · Nov 17, 2023 · c6d83bc · c6d83bc
1 parent 30464ee
commit c6d83bc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/climada/hazard/base.py b/climada/hazard/base.py
@@ -840,7 +840,7 @@ def maybe_repeat(values: np.ndarray, times: int) -> np.ndarray:
                     None,
                     np.ones(num_events),
                     np.array(range(num_events), dtype=int) + 1,
-                    list(data[coords["event"]].dt.strftime("%Y-%m-%d")),
+                    data[coords["event"]].dt.strftime("%Y-%m-%d").values.flatten().tolist(),
                     np.array(u_dt.datetime64_to_ordinal(data[coords["event"]].values)),
                 ],
                 # The accessor for the data in the Dataset

diff --git a/climada/hazard/test/test_base_xarray.py b/climada/hazard/test/test_base_xarray.py
@@ -551,6 +551,7 @@ def test_2D_time(self):
         np.testing.assert_array_equal(
             hazard.date, [val.toordinal() for val in time.flat]
         )
+        np.testing.assert_array_equal(hazard.event_name, ["1999-01-01", "1999-02-01", "2000-01-01", "2000-02-01"])
 
     def test_errors(self):
         """Check if expected errors are thrown"""