| 2060 | |
| 2061 | |
| 2062 | def test_store_kwargs(): |
| 2063 | d = da.ones((10, 10), chunks=(2, 2)) |
| 2064 | a = d + 1 |
| 2065 | |
| 2066 | called = [False] |
| 2067 | |
| 2068 | def get_func(*args, **kwargs): |
| 2069 | assert kwargs.pop("foo") == "test kwarg" |
| 2070 | r = dask.get(*args, **kwargs) |
| 2071 | called[0] = True |
| 2072 | return r |
| 2073 | |
| 2074 | called[0] = False |
| 2075 | at = np.zeros(shape=(10, 10)) |
| 2076 | store([a], [at], scheduler=get_func, foo="test kwarg") |
| 2077 | assert called[0] |
| 2078 | |
| 2079 | called[0] = False |
| 2080 | at = np.zeros(shape=(10, 10)) |
| 2081 | a.store(at, scheduler=get_func, foo="test kwarg") |
| 2082 | assert called[0] |
| 2083 | |
| 2084 | called[0] = False |
| 2085 | at = np.zeros(shape=(10, 10)) |
| 2086 | store([a], [at], scheduler=get_func, return_stored=True, foo="test kwarg") |
| 2087 | assert called[0] |
| 2088 | |
| 2089 | |
| 2090 | def test_store_delayed_target(): |