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
« prev ^ index » next coverage.py v7.5.1, created at 2024-05-08 14:15 +0200
1from imod.typing.grid import enforce_dim_order
4def enforced_dim_order(func):
5 """Decorator to enforce dimension order after function call"""
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)
14 return decorator