1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
# Author: Daniel G Wakeman <dwakeman@nmr.mgh.harvard.edu>
# Denis A. Engemann <denis.engemann@gmail.com>
#
# License: BSD (3-clause)
import os.path as op
from copy import deepcopy
import numpy as np
from nose.tools import assert_raises, assert_true, assert_equal
from scipy.io import savemat
from mne.channels import (rename_channels, read_ch_connectivity,
ch_neighbor_connectivity)
from mne.io import read_info
from mne.io.constants import FIFF
from mne.fixes import partial
from mne.utils import _TempDir
tempdir = _TempDir()
base_dir = op.join(op.dirname(__file__), '..', 'io', 'tests', 'data')
raw_fname = op.join(base_dir, 'test_raw.fif')
def test_rename_channels():
"""Test rename channels
"""
info = read_info(raw_fname)
# Error Tests
# Test channel name exists in ch_names
mapping = {'EEG 160': 'EEG060'}
assert_raises(ValueError, rename_channels, info, mapping)
# Test change to EEG channel
mapping = {'EOG 061': ('EEG 061', 'eeg')}
assert_raises(ValueError, rename_channels, info, mapping)
# Test change to illegal channel type
mapping = {'EOG 061': ('MEG 061', 'meg')}
assert_raises(ValueError, rename_channels, info, mapping)
# Test channel type which you are changing from e.g. MEG
mapping = {'MEG 2641': ('MEG2641', 'eeg')}
assert_raises(ValueError, rename_channels, info, mapping)
# Test improper mapping configuration
mapping = {'MEG 2641': 1.0}
assert_raises(ValueError, rename_channels, info, mapping)
# Test duplicate named channels
mapping = {'EEG 060': 'EOG 061'}
assert_raises(ValueError, rename_channels, info, mapping)
# Test successful changes
# Test ch_name and ch_names are changed
info2 = deepcopy(info) # for consistency at the start of each test
info2['bads'] = ['EEG 060', 'EOG 061']
mapping = {'EEG 060': 'EEG060', 'EOG 061': 'EOG061'}
rename_channels(info2, mapping)
assert_true(info2['chs'][374]['ch_name'] == 'EEG060')
assert_true(info2['ch_names'][374] == 'EEG060')
assert_true('EEG060' in info2['bads'])
assert_true(info2['chs'][375]['ch_name'] == 'EOG061')
assert_true(info2['ch_names'][375] == 'EOG061')
assert_true('EOG061' in info2['bads'])
# Test type change
info2 = deepcopy(info)
info2['bads'] = ['EEG 060', 'EEG 059']
mapping = {'EEG 060': ('EOG 060', 'eog'), 'EEG 059': ('EOG 059', 'eog')}
rename_channels(info2, mapping)
assert_true(info2['chs'][374]['ch_name'] == 'EOG 060')
assert_true(info2['ch_names'][374] == 'EOG 060')
assert_true('EOG 060' in info2['bads'])
assert_true(info2['chs'][374]['kind'] is FIFF.FIFFV_EOG_CH)
assert_true(info2['chs'][373]['ch_name'] == 'EOG 059')
assert_true(info2['ch_names'][373] == 'EOG 059')
assert_true('EOG 059' in info2['bads'])
assert_true(info2['chs'][373]['kind'] is FIFF.FIFFV_EOG_CH)
def test_read_ch_connectivity():
"Test reading channel connectivity templates"
a = partial(np.array, dtype='<U7')
# no pep8
nbh = np.array([[(['MEG0111'], [[a(['MEG0131'])]]),
(['MEG0121'], [[a(['MEG0111'])],
[a(['MEG0131'])]]),
(['MEG0131'], [[a(['MEG0111'])],
[a(['MEG0121'])]])]],
dtype=[('label', 'O'), ('neighblabel', 'O')])
mat = dict(neighbours=nbh)
mat_fname = op.join(tempdir, 'test_mat.mat')
savemat(mat_fname, mat)
ch_connectivity = read_ch_connectivity(mat_fname)
x = ch_connectivity
assert_equal(x.shape, (3, 3))
assert_equal(x[0, 1], False)
assert_equal(x[0, 2], True)
assert_true(np.all(x.diagonal()))
assert_raises(ValueError, read_ch_connectivity, mat_fname, [0, 3])
ch_connectivity = read_ch_connectivity(mat_fname, picks=[0, 2])
assert_equal(ch_connectivity.shape[0], 2)
ch_names = ['EEG01', 'EEG02', 'EEG03']
neighbors = [['EEG02'], ['EEG04'], ['EEG02']]
assert_raises(ValueError, ch_neighbor_connectivity, ch_names, neighbors)
neighbors = [['EEG02'], ['EEG01', 'EEG03'], ['EEG 02']]
assert_raises(ValueError, ch_neighbor_connectivity, ch_names[:2],
neighbors)
neighbors = [['EEG02'], 'EEG01', ['EEG 02']]
assert_raises(ValueError, ch_neighbor_connectivity, ch_names, neighbors)
|
