Coverage for C:\src\imod-python\imod\typing\__init__.py: 100%

11 statements  

« prev     ^ index     » next       coverage.py v7.4.4, created at 2024-04-08 10:26 +0200

1""" 

2Module to define type aliases. 

3""" 

4 

5from typing import TypeAlias, Union 

6 

7import numpy as np 

8import xarray as xr 

9import xugrid as xu 

10 

11GridDataArray: TypeAlias = Union[xr.DataArray, xu.UgridDataArray] 

12GridDataset: TypeAlias = Union[xr.Dataset, xu.UgridDataset] 

13ScalarAsDataArray: TypeAlias = Union[xr.DataArray, xu.UgridDataArray] 

14ScalarAsDataset: TypeAlias = Union[xr.Dataset, xu.UgridDataset] 

15UnstructuredData: TypeAlias = Union[xu.UgridDataset, xu.UgridDataArray] 

16FloatArray: TypeAlias = np.ndarray 

17IntArray: TypeAlias = np.ndarray