def test_raw_data_1(): data = np.random.randn(200, 4) rd = NumPyRawDataReader(data) chunks = rd.chunks(chunk_size=100, chunk_overlap=20) ch = 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]) ch = 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]) assert str(ch)
def test_raw_data_iterator(): data = np.random.randn(200, 4) rd = NumPyRawDataReader(data) assert len([ch for ch in rd.chunks(chunk_size=100, chunk_overlap=20)]) == 3