def test_fromfunction():
def f(x, y):
return x + y
d = da.fromfunction(f, shape=(5, 5), chunks=(2, 2), dtype='f8')
assert_eq(d, np.fromfunction(f, shape=(5, 5)))
assert same_keys(d, da.fromfunction(f, shape=(5, 5), chunks=(2, 2), dtype='f8'))
def test_fromfunction(func, dtype, kwargs):
a = np.fromfunction(func, shape=(5, 5), dtype=dtype, **kwargs)
d = da.fromfunction(func, shape=(5, 5), chunks=(2, 2), dtype=dtype, **kwargs)
assert_eq(d, a)
d2 = da.fromfunction(func, shape=(5, 5), chunks=(2, 2), dtype=dtype, **kwargs)
assert same_keys(d, d2)
def test_fromfunction(func, dtype, kwargs):
a = np.fromfunction(func, shape=(5, 5), dtype=dtype, **kwargs)
d = da.fromfunction(
func, shape=(5, 5), chunks=(2, 2), dtype=dtype, **kwargs
)
assert_eq(d, a)
d2 = da.fromfunction(
func, shape=(5, 5), chunks=(2, 2), dtype=dtype, **kwargs
)
assert same_keys(d, d2)
def complex_big_data():
return DataArray(da.fromfunction(lambda y, x, E, T, t: (y + x) * np.cos(E/10) + T + t ** 2,
shape=(size, size, size, size, size),
chunks=10,
dtype=float),
dims=['y (μm)', 'x (μm)', 'E (eV)', 'T (K)', 't (s)'],
coords=[np.arange(size) / 10.,
np.arange(size) / 10.,
np.arange(size) * 100,
np.arange(size) / 10.,
np.arange(size) / 10.])
def da_date_idx(start, N, freq, chunks):
return da.fromfunction(IdxToDate(start, freq),
shape=(N, ),
chunks=chunks,
dtype='M8[ns]')
