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

8 statements  

« prev     ^ index     » next       coverage.py v7.5.1, created at 2024-05-08 14:15 +0200

1from imod.typing.grid import enforce_dim_order 

2 

3 

4def enforced_dim_order(func): 

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

6 

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

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

9 # Multiple grids returned 

10 if isinstance(x, tuple): 

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

12 return enforce_dim_order(x) 

13 

14 return decorator