def test_epoch_trigger_shift():
"Test the shift_mne_epoch_trigger() function"
epochs = datasets.get_mne_sample(sns=True, sub="[1,2,3]")['epochs']
epochs.info['projs'] = []
n_lost_start = np.sum(epochs.times < epochs.tmin + 0.05)
n_lost_end = np.sum(epochs.times > epochs.tmax - 0.05)
data = epochs.get_data()
# don't shift
epochs_s = shift_mne_epoch_trigger(epochs, [0, 0, 0])
assert_array_equal(epochs_s.get_data(), data)
epochs_s = shift_mne_epoch_trigger(epochs, [-0.05, 0., 0.05])
data_s = epochs_s.get_data()
assert_array_equal(data_s[0], data[0, :, :-(n_lost_end + n_lost_start)])
assert_array_equal(data_s[1], data[1, :, n_lost_start:-n_lost_end])
assert_array_equal(data_s[2], data[2, :, n_lost_end + n_lost_start:])
assert_allclose(epochs_s.times,
epochs.times[n_lost_start:-n_lost_end],
rtol=1e-1,
atol=1e-3) # ms accuracy
epochs_s = shift_mne_epoch_trigger(epochs, [0.05, 0., 0.05])
data_s = epochs_s.get_data()
assert_array_equal(data_s[0], data[0, :, n_lost_end:])
assert_array_equal(data_s[1], data[1, :, :-n_lost_end])
assert_array_equal(data_s[2], data[2, :, n_lost_end:])
assert_allclose(epochs_s.times,
epochs.times[:-n_lost_end],
rtol=1e-1,
atol=1e-3) # ms accuracy
def test_epoch_trigger_shift():
"Test the shift_mne_epoch_trigger() function"
epochs = datasets.get_mne_sample(sns=True, sub="[1,2,3]")['epochs']
n_lost_start = np.sum(epochs.times < epochs.tmin + 0.05)
n_lost_end = np.sum(epochs.times > epochs.tmax - 0.05)
data = epochs.get_data()
epochs_s = shift_mne_epoch_trigger(epochs, [0, 0, 0])
assert_array_equal(epochs_s.get_data(), data)
epochs_s = shift_mne_epoch_trigger(epochs, [-0.05, 0., 0.05])
data_s = epochs_s.get_data()
assert_array_equal(data_s[0], data[0, :, : -(n_lost_end + n_lost_start)])
assert_array_equal(data_s[1], data[1, :, n_lost_start: -n_lost_end])
assert_array_equal(data_s[2], data[2, :, n_lost_end + n_lost_start:])
assert_allclose(epochs_s.times, epochs.times[n_lost_start: -n_lost_end],
rtol=1e-1, atol=1e-3) # ms accuracy
epochs_s = shift_mne_epoch_trigger(epochs, [0.05, 0., 0.05])
data_s = epochs_s.get_data()
assert_array_equal(data_s[0], data[0, :, n_lost_end:])
assert_array_equal(data_s[1], data[1, :, :-n_lost_end])
assert_array_equal(data_s[2], data[2, :, n_lost_end:])
assert_allclose(epochs_s.times, epochs.times[:-n_lost_end],
rtol=1e-1, atol=1e-3) # ms accuracy
def test_dataobjects():
"Test handing MNE-objects as data-objects"
shift = np.array([0.1, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.1, -0.1])
epochs = datasets.get_mne_epochs()
ds = Dataset(('a', Factor('ab', repeat=8)),
('epochs', epochs))
ds['ets'] = shift_mne_epoch_trigger(epochs, shift, min(shift), max(shift))
# ds operations
sds = ds.sub("a == 'a'")
ads = ds.aggregate('a')
# asndvar
ndvar = asndvar(ds['epochs'])
ndvar = asndvar(ds['ets'])
# connectivity
ds = datasets.get_mne_sample(sub=[0], sns=True)
sensor = ds['meg'].sensor
c = sensor.connectivity()
assert_array_equal(c[:, 0] < c[:, 1], True)
eq_(c.max(), len(sensor) - 1)
def test_dataobjects():
"Test handing MNE-objects as data-objects"
shift = np.array([0.1, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.1, -0.1])
epochs = datasets.get_mne_epochs()
ds = Dataset(('a', Factor('ab', repeat=8)),
('epochs', epochs))
ds['ets'] = shift_mne_epoch_trigger(epochs, shift, min(shift), max(shift))
# ds operations
sds = ds.sub("a == 'a'")
ads = ds.aggregate('a')
# asndvar
ndvar = asndvar(ds['epochs'])
ndvar = asndvar(ds['ets'])
# connectivity
ds = datasets.get_mne_sample(sub=[0], sns=True)
sensor = ds['meg'].sensor
c = sensor.connectivity()
assert_array_equal(c[:, 0] < c[:, 1], True)
eq_(c.max(), len(sensor) - 1)