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)