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