Coverage for C:\src\imod-python\imod\util\dims.py: 100%

10 statements  

« prev     ^ index     » next       coverage.py v7.5.1, created at 2024-05-16 11:25 +0200

1from functools import wraps 

2 

3from imod.typing.grid import enforce_dim_order 

4 

5 

6def enforced_dim_order(func): 

7 """Decorator to enforce dimension order after function call""" 

8 

9 @wraps(func) 

10 def decorator(*args, **kwargs): 

11 x = func(*args, **kwargs) 

12 # Multiple grids returned 

13 if isinstance(x, tuple): 

14 return tuple(enforce_dim_order(i) for i in x) 

15 return enforce_dim_order(x) 

16 

17 return decorator