Coverage for C:\src\imod-python\imod\mf6\utilities\package_utils.py: 100%
7 statements
« prev ^ index » next coverage.py v7.4.3, created at 2024-03-01 15:36 +0100
« prev ^ index » next coverage.py v7.4.3, created at 2024-03-01 15:36 +0100
1import numpy as np
2import xarray as xr
4import imod
7def get_repeat_stress(times) -> xr.DataArray:
8 """
9 Set repeat stresses: re-use data of earlier periods.
11 Parameters
12 ----------
13 times: Dict of datetime-like to datetime-like.
14 The data of the value datetime is used for the key datetime.
15 """
16 keys = [imod.util.time.to_datetime_internal(key, use_cftime=False) for key in times.keys()]
17 values = [
18 imod.util.time.to_datetime_internal(value, use_cftime=False)
19 for value in times.values()
20 ]
21 return xr.DataArray(
22 data=np.column_stack((keys, values)),
23 dims=("repeat", "repeat_items"),
24 )