()
| 1528 | |
| 1529 | @requires_dask |
| 1530 | def test_vectorize_exclude_dims_dask() -> None: |
| 1531 | # GH 3890 |
| 1532 | data_array_a = xr.DataArray([[0, 1, 2], [1, 2, 3]], dims=("x", "y")) |
| 1533 | data_array_b = xr.DataArray([[0, 1, 2, 3, 4], [1, 2, 3, 4, 5]], dims=("x", "y")) |
| 1534 | |
| 1535 | expected = xr.DataArray([3, 5], dims=["x"]) |
| 1536 | actual = apply_ufunc( |
| 1537 | pandas_median_add, |
| 1538 | data_array_a.chunk({"x": 1}), |
| 1539 | data_array_b.chunk({"x": 1}), |
| 1540 | input_core_dims=[["y"], ["y"]], |
| 1541 | exclude_dims=set("y"), |
| 1542 | vectorize=True, |
| 1543 | dask="parallelized", |
| 1544 | output_dtypes=[float], |
| 1545 | ) |
| 1546 | assert_identical(expected, actual) |
| 1547 | |
| 1548 | |
| 1549 | def test_corr_only_dataarray() -> None: |
nothing calls this directly
no test coverage detected
searching dependent graphs…