def test_apply_function_evk(): """Check the apply_function method for evoked data.""" # create fake evoked data to use for checking apply_function data = np.random.rand(10, 1000) info = create_info(10, 1000., 'eeg') evoked = EvokedArray(data, info) evoked_data = evoked.data.copy() # check apply_function channel-wise def fun(data, multiplier): return data * multiplier mult = -1 applied = evoked.apply_function(fun, n_jobs=None, multiplier=mult) assert np.shape(applied.data) == np.shape(evoked_data) assert np.equal(applied.data, evoked_data * mult).all()
def test_apply_function_evk(): """Check the apply_function method for evoked data.""" # create fake evoked data to use for checking apply_function data = np.random.rand(10, 1000) info = create_info(10, 1000., 'eeg') evoked = EvokedArray(data, info) evoked_data = evoked.data.copy() # check apply_function channel-wise mult_param = -1 kwargs = dict(mult_param=mult_param) applied = evoked.apply_function(fun, picks=None, dtype=None, n_jobs=1, **kwargs) assert np.shape(applied.data) == np.shape(evoked_data) assert np.equal(applied.data, evoked_data * mult_param).all()