def test_futures_to_dask_arrays(loop):
with cluster() as (c, [a, b]):
with Executor(('127.0.0.1', c['port']), loop=loop) as e:
futures = e.map(np.ones, [(5, i) for i in range(1, 6)])
x = future_to_dask_array(futures[0])
assert x.shape == (5, 1)
assert (x.compute(get=e.get) == 1).all()
xs = futures_to_dask_arrays(futures)
assert [x.shape for x in xs] == [(5, i) for i in range(1, 6)]
def test_futures_to_dask_arrays(loop):
with cluster() as (c, [a, b]):
with Client(('127.0.0.1', c['port']), loop=loop) as c:
futures = c.map(np.ones, [(5, i) for i in range(1, 6)])
x = future_to_dask_array(futures[0])
assert x.shape == (5, 1)
assert (x.compute(get=c.get) == 1).all()
xs = futures_to_dask_arrays(futures)
assert [x.shape for x in xs] == [(5, i) for i in range(1, 6)]
def test_futures_to_dask_arrays(loop):
with cluster() as (c, [a, b]):
with Client(("127.0.0.1", c["port"]), loop=loop) as c:
futures = c.map(np.ones, [(5, i) for i in range(1, 6)])
x = future_to_dask_array(futures[0])
assert x.shape == (5, 1)
assert (x.compute(get=c.get) == 1).all()
xs = futures_to_dask_arrays(futures)
assert [x.shape for x in xs] == [(5, i) for i in range(1, 6)]