def test_merge_negativ_overlap(): """Try to merge multiple windows with negativ overlap """ X = np.concatenate([np.ones((1, 10, 2)), np.zeros((1, 10, 2))]) with pytest.raises(ValueError, match="window_overlap should be non-negative."): Xtransform = _merge_overlap(X, window_overlap=-1)
def test_merge_simple2(): """Merge multiple windows with overlap """ X = np.reshape(np.arange(1., 21.), (5, 2, 2)) Xtransform = _merge_overlap(X, window_overlap=1) npt.assert_array_almost_equal(Xtransform, np.concatenate([X[0:-1, 0, :], X[-1, :, :]]))
def test_merge_too_big_overlap(): """Try to merge multiple windows with too big overlap """ X = np.concatenate([np.ones((1, 10, 2)), np.zeros((1, 10, 2))]) with pytest.raises( ValueError, match="window_overlap cannot be higher than n_samples."): Xtransform = _merge_overlap(X, window_overlap=11)
def test_merge_wrong_dim(): X = np.concatenate([np.ones((10, 2)), np.zeros((10, 2))]) with pytest.raises( ValueError, match='X.shape should be \(n_trials, n_samples, n_electrodes\).'): Xtransform = _merge_overlap(X, window_overlap=1)
def test_merge_simple(): """Merge multiple windows with overlap """ X = np.concatenate([np.ones((1, 10, 2)), np.zeros((1, 10, 2))]) Xtransform = _merge_overlap(X, window_overlap=10) npt.assert_array_almost_equal(Xtransform, X[-1, :, :])