def test_piecewise():
np.random.seed(1337)
x = np.random.randint(10, size=(15, 16))
d = da.from_array(x, chunks=(4, 5))
assert_eq(
np.piecewise(x, [x < 5, x >= 5], [lambda e, v, k: e + 1, 5], 1, k=2),
da.piecewise(d, [d < 5, d >= 5], [lambda e, v, k: e + 1, 5], 1, k=2))
def test_piecewise():
np.random.seed(1337)
x = np.random.randint(10, size=(15, 16))
d = da.from_array(x, chunks=(4, 5))
assert_eq(
np.piecewise(x, [x < 5, x >= 5], [lambda e, v, k: e + 1, 5], 1, k=2),
da.piecewise(d, [d < 5, d >= 5], [lambda e, v, k: e + 1, 5], 1, k=2)
)
def test_piecewise():
x = np.random.randint(10, size=(15, 16))
d = da.from_array(x, chunks=(4, 5))
assert_eq(
np.piecewise(x, [x < 5, x >= 5], [lambda e, v, k: e + 1, 5], 1, k=2),
da.piecewise(d, [d < 5, d >= 5], [lambda e, v, k: e + 1, 5], 1, k=2))
assert_eq(
np.piecewise(
x, [x > 2, x <= 5],
[lambda e, v, k: e + 1, lambda e, v, k: v * e, lambda e, v, k: 0],
1,
k=2),
da.piecewise(
d, [d > 5, d <= 5],
[lambda e, v, k: e + 1, lambda e, v, k: v * e, lambda e, v, k: 0],
1,
k=2))
def test_piecewise_otherwise():
np.random.seed(1337)
x = np.random.randint(10, size=(15, 16))
d = da.from_array(x, chunks=(4, 5))
assert_eq(
np.piecewise(
x, [x > 5, x <= 2],
[lambda e, v, k: e + 1, lambda e, v, k: v * e, lambda e, v, k: 0],
1,
k=2),
da.piecewise(
d, [d > 5, d <= 2],
[lambda e, v, k: e + 1, lambda e, v, k: v * e, lambda e, v, k: 0],
1,
k=2))
def test_piecewise_otherwise():
np.random.seed(1337)
x = np.random.randint(10, size=(15, 16))
d = da.from_array(x, chunks=(4, 5))
assert_eq(
np.piecewise(
x,
[x > 5, x <= 2],
[lambda e, v, k: e + 1, lambda e, v, k: v * e, lambda e, v, k: 0],
1, k=2
),
da.piecewise(
d,
[d > 5, d <= 2],
[lambda e, v, k: e + 1, lambda e, v, k: v * e, lambda e, v, k: 0],
1, k=2
)
)
