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()
