示例#1
0
文件: testing.py 项目: nbren12/xarray
def _data_allclose_or_equiv(arr1, arr2, rtol=1e-05, atol=1e-08, decode_bytes=True):
    if any(arr.dtype.kind == "S" for arr in [arr1, arr2]) and decode_bytes:
        arr1 = _decode_string_data(arr1)
        arr2 = _decode_string_data(arr2)
    exact_dtypes = ["M", "m", "O", "S", "U"]
    if any(arr.dtype.kind in exact_dtypes for arr in [arr1, arr2]):
        return duck_array_ops.array_equiv(arr1, arr2)
    else:
        return duck_array_ops.allclose_or_equiv(arr1, arr2, rtol=rtol, atol=atol)
示例#2
0
def _data_allclose_or_equiv(arr1, arr2, rtol=1e-05, atol=1e-08,
                            decode_bytes=True):
    if any(arr.dtype.kind == 'S' for arr in [arr1, arr2]) and decode_bytes:
        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 duck_array_ops.array_equiv(arr1, arr2)
    else:
        return duck_array_ops.allclose_or_equiv(
            arr1, arr2, rtol=rtol, atol=atol)
示例#3
0
def assert_duckarray_allclose(actual,
                              desired,
                              rtol=1e-07,
                              atol=0,
                              err_msg="",
                              verbose=True):
    """Like `np.testing.assert_allclose`, but for duckarrays."""
    __tracebackhide__ = True

    allclose = duck_array_ops.allclose_or_equiv(actual,
                                                desired,
                                                rtol=rtol,
                                                atol=atol)
    assert allclose, _format_message(actual,
                                     desired,
                                     err_msg=err_msg,
                                     verbose=verbose)
示例#4
0
 def assertAllClose(self, a1, a2, rtol=1e-05, atol=1e-8):
     assert allclose_or_equiv(a1, a2, rtol=rtol, atol=atol)
示例#5
0
 def assertAllClose(self, a1, a2, rtol=1e-05, atol=1e-8):
     __tracebackhide__ = True  # noqa: F841
     assert allclose_or_equiv(a1, a2, rtol=rtol, atol=atol)
示例#6
0
 def assertAllClose(self, a1, a2, rtol=1e-05, atol=1e-8):
     assert allclose_or_equiv(a1, a2, rtol=rtol, atol=atol)
示例#7
0
 def assertAllClose(self, a1, a2, rtol=1e-05, atol=1e-8):
     __tracebackhide__ = True  # noqa: F841
     assert allclose_or_equiv(a1, a2, rtol=rtol, atol=atol)