Function custom_scheduler_get

dask/array/tests/test_xarray.py:101–106

(dsk, keys, expected, **kwargs)

    pytest.importorskip("zarr")

    def custom_scheduler_get(dsk, keys, expected, **kwargs):
        dsk = dsk.__dask_graph__()
        assert (
            len(dsk) == expected
        ), f"False number of tasks got {len(dsk)} but expected {expected}"
        return [42 for _ in keys]

    # First test that this mocking stuff works as expected
    with pytest.raises(AssertionError, match="False number of tasks"):