()
| 174 | |
| 175 | |
| 176 | def test_complex_ordering(): |
| 177 | pytest.importorskip("numpy") |
| 178 | da = pytest.importorskip("dask.array") |
| 179 | from dask.diagnostics import Callback |
| 180 | |
| 181 | actual_order = [] |
| 182 | |
| 183 | def track_order(key, dask, state): |
| 184 | actual_order.append(key) |
| 185 | |
| 186 | x = da.random.normal(size=(20, 20), chunks=(-1, -1)) |
| 187 | res = (x.dot(x.T) - x.mean(axis=0)).std() |
| 188 | dsk = dict(res.__dask_graph__()) |
| 189 | exp_order_dict = order(dsk) |
| 190 | exp_order = sorted(exp_order_dict.keys(), key=exp_order_dict.get) |
| 191 | with Callback(pretask=track_order): |
| 192 | get_sync(dsk, exp_order[-1]) |
| 193 | assert actual_order == exp_order |
| 194 | |
| 195 | |
| 196 | def test_ensure_calculate_only_whats_needed(): |
nothing calls this directly
no test coverage detected
searching dependent graphs…