7
7
import pytest
8
8
from numpy import array , nan
9
9
10
- from xarray import DataArray , Dataset , cftime_range , concat
10
+ from xarray import DataArray , Dataset , cftime_range , concat , testing
11
11
from xarray .core import dtypes , duck_array_ops
12
12
from xarray .core .duck_array_ops import (
13
13
array_notnull_equiv ,
@@ -902,7 +902,6 @@ def test_push_dask():
902
902
def test_isin_tolerance (shape , tolerance , dask_for_A , dask_for_B ):
903
903
if (dask_for_A or dask_for_B ) and not has_dask :
904
904
pytest .skip ("requires dask" )
905
- from xarray .testing import assert_duckarray_equal
906
905
907
906
in_margin = tolerance / 2 # measure within acceptable margin
908
907
arrayA = np .arange (- 10.0 , 10.0 , 0.1 ).reshape (shape )
@@ -926,7 +925,7 @@ def test_isin_tolerance(shape, tolerance, dask_for_A, dask_for_B):
926
925
# test function
927
926
actual = isin_tolerance (arrayA , arrayB , tolerance )
928
927
929
- assert_duckarray_equal (actual , expected )
928
+ testing . assert_duckarray_equal (actual , expected )
930
929
if dask_for_A :
931
930
assert isinstance (actual , dask_array_type )
932
931
else :
@@ -938,5 +937,5 @@ def test_isin_tolerance(shape, tolerance, dask_for_A, dask_for_B):
938
937
# test function
939
938
actual = isin_tolerance (arrayA , arrayB , tolerance )
940
939
941
- assert_duckarray_equal (actual , expected )
940
+ testing . assert_duckarray_equal (actual , expected )
942
941
assert isinstance (actual , np .ndarray )
0 commit comments