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
« prev ^ index » next coverage.py v7.5.1, created at 2024-05-16 11:25 +0200
1from functools import wraps
3from imod.typing.grid import enforce_dim_order
6def enforced_dim_order(func):
7 """Decorator to enforce dimension order after function call"""
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)
17 return decorator