def test_pickle(fname_info, unlocked):
"""Test that Info can be (un)pickled."""
if fname_info == 'create_info':
info = create_info(3, 1000., 'eeg')
else:
info = read_info(fname_info)
assert not info._unlocked
info._unlocked = unlocked
data = pickle.dumps(info)
info_un = pickle.loads(data)
assert isinstance(info_un, Info)
assert_object_equal(info, info_un)
assert info_un._unlocked == unlocked
def test_field_round_trip(tmpdir):
"""Test round-trip for new fields."""
info = create_info(1, 1000., 'eeg')
for key in ('file_id', 'meas_id'):
info[key] = _generate_meas_id()
info['device_info'] = dict(
type='a', model='b', serial='c', site='d')
info['helium_info'] = dict(
he_level_raw=1., helium_level=2., orig_file_guid='e', meas_date=(1, 2))
fname = tmpdir.join('temp-info.fif')
write_info(fname, info)
info_read = read_info(fname)
info_read['dig'] = None # XXX eventually this should go away
assert_object_equal(info, info_read)
def test_field_round_trip(tmp_path):
"""Test round-trip for new fields."""
info = create_info(1, 1000., 'eeg')
with info._unlock():
for key in ('file_id', 'meas_id'):
info[key] = _generate_meas_id()
info['device_info'] = dict(type='a', model='b', serial='c', site='d')
info['helium_info'] = dict(he_level_raw=1.,
helium_level=2.,
orig_file_guid='e',
meas_date=(1, 2))
fname = tmp_path / 'temp-info.fif'
write_info(fname, info)
info_read = read_info(fname)
assert_object_equal(info, info_read)
def assert_indexing(info, picks_by_type, ref_meg=False, all_data=True):
"""Assert our indexing functions work properly."""
# First that our old and new channel typing functions are equivalent
_assert_channel_types(info)
# Next that channel_indices_by_type works
if not ref_meg:
idx = channel_indices_by_type(info)
for key in idx:
for p in picks_by_type:
if key == p[0]:
assert_array_equal(idx[key], p[1])
break
else:
assert len(idx[key]) == 0
# Finally, picks_by_type (if relevant)
if not all_data:
picks_by_type = [p for p in picks_by_type
if p[0] in _DATA_CH_TYPES_SPLIT]
picks_by_type = [(p[0], np.array(p[1], int)) for p in picks_by_type]
actual = _picks_by_type(info, ref_meg=ref_meg)
assert_object_equal(actual, picks_by_type)
if not ref_meg and idx['hbo']: # our old code had a bug
with pytest.raises(TypeError, match='unexpected keyword argument'):
_picks_by_type_old(info, ref_meg=ref_meg)
else:
old = _picks_by_type_old(info, ref_meg=ref_meg)
assert_object_equal(old, picks_by_type)
# test bads
info = info.copy()
info['bads'] = [info['chs'][picks_by_type[0][1][0]]['ch_name']]
picks_by_type = deepcopy(picks_by_type)
picks_by_type[0] = (picks_by_type[0][0], picks_by_type[0][1][1:])
actual = _picks_by_type(info, ref_meg=ref_meg)
assert_object_equal(actual, picks_by_type)
def assert_indexing(info, picks_by_type, ref_meg=False, all_data=True):
"""Assert our indexing functions work properly."""
# First that our old and new channel typing functions are equivalent
_assert_channel_types(info)
# Next that channel_indices_by_type works
if not ref_meg:
idx = channel_indices_by_type(info)
for key in idx:
for p in picks_by_type:
if key == p[0]:
assert_array_equal(idx[key], p[1])
break
else:
assert len(idx[key]) == 0
# Finally, picks_by_type (if relevant)
if not all_data:
picks_by_type = [p for p in picks_by_type
if p[0] in _DATA_CH_TYPES_SPLIT]
picks_by_type = [(p[0], np.array(p[1], int)) for p in picks_by_type]
actual = _picks_by_type(info, ref_meg=ref_meg)
assert_object_equal(actual, picks_by_type)
if not ref_meg and idx['hbo']: # our old code had a bug
with pytest.raises(TypeError, match='unexpected keyword argument'):
_picks_by_type_old(info, ref_meg=ref_meg)
else:
old = _picks_by_type_old(info, ref_meg=ref_meg)
assert_object_equal(old, picks_by_type)
# test bads
info = info.copy()
info['bads'] = [info['chs'][picks_by_type[0][1][0]]['ch_name']]
picks_by_type = deepcopy(picks_by_type)
picks_by_type[0] = (picks_by_type[0][0], picks_by_type[0][1][1:])
actual = _picks_by_type(info, ref_meg=ref_meg)
assert_object_equal(actual, picks_by_type)
def _test_anonymize_info(base_info):
"""Test that sensitive information can be anonymized."""
pytest.raises(TypeError, anonymize_info, 'foo')
default_anon_dos = datetime(2000, 1, 1, 0, 0, 0, tzinfo=timezone.utc)
default_str = "mne_anonymize"
default_subject_id = 0
default_desc = ("Anonymized using a time shift" +
" to preserve age at acquisition")
# Test no error for incomplete info
info = base_info.copy()
info.pop('file_id')
anonymize_info(info)
# Fake some subject data
meas_date = datetime(2010, 1, 1, 0, 0, 0, tzinfo=timezone.utc)
base_info['meas_date'] = meas_date
base_info['subject_info'] = dict(id=1,
his_id='foobar',
last_name='bar',
first_name='bar',
birthday=(1987, 4, 8),
sex=0,
hand=1)
# generate expected info...
# first expected result with no options.
# will move DOS from 2010/1/1 to 2000/1/1 which is 3653 days.
exp_info = base_info.copy()
exp_info['description'] = default_desc
exp_info['experimenter'] = default_str
exp_info['proj_name'] = default_str
exp_info['proj_id'] = np.array([0])
exp_info['subject_info']['first_name'] = default_str
exp_info['subject_info']['last_name'] = default_str
exp_info['subject_info']['id'] = default_subject_id
exp_info['subject_info']['his_id'] = str(default_subject_id)
# this bday is 3653 days different. the change in day is due to a
# different number of leap days between 1987 and 1977 than between
# 2010 and 2000.
exp_info['subject_info']['birthday'] = (1977, 4, 7)
exp_info['meas_date'] = default_anon_dos
# make copies
exp_info_3 = exp_info.copy()
# adjust each expected outcome
delta_t = timedelta(days=3653)
for key in ('file_id', 'meas_id'):
value = exp_info.get(key)
if value is not None:
assert 'msecs' not in value
tmp = _add_timedelta_to_stamp((value['secs'], value['usecs']),
-delta_t)
value['secs'] = tmp[0]
value['usecs'] = tmp[1]
value['machid'][:] = 0
# exp 2 tests the keep_his option
exp_info_2 = exp_info.copy()
exp_info_2['subject_info']['his_id'] = 'foobar'
# exp 3 tests is a supplied daysback
delta_t_2 = timedelta(days=43)
exp_info_3['subject_info']['birthday'] = (1987, 2, 24)
exp_info_3['meas_date'] = meas_date - delta_t_2
for key in ('file_id', 'meas_id'):
value = exp_info_3.get(key)
if value is not None:
assert 'msecs' not in value
tmp = _add_timedelta_to_stamp((value['secs'], value['usecs']),
-delta_t_2)
value['secs'] = tmp[0]
value['usecs'] = tmp[1]
value['machid'][:] = 0
# exp 4 tests is a supplied daysback
delta_t_3 = timedelta(days=223 + 364 * 500)
new_info = anonymize_info(base_info.copy())
assert_object_equal(new_info, exp_info)
new_info = anonymize_info(base_info.copy(), keep_his=True)
assert_object_equal(new_info, exp_info_2)
new_info = anonymize_info(base_info.copy(), daysback=delta_t_2.days)
assert_object_equal(new_info, exp_info_3)
with pytest.raises(RuntimeError, match='anonymize_info generated'):
anonymize_info(base_info.copy(), daysback=delta_t_3.days)
# assert_object_equal(new_info, exp_info_4)
# test with meas_date = None
base_info['meas_date'] = None
exp_info_3['meas_date'] = None
exp_info_3['file_id']['secs'] = DATE_NONE[0]
exp_info_3['file_id']['usecs'] = DATE_NONE[1]
exp_info_3['meas_id']['secs'] = DATE_NONE[0]
exp_info_3['meas_id']['usecs'] = DATE_NONE[1]
exp_info_3['subject_info'].pop('birthday', None)
if base_info['meas_date'] is None:
with pytest.warns(RuntimeWarning, match='all information'):
new_info = anonymize_info(base_info.copy(),
daysback=delta_t_2.days)
else:
new_info = anonymize_info(base_info.copy(), daysback=delta_t_2.days)
assert_object_equal(new_info, exp_info_3)
with pytest.warns(None): # meas_date is None
new_info = anonymize_info(base_info.copy())
assert_object_equal(new_info, exp_info_3)
def _test_anonymize_info(base_info):
"""Test that sensitive information can be anonymized."""
pytest.raises(TypeError, anonymize_info, 'foo')
default_anon_dos = datetime(2000, 1, 1, 0, 0, 0, tzinfo=timezone.utc)
default_str = "mne_anonymize"
default_subject_id = 0
default_desc = ("Anonymized using a time shift" +
" to preserve age at acquisition")
# Test no error for incomplete info
info = base_info.copy()
info.pop('file_id')
anonymize_info(info)
# Fake some subject data
meas_date = datetime(2010, 1, 1, 0, 0, 0, tzinfo=timezone.utc)
base_info['meas_date'] = _dt_to_stamp(meas_date)
base_info['subject_info'] = dict(id=1,
his_id='foobar',
last_name='bar',
first_name='bar',
birthday=(1987, 4, 8),
sex=0,
hand=1)
# generate expected info...
# first expected result with no options.
# will move DOS from 2010/1/1 to 2000/1/1 which is 3653 days.
exp_info = base_info.copy()
exp_info['description'] = default_desc
exp_info['experimenter'] = default_str
exp_info['proj_name'] = default_str
exp_info['proj_id'][:] = 0
exp_info['subject_info']['first_name'] = default_str
exp_info['subject_info']['last_name'] = default_str
exp_info['subject_info']['id'] = default_subject_id
exp_info['subject_info']['his_id'] = str(default_subject_id)
# this bday is 3653 days different. the change in day is due to a
# different number of leap days between 1987 and 1977 than between
# 2010 and 2000.
exp_info['subject_info']['birthday'] = (1977, 4, 7)
exp_info['meas_date'] = _dt_to_stamp(default_anon_dos)
for key in ('file_id', 'meas_id'):
value = exp_info.get(key)
if value is not None:
assert 'msecs' not in value
value['secs'] = exp_info['meas_date'][0]
value['usecs'] = exp_info['meas_date'][1]
value['machid'][:] = 0
# exp 2 tests the keep_his option
exp_info_2 = exp_info.copy()
exp_info_2['subject_info']['his_id'] = 'foobar'
# exp 3 tests is a supplied daysback
dt = timedelta(days=43)
exp_info_3 = exp_info.copy()
exp_info_3['subject_info']['birthday'] = (1987, 2, 24)
exp_info_3['meas_date'] = _dt_to_stamp(meas_date - dt)
for key in ('file_id', 'meas_id'):
value = exp_info_3.get(key)
if value is not None:
assert 'msecs' not in value
value['secs'] = exp_info_3['meas_date'][0]
value['usecs'] = exp_info_3['meas_date'][1]
value['machid'][:] = 0
new_info = anonymize_info(base_info.copy())
assert_object_equal(new_info, exp_info)
new_info = anonymize_info(base_info.copy(), keep_his=True)
assert_object_equal(new_info, exp_info_2)
new_info = anonymize_info(base_info.copy(), daysback=dt.days)
assert_object_equal(new_info, exp_info_3)
