def test_window_shapes():
x = np.array([0,1,2,3,4])
def winshape(x):
return x.shape[0]
y = parakeet.pmap1(winshape, x, 3)
expected = np.array([2,3,3,3,2])
assert y.shape == expected.shape, "Expected shape %s but got %s" % (expected.shape, y.shape)
assert np.all(y == expected), "Expected array %s but got %s" % (expected, y)
def test_window_shapes():
x = np.array([0, 1, 2, 3, 4])
def winshape(x):
return x.shape[0]
y = parakeet.pmap1(winshape, x, 3)
expected = np.array([2, 3, 3, 3, 2])
assert y.shape == expected.shape, "Expected shape %s but got %s" % (
expected.shape, y.shape)
assert np.all(
y == expected), "Expected array %s but got %s" % (expected, y)
def winmap_avg1d(x, w=3):
return parakeet.pmap1(avg1d, x, w)
def winmap_avg1d(x, w=3):
return parakeet.pmap1(avg1d, x, w)
