def test1_split_into_blocks(self):
"""1D test
"""
arr_size = (3,)
arr_in = np.random.randn(*arr_size)
window_shape = (3,)
out = split_into_blocks(arr_in, window_shape, step=1)
out_real = [(0,)]
self.assertTrue(out == out_real)
def test4_split_into_blocks(self):
"""2D test
"""
arr_size = (3, 3)
arr_in = np.random.randn(*arr_size)
window_shape = (1, 1)
out = split_into_blocks(arr_in, window_shape, step=2)
out_real = [(0, 0), (0, 2), (2, 0), (2, 2)]
self.assertTrue(out == out_real)
