Skip to content
Snippets Groups Projects
Commit 82ba326f authored by Felix Kleinert's avatar Felix Kleinert
Browse files

Update tests

parent f6d93a00
No related branches found
No related tags found
1 merge request!231Draft: Resolve "Create WRF-Chem data handler"
Pipeline #60248 passed
...@@ -70,8 +70,7 @@ class TestHaversineDist: ...@@ -70,8 +70,7 @@ class TestHaversineDist:
np.array([0., 0., 0.]), True, np.array([0., 0., 0.]), True,
np.array([[np.pi, np.pi/2., np.pi/4.], np.array([[np.pi, np.pi/2., np.pi/4.],
[np.pi/2., 0., np.pi/4.], [np.pi/2., 0., np.pi/4.],
[3./4.*np.pi, np.pi/4., 0.]]) [3./4.*np.pi, np.pi/4., 0.]])),
)
)) ))
def test_haversine_dist_on_unit_sphere_fields_and_scalars(self, lat1, lon1, lat2, lon2, to_radians, expected_dist): def test_haversine_dist_on_unit_sphere_fields_and_scalars(self, lat1, lon1, lat2, lon2, to_radians, expected_dist):
dist = haversine_dist(lat1=lat1, lon1=lon1, lat2=lat2, lon2=lon2, to_radians=to_radians, earth_radius=1.) dist = haversine_dist(lat1=lat1, lon1=lon1, lat2=lat2, lon2=lon2, to_radians=to_radians, earth_radius=1.)
......
import numpy as np import numpy as np
import xarray as xr import xarray as xr
import dask.array as da
import datetime as dt import datetime as dt
import logging import logging
...@@ -352,6 +353,10 @@ class TestConvert2xrDa: ...@@ -352,6 +353,10 @@ class TestConvert2xrDa:
def custom_xr_dataset(self, custom_xr_dataarray): def custom_xr_dataset(self, custom_xr_dataarray):
return xr.Dataset({'test_1': custom_xr_dataarray}) return xr.Dataset({'test_1': custom_xr_dataarray})
@pytest.fixture
def custom_1d_daarray(self, custom_1d_npdata):
return da.array(custom_1d_npdata)
def test_convert2xrda_xrdata_in(self, custom_xr_dataarray, custom_xr_dataset): def test_convert2xrda_xrdata_in(self, custom_xr_dataarray, custom_xr_dataset):
assert (convert2xrda(custom_xr_dataarray) == custom_xr_dataarray).all() assert (convert2xrda(custom_xr_dataarray) == custom_xr_dataarray).all()
assert (convert2xrda(custom_xr_dataset) == custom_xr_dataset).all() assert (convert2xrda(custom_xr_dataset) == custom_xr_dataset).all()
...@@ -416,3 +421,10 @@ class TestConvert2xrDa: ...@@ -416,3 +421,10 @@ class TestConvert2xrDa:
converted_data = convert2xrda(wrong_input, use_1d_default=True) converted_data = convert2xrda(wrong_input, use_1d_default=True)
assert f"`arr' must be arry-like, int or float. But is of type {type(wrong_input)}" in e.value.args[0] assert f"`arr' must be arry-like, int or float. But is of type {type(wrong_input)}" in e.value.args[0]
def test_convert2xrda_dask_in_default_true_nokwargs(self, custom_1d_daarray):
with pytest.raises(TypeError) as e:
convert2xrda(custom_1d_daarray, True)
assert "`use_1d_default=True' is used with `arr' of type da.array. For da.arrays please pass `use_1d_default=False' and specify keywords for xr.DataArray via kwargs." in \
e.value.args[0]
assert "`use_1d_default=True' is used with `arr' of type da.array. For da.arrays please pass" + \
" `use_1d_default=False' and specify keywords for xr.DataArray via kwargs." in e.value.args[0]
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment