Coverage for tests/utils_test_data.py: 100%

18 statements  

« prev     ^ index     » next       coverage.py v7.3.2, created at 2024-02-28 12:51 +1100

1""" 

2Test data for scores.utils 

3""" 

4import numpy as np 

5import xarray as xr 

6 

7""" 

8Test data for scores.utils.check_dims 

9""" 

10DA_R = xr.DataArray(np.array(1).reshape((1,)), dims=["red"]) 

11DA_G = xr.DataArray(np.array(1).reshape((1,)), dims=["green"]) 

12DA_B = xr.DataArray(np.array(1).reshape((1,)), dims=["blue"]) 

13 

14DA_RG = xr.DataArray(np.array(1).reshape((1, 1)), dims=["red", "green"]) 

15DA_GB = xr.DataArray(np.array(1).reshape((1, 1)), dims=["green", "blue"]) 

16 

17DA_RGB = xr.DataArray(np.array(1).reshape((1, 1, 1)), dims=["red", "green", "blue"]) 

18 

19DS_R = xr.Dataset({"DA_R": DA_R}) 

20DS_R_2 = xr.Dataset({"DA_R": DA_R, "DA_R_2": DA_R}) 

21DS_G = xr.Dataset({"DA_G": DA_G}) 

22 

23DS_RG = xr.Dataset({"DA_RG": DA_RG}) 

24DS_GB = xr.Dataset({"DA_GB": DA_GB}) 

25 

26DS_RG_RG = xr.Dataset({"DA_RG": DA_RG, "DA_RG_2": DA_RG}) 

27DS_RG_R = xr.Dataset({"DA_RG": DA_RG, "DA_R": DA_R}) 

28 

29DS_RGB_GB = xr.Dataset({"DA_RGB": DA_RGB, "DA_GB": DA_GB}) 

30 

31DS_RGB = xr.Dataset({"DA_RGB": DA_RGB})