def test__futures_to_dask_arrays(c, s, a, b): import dask.array as da fs = c.map(np.ones, [(5, i) for i in range(1, 6)], pure=False) arrs = yield _futures_to_dask_arrays(fs) assert len(fs) == len(arrs) == 5 assert all(a.shape == (5, i + 1) for i, a in enumerate(arrs)) assert all(a.dtype == np.ones(1).dtype for a in arrs) assert all(isinstance(a, da.Array) for a in arrs) xs = yield c._gather(c.compute(arrs)) assert all((x == 1).all() for x in xs)
def test__futures_to_dask_arrays(e, s, a, b): import dask.array as da fs = e.map(np.ones, [(5, i) for i in range(1, 6)], pure=False) arrs = yield _futures_to_dask_arrays(fs) assert len(fs) == len(arrs) == 5 assert all(a.shape == (5, i + 1) for i, a in enumerate(arrs)) assert all(a.dtype == np.ones(1).dtype for a in arrs) assert all(isinstance(a, da.Array) for a in arrs) xs = yield e._gather(e.compute(arrs)) assert all((x == 1).all() for x in xs)