def data_allclose_or_equiv(arr1, arr2, rtol=1e-05, atol=1e-08):
if any(arr.dtype.kind == 'S' for arr in [arr1, arr2]):
arr1 = decode_string_data(arr1)
arr2 = decode_string_data(arr2)
exact_dtypes = ['M', 'm', 'O', 'U']
if any(arr.dtype.kind in exact_dtypes for arr in [arr1, arr2]):
return ops.array_equiv(arr1, arr2)
else:
return ops.allclose_or_equiv(arr1, arr2, rtol=rtol, atol=atol)
def data_allclose_or_equiv(arr1, arr2, rtol=1e-05, atol=1e-08):
if any(arr.dtype.kind == 'S' for arr in [arr1, arr2]):
arr1 = decode_string_data(arr1)
arr2 = decode_string_data(arr2)
exact_dtypes = ['M', 'm', 'O', 'U']
if any(arr.dtype.kind in exact_dtypes for arr in [arr1, arr2]):
return ops.array_equiv(arr1, arr2)
else:
return ops.allclose_or_equiv(arr1, arr2, rtol=rtol, atol=atol)
def data_allclose_or_equiv(arr1, arr2, rtol=1e-05, atol=1e-08):
from xarray.core import ops
if any(arr.dtype.kind == "S" for arr in [arr1, arr2]):
arr1 = decode_string_data(arr1)
arr2 = decode_string_data(arr2)
exact_dtypes = ["M", "m", "O", "U"]
if any(arr.dtype.kind in exact_dtypes for arr in [arr1, arr2]):
return ops.array_equiv(arr1, arr2)
else:
return ops.allclose_or_equiv(arr1, arr2, rtol=rtol, atol=atol)
def assertAllClose(self, a1, a2, rtol=1e-05, atol=1e-8):
assert allclose_or_equiv(a1, a2, rtol=rtol, atol=atol)
