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
|
# Authors: The MNE-Python contributors.
# License: BSD-3-Clause
# Copyright the MNE-Python contributors.
import pytest
from numpy.testing import assert_array_almost_equal
from mne.datasets import testing
from mne.io import read_raw_edf, read_raw_nihon
from mne.io.nihon import nihon
from mne.io.nihon.nihon import (
_read_nihon_annotations,
_read_nihon_header,
_read_nihon_metadata,
)
from mne.io.tests.test_raw import _test_raw_reader
data_path = testing.data_path(download=False)
@testing.requires_testing_data
def test_nihon_eeg():
"""Test reading Nihon Kohden EEG files."""
fname = data_path / "NihonKohden" / "MB0400FU.EEG"
raw = read_raw_nihon(fname.as_posix(), preload=True)
assert "RawNihon" in repr(raw)
_test_raw_reader(read_raw_nihon, fname=fname, test_scaling=False)
fname_edf = data_path / "NihonKohden" / "MB0400FU.EDF"
raw_edf = read_raw_edf(fname_edf, preload=True)
assert raw._data.shape == raw_edf._data.shape
assert raw.info["sfreq"] == raw.info["sfreq"]
# ch names and order are switched in the EDF
edf_ch_names = {x: x.split(" ")[1].replace("-Ref", "") for x in raw_edf.ch_names}
raw_edf.rename_channels(edf_ch_names)
assert raw.ch_names == raw_edf.ch_names
for i, an1 in enumerate(raw.annotations):
# EDF has some weird annotations, which are not in the LOG file
an2 = raw_edf.annotations[i * 2 + 1]
assert an1["onset"] == an2["onset"]
assert an1["duration"] == an2["duration"]
# Also, it prepends 'Segment: ' to some annotations
t_desc = an2["description"].replace("Segment: ", "")
assert an1["description"] == t_desc
assert_array_almost_equal(raw._data, raw_edf._data)
with pytest.raises(ValueError, match="Not a valid Nihon Kohden EEG file"):
raw = read_raw_nihon(fname_edf, preload=True)
with pytest.raises(ValueError, match="Not a valid Nihon Kohden EEG file"):
raw = _read_nihon_header(fname_edf)
bad_fname = data_path / "eximia" / "text_eximia.nxe"
msg = "No PNT file exists. Metadata will be blank"
with pytest.warns(RuntimeWarning, match=msg):
meta = _read_nihon_metadata(bad_fname)
assert len(meta) == 0
msg = "No LOG file exists. Annotations will not be read"
with pytest.warns(RuntimeWarning, match=msg):
annot = _read_nihon_annotations(bad_fname)
assert all(len(x) == 0 for x in annot.values())
# the nihon test file has $A1 and $A2 in it, which are not EEG
assert "$A1" in raw.ch_names
# assert that channels with $ are 'misc'
picks = [ch for ch in raw.ch_names if ch.startswith("$")]
ch_types = raw.get_channel_types(picks=picks)
assert all(ch == "misc" for ch in ch_types)
@testing.requires_testing_data
def test_nihon_duplicate_channels(monkeypatch):
"""Test deduplication of channel names."""
fname = data_path / "NihonKohden" / "MB0400FU.EEG"
def return_channel_duplicates(fname):
ch_names = nihon._default_chan_labels
ch_names[1] = ch_names[0]
return ch_names
monkeypatch.setattr(nihon, "_read_21e_file", return_channel_duplicates)
assert len(nihon._read_21e_file(fname)) > len(set(nihon._read_21e_file(fname)))
msg = (
"Channel names are not unique, found duplicates for: "
"{'FP1'}. Applying running numbers for duplicates."
)
with pytest.warns(RuntimeWarning, match=msg):
read_raw_nihon(fname)
|
