-
-
Notifications
You must be signed in to change notification settings - Fork 390
/
tests_events.py
68 lines (48 loc) · 1.91 KB
/
tests_events.py
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
import matplotlib.pyplot as plt
import numpy as np
import pytest
import neurokit2 as nk
# =============================================================================
# Events
# =============================================================================
def test_events_find():
signal = np.cos(np.linspace(start=0, stop=20, num=1000))
events = nk.events_find(signal)
assert list(events["onset"]) == [0, 236, 550, 864]
events = nk.events_find(signal, duration_min=150)
assert list(events["onset"]) == [236, 550]
events = nk.events_find(signal, inter_min=300)
assert list(events["onset"]) == [0, 550, 864]
# No events found warning
signal = np.zeros(1000)
with pytest.warns(nk.misc.NeuroKitWarning, match=r'No events found.*'):
nk.events_find(signal)
def test_events_to_mne():
signal = np.cos(np.linspace(start=0, stop=20, num=1000))
events = nk.events_find(signal)
events, event_id = nk.events_to_mne(events)
assert event_id == {"event": 0}
def test_events_plot():
signal = np.cos(np.linspace(start=0, stop=20, num=1000))
events = nk.events_find(signal)
data = nk.events_plot(events, signal, show=False)
assert len(data["Event_Onset"]) == 1000
# Different events
events1 = events["onset"]
events2 = np.linspace(0, len(signal), 8)
nk.events_plot([events1, events2], signal)
fig = plt.gcf()
for ax in fig.get_axes():
handles, labels = ax.get_legend_handles_labels()
assert len(handles) == len(events1) + len(events2) + 1
assert len(labels) == len(handles)
plt.close(fig)
# Different conditions
events = nk.events_find(signal, event_conditions=["A", "B", "A", "B"])
nk.events_plot(events, signal)
fig = plt.gcf()
for ax in fig.get_axes():
handles, labels = ax.get_legend_handles_labels()
assert len(handles) == len(events) + 1
assert len(labels) == len(handles)
plt.close(fig)