Function get

dask/tests/test_base.py:922–924

(dsk, keys, *args, **kwargs)

    called = [False]

    def get(dsk, keys, *args, **kwargs):
        called[0] = True
        return dask.get(dsk, keys)

    assert delayed(lambda: 1)().compute(scheduler=get) == 1
    assert called[0]