This is an automated email from the git hooks/post-receive script. yoh pushed a commit to tag 0.4 in repository python-mne.
commit a3d41d41ee1907ec51cc0da2fe054c45b25f2c3a Author: Daniel Strohmeier <[email protected]> Date: Thu Jul 12 19:00:41 2012 +0200 TST : better tests in stim artifact --- mne/artifacts/stim.py | 2 +- mne/artifacts/tests/test_stim.py | 13 ++++++++++++- 2 files changed, 13 insertions(+), 2 deletions(-) diff --git a/mne/artifacts/stim.py b/mne/artifacts/stim.py index 01af320..0bdabcb 100644 --- a/mne/artifacts/stim.py +++ b/mne/artifacts/stim.py @@ -5,7 +5,7 @@ from scipy import signal, interpolate def eliminate_stim_artifact(raw, events, event_id, tmin=-0.005, tmax=0.01, mode='linear'): """Eliminates stimulations artifacts from raw data - + The raw object will be modified in place (no copy) Parameters diff --git a/mne/artifacts/tests/test_stim.py b/mne/artifacts/tests/test_stim.py index 8cbdc85..3a6827d 100644 --- a/mne/artifacts/tests/test_stim.py +++ b/mne/artifacts/tests/test_stim.py @@ -1,4 +1,5 @@ import os.path as op +import numpy as np from nose.tools import assert_true from numpy.testing import assert_array_almost_equal @@ -16,6 +17,16 @@ def test_find_ecg(): """Test eliminate stim artifact""" raw = Raw(raw_fname, preload=True) events = read_events(event_fname) - n_events = len(events) + event_idx = np.where(events[:, 2] == 1)[0][0] + tidx = events[event_idx, 0] - raw.first_samp + raw = eliminate_stim_artifact(raw, events, event_id=1, tmin=-0.005, tmax=0.01, mode='linear') + data, times = raw[:, tidx - 3:tidx + 5] + diff_data0 = np.diff(data[0]) + diff_data0 -= np.mean(diff_data0) + assert_array_almost_equal(diff_data0, np.zeros(len(diff_data0))) + raw = eliminate_stim_artifact(raw, events, event_id=1, tmin=-0.005, + tmax=0.01, mode='window') + data, times = raw[:, tidx:tidx + 1] + assert_true(np.all(data) == 0.) -- Alioth's /usr/local/bin/git-commit-notice on /srv/git.debian.org/git/debian-med/python-mne.git _______________________________________________ debian-med-commit mailing list [email protected] http://lists.alioth.debian.org/cgi-bin/mailman/listinfo/debian-med-commit
