def test_stim_elim():
"""Test eliminate stim artifact"""
raw = Raw(raw_fname, preload=True)
events = read_events(event_fname)
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.)
def test_stim_elim():
"""Test eliminate stim artifact"""
raw = Raw(raw_fname, preload=True)
events = read_events(event_fname)
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.)
def test_stim_elim():
"""Test eliminate stim artifact"""
raw = Raw(raw_fname, preload=True)
events = read_events(event_fname)
event_idx = np.where(events[:, 2] == 1)[0][0]
tidx = int(events[event_idx, 0] - raw.first_samp)
# use window around stimulus
tmin = -0.02
tmax = 0.02
test_tminidx = int(-0.01 * raw.info['sfreq'])
test_tmaxidx = int(0.01 * raw.info['sfreq'])
raw = eliminate_stim_artifact(raw, events, event_id=1, tmin=tmin,
tmax=tmax, mode='linear')
data, times = raw[:, (tidx + test_tminidx):(tidx + test_tmaxidx)]
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=tmin,
tmax=tmax, mode='window')
data, times = raw[:, (tidx + test_tminidx):(tidx + test_tmaxidx)]
assert_true(np.all(data) == 0.)
# use window before stimulus
tmin = -0.045
tmax = 0.015
test_tminidx = int(-0.035 * raw.info['sfreq'])
test_tmaxidx = int(-0.015 * raw.info['sfreq'])
raw = eliminate_stim_artifact(raw, events, event_id=1, tmin=tmin,
tmax=tmax, mode='linear')
data, times = raw[:, (tidx + test_tminidx):(tidx + test_tmaxidx)]
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=tmin,
tmax=tmax, mode='window')
data, times = raw[:, (tidx + test_tminidx):(tidx + test_tmaxidx)]
assert_true(np.all(data) == 0.)
# use window after stimulus
tmin = 0.005
tmax = 0.045
test_tminidx = int(0.015 * raw.info['sfreq'])
test_tmaxidx = int(0.035 * raw.info['sfreq'])
raw = eliminate_stim_artifact(raw, events, event_id=1, tmin=tmin,
tmax=tmax, mode='linear')
data, times = raw[:, (tidx + test_tminidx):(tidx + test_tmaxidx)]
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=tmin,
tmax=tmax, mode='window')
data, times = raw[:, (tidx + test_tminidx):(tidx + test_tmaxidx)]
assert_true(np.all(data) == 0.)