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
|
import pathlib
import pytest
from pytestqt.qt_compat import qt_api
@pytest.fixture
def widget(qtbot):
w = qt_api.QtWidgets.QWidget()
qtbot.addWidget(w)
w.setAttribute(qt_api.QtCore.Qt.WidgetAttribute.WA_StyledBackground)
w.setStyleSheet("background-color: magenta")
return w
def test_basic(qtbot, widget):
path = qtbot.screenshot(widget)
assert path.exists()
pixmap = qt_api.QtGui.QPixmap()
assert pixmap.load(str(path))
image = pixmap.toImage()
color = image.pixelColor(image.rect().center())
assert (color.red(), color.green(), color.blue()) == (255, 0, 255)
def test_region(qtbot, widget):
region = qt_api.QtCore.QRect(0, 0, 25, 25)
path = qtbot.screenshot(widget, region=region)
pixmap = qt_api.QtGui.QPixmap()
assert pixmap.load(str(path))
assert pixmap.rect() == region
def test_filename_class(qtbot, widget):
path = qtbot.screenshot(widget)
assert path.name == "screenshot_QWidget.png"
def test_filename_objectname(qtbot, widget):
widget.setObjectName("shotgun")
path = qtbot.screenshot(widget)
assert path.name == "screenshot_QWidget_shotgun.png"
def test_filename_suffix(qtbot, widget):
path = qtbot.screenshot(widget, suffix="before")
assert path.name == "screenshot_QWidget_before.png"
def test_filename_both(qtbot, widget):
widget.setObjectName("shotgun")
path = qtbot.screenshot(widget, suffix="before")
assert path.name == "screenshot_QWidget_shotgun_before.png"
def test_filename_endless(qtbot, widget, monkeypatch):
monkeypatch.setattr(pathlib.Path, "exists", lambda _self: True)
with pytest.raises(qtbot.ScreenshotError, match="Failed to find unique filename"):
qtbot.screenshot(widget, suffix="before")
def test_filename_invalid(qtbot, widget):
with pytest.raises(qtbot.ScreenshotError, match="Saving to .* failed"):
qtbot.screenshot(widget, suffix=r"invalid/path\everywhere")
def test_folder(qtbot, tmp_path, widget):
path = qtbot.screenshot(widget)
assert path.parent == tmp_path
@pytest.mark.parametrize(
"existing, expected",
[
(["QLineEdit"], "QWidget"),
(["QWidget"], "QWidget_2"),
(["QWidget_2"], "QWidget"),
(["QWidget", "QWidget_2"], "QWidget_3"),
],
)
def test_filename_dedup(qtbot, widget, tmp_path, existing, expected):
for name in existing:
path = tmp_path / f"screenshot_{name}.png"
path.touch()
path = qtbot.screenshot(widget)
assert path.name == f"screenshot_{expected}.png"
|
