Function test_boolean_mask_with_unknown_shape

dask/array/tests/test_slicing.py:1133–1153

(
    shapes: tuple[float | int, float | int],
)

    ],
)
def test_boolean_mask_with_unknown_shape(
    shapes: tuple[float | int, float | int],
) -> None:
    x_shape, mask_shape = shapes
    arr = delayed(np.ones(10))
    x = da.concatenate(
        [
            da.from_delayed(arr, shape=(x_shape,), dtype=float),
            da.from_delayed(arr, shape=(x_shape,), dtype=float),
        ]
    )
    mask = da.concatenate(
        [
            da.from_delayed(arr, shape=(mask_shape,), dtype=bool),
            da.from_delayed(arr, shape=(mask_shape,), dtype=bool),
        ]
    )
    x[mask] = 2

    expected = np.full(20, 2.0)
    assert_eq(x, expected)