def test_mae_r_xr_dask(a_dask, b_dask, dim):
actual = mae(a_dask, b_dask, dim)
dim, axis = _preprocess(dim)
_a_dask = a_dask.values
_b_dask = b_dask.values
axis = tuple(a_dask.dims.index(d) for d in dim)
res = _mae(_a_dask, _b_dask, axis)
expected = actual.copy()
expected.values = res
assert_allclose(actual, expected)
def test_mae_r_xr(a, b, dim):
actual = mae(a, b, dim)
dim, axis = _preprocess(dim)
_a = a.values
_b = b.values
axis = tuple(a.dims.index(d) for d in dim)
res = _mae(_a, _b, axis)
expected = actual.copy()
expected.values = res
assert_allclose(actual, expected)
def test_pearson_r_xr_dask(a_dask, b_dask, dim):
actual = pearson_r(a_dask, b_dask, dim)
dim, _ = _preprocess(dim)
if len(dim) > 1:
new_dim = '_'.join(dim)
_a_dask = a_dask.stack(**{new_dim: dim})
_b_dask = b_dask.stack(**{new_dim: dim})
else:
new_dim = dim[0]
_a_dask = a_dask
_b_dask = b_dask
axis = _a_dask.dims.index(new_dim)
res = _pearson_r(_a_dask.values, _b_dask.values, axis)
expected = actual.copy()
expected.values = res
assert_allclose(actual, expected)
def test_pearson_r_xr(a, b, dim):
actual = pearson_r(a, b, dim)
dim, _ = _preprocess(dim)
if len(dim) > 1:
new_dim = '_'.join(dim)
_a = a.stack(**{new_dim: dim})
_b = b.stack(**{new_dim: dim})
else:
new_dim = dim[0]
_a = a
_b = b
axis = _a.dims.index(new_dim)
res = _pearson_r(_a.values, _b.values, axis)
expected = actual.copy()
expected.values = res
assert_allclose(actual, expected)