Coverage for C:\src\imod-python\imod\mf6\utilities\package.py: 100%
7 statements
« prev ^ index » next coverage.py v7.4.4, created at 2024-04-08 13:27 +0200
« prev ^ index » next coverage.py v7.4.4, created at 2024-04-08 13:27 +0200
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 = [
17 imod.util.time.to_datetime_internal(key, use_cftime=False)
18 for key in times.keys()
19 ]
20 values = [
21 imod.util.time.to_datetime_internal(value, use_cftime=False)
22 for value in times.values()
23 ]
24 return xr.DataArray(
25 data=np.column_stack((keys, values)),
26 dims=("repeat", "repeat_items"),
27 )