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
|
# Authors: The MNE-Python contributors.
# License: BSD-3-Clause
# Copyright the MNE-Python contributors.
import matplotlib
import pytest
from mne.viz import plot_channel_labels_circle
@pytest.mark.filterwarnings(
"ignore:invalid value encountered in greater_equal:RuntimeWarning"
)
def test_plot_channel_labels_circle():
"""Test plotting channel labels in a circle."""
fig, axes = plot_channel_labels_circle(
dict(brain=["big", "great", "smart"]),
colors=dict(big="r", great="y", smart="b"),
)
texts = [
child.get_text()
for child in axes.get_children()
if isinstance(child, matplotlib.text.Text)
]
for text in ("brain", "big", "great", "smart"):
assert text in texts
# check inputs
with pytest.raises(ValueError, match="No color provided"):
plot_channel_labels_circle(
dict(brain=["big", "great", "smart"]), colors=dict(big="r", great="y")
)
|
