def test_edf_stim_ch_pick_up(test_input, EXPECTED): """Test stim_channel.""" TYPE_LUT = {v[0]: k for k, v in _KIND_DICT.items()} fname = op.join(data_dir, 'test_stim_channel.edf') raw = read_raw_edf(fname, stim_channel=test_input) ch_types = {ch['ch_name']: TYPE_LUT[ch['kind']] for ch in raw.info['chs']} assert ch_types == EXPECTED
def test_edf_stim_ch_pick_up(test_input, EXPECTED): """Test stim_channel.""" TYPE_LUT = {v[0]: k for k, v in _KIND_DICT.items()} fname = op.join(data_dir, 'test_stim_channel.edf') raw = read_raw_edf(fname, stim_channel=test_input) ch_types = {ch['ch_name']: TYPE_LUT[ch['kind']] for ch in raw.info['chs']} assert ch_types == EXPECTED
def test_edf_stim_ch_pick_up(test_input, EXPECTED): """Test stim_channel.""" # This is fragile for EEG/EEG-CSD, so just omit csd TYPE_LUT = {v[0]: k for k, v in _KIND_DICT.items() if k != 'csd'} fname = op.join(data_dir, 'test_stim_channel.edf') raw = read_raw_edf(fname, stim_channel=test_input) ch_types = {ch['ch_name']: TYPE_LUT[ch['kind']] for ch in raw.info['chs']} assert ch_types == EXPECTED