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)