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

1import numpy as np 

2import xarray as xr 

3 

4import imod 

5 

6 

7def get_repeat_stress(times) -> xr.DataArray: 

8 """ 

9 Set repeat stresses: re-use data of earlier periods. 

10 

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 )