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
|
# Authors: The MNE-Python contributors.
# License: BSD-3-Clause
# Copyright the MNE-Python contributors.
import platform
from colorsys import rgb_to_hls
import numpy as np
import pytest
from mne import create_info
from mne.io import RawArray
from mne.utils import _check_qt_version
from mne.viz.backends._utils import (
_check_color,
_get_colormap_from_array,
_pixmap_to_ndarray,
_qt_is_dark,
)
def test_get_colormap_from_array():
"""Test setting a colormap."""
from matplotlib.colors import LinearSegmentedColormap, ListedColormap
cmap = _get_colormap_from_array()
assert isinstance(cmap, LinearSegmentedColormap)
cmap = _get_colormap_from_array(colormap="viridis")
assert isinstance(cmap, ListedColormap)
cmap = _get_colormap_from_array(colormap=[1, 1, 1], normalized_colormap=True)
assert isinstance(cmap, ListedColormap)
cmap = _get_colormap_from_array(colormap=[255, 255, 255], normalized_colormap=False)
assert isinstance(cmap, ListedColormap)
def test_check_color():
"""Test color format."""
assert _check_color("red") == (1.0, 0.0, 0.0)
assert _check_color((0.0, 1.0, 0.0, 1.0)) == (0.0, 1.0, 0.0, 1.0)
assert _check_color((0, 0, 255, 255)) == (0, 0, 255, 255)
with pytest.raises(ValueError, match="RGB or RGBA"):
_check_color([255, 0])
with pytest.raises(ValueError, match="out of range"):
_check_color([256, 0, 0])
with pytest.raises(ValueError, match="out of range"):
_check_color([-1.0, 0.0, 0.0])
with pytest.raises(TypeError, match="Expected data type"):
_check_color(["foo", "bar", "foo"])
with pytest.raises(TypeError, match="Expected type"):
_check_color(None)
@pytest.mark.pgtest
@pytest.mark.parametrize("theme", ("auto", "light", "dark"))
def test_theme_colors(pg_backend, theme, monkeypatch, tmp_path):
"""Test that theme colors propagate properly."""
darkdetect = pytest.importorskip("darkdetect")
monkeypatch.setenv("_MNE_FAKE_HOME_DIR", str(tmp_path))
monkeypatch.delenv("MNE_BROWSER_THEME", raising=False)
# make it seem like the system is always in light mode
monkeypatch.setattr(darkdetect, "theme", lambda: "light")
raw = RawArray(np.zeros((1, 1000)), create_info(1, 1000.0, "eeg"))
_, api = _check_qt_version(return_api=True)
fig = raw.plot(theme=theme)
is_dark = _qt_is_dark(fig)
# on Darwin these checks get complicated, so don't bother for now
if platform.system() == "Darwin":
pytest.skip("Problems on macOS")
if theme == "dark":
assert is_dark, theme
elif theme == "light":
assert not is_dark, theme
def assert_correct_darkness(widget, want_dark):
__tracebackhide__ = True # noqa
# This should work, but it just picks up the parent in the errant case!
bgcolor = widget.palette().color(widget.backgroundRole()).getRgbF()[:3]
dark = rgb_to_hls(*bgcolor)[1] < 0.5
assert dark == want_dark, f"{widget} dark={dark} want_dark={want_dark}"
# ... so we use a more direct test
colors = _pixmap_to_ndarray(widget.grab())[:, :, :3]
dark = colors.mean() < 0.5
assert dark == want_dark, f"{widget} dark={dark} want_dark={want_dark}"
for widget in (fig.mne.toolbar, fig.statusBar()):
assert_correct_darkness(widget, is_dark)
|
