def test_chunk(): data = np.random.randn(200, 4) chunks = chunk_bounds(data.shape[0], 100, overlap=20) # Chunk 1. ch = Chunk(data=data, bounds=next(chunks)) assert ch.window_full == (0, 100) assert ch.window_keep == (0, 90) assert np.array_equal(ch.data_chunk_full, data[0:100]) assert np.array_equal(ch.data_chunk_keep, data[0:90]) # Chunk 2. ch = Chunk(data=data, bounds=next(chunks)) assert ch.window_full == (80, 180) assert ch.window_keep == (90, 170) assert np.array_equal(ch.data_chunk_full, data[80:180]) assert np.array_equal(ch.data_chunk_keep, data[90:170])
def test_chunk_bounds(): chunks = chunk_bounds(200, 100, overlap=20) assert next(chunks) == (0, 100, 0, 90) assert next(chunks) == (80, 180, 90, 170) assert next(chunks) == (160, 200, 170, 200)