BreezeStyleSheets/test/ui.py

1032 lines
34 KiB
Python

#!/usr/bin/env python
#
# The MIT License (MIT)
#
# Copyright (c) <2021> <Alex Huszagh>
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the 'Software'), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED 'AS IS', WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
# THE SOFTWARE.
'''
test
====
Test styles of a single widget.
'''
import argparse
import gc
import os
import random
import sys
import time
tests_dir = os.path.dirname(os.path.realpath(__file__))
home = os.path.dirname(tests_dir)
# Create our arguments.
parser = argparse.ArgumentParser(description='Configurations for the Qt5 application.')
parser.add_argument(
'--widget',
help='''widget to test. can provide `all` to test all'''
)
parser.add_argument(
'--stylesheet',
help='''stylesheet name''',
default='native'
)
# Know working styles include:
# 1. Fusion
# 2. Windows
parser.add_argument(
'--style',
help='''application style, which is different than the stylesheet''',
default='native'
)
parser.add_argument(
'--font-size',
help='''font size for the application''',
type=float,
default=-1
)
parser.add_argument(
'--font-family',
help='''the font family'''
)
parser.add_argument(
'--width',
help='''the window width''',
type=int,
default=1068,
)
parser.add_argument(
'--height',
help='''the window height''',
type=int,
default=824,
)
parser.add_argument(
'--alignment',
help='''the layout alignment''',
)
parser.add_argument(
'--compress',
help='''add stretch on both sides''',
action='store_true',
)
parser.add_argument(
'--scale',
help='''scale factor for the UI''',
type=float,
default=1,
)
parser.add_argument(
'--pyqt6',
help='''use PyQt6 rather than PyQt5.''',
action='store_true'
)
parser.add_argument(
'--use-x11',
help='''force the use of x11 on compatible systems.''',
action='store_true'
)
args, unknown = parser.parse_known_args()
if args.pyqt6:
from PyQt6 import QtCore, QtGui, QtWidgets
QtCore.QDir.addSearchPath(args.stylesheet, f'{home}/pyqt6/{args.stylesheet}/')
style_prefix = f'{args.stylesheet}:'
stylesheet = f'{style_prefix}stylesheet.qss'
else:
sys.path.insert(0, home)
from PyQt5 import QtCore, QtGui, QtWidgets
import breeze_resources
style_prefix = f':/{args.stylesheet}/'
stylesheet = f'{style_prefix}stylesheet.qss'
# Compat definitions, between Qt5 and Qt6.
if args.pyqt6:
QAction = QtGui.QAction
AlignTop = QtCore.Qt.AlignmentFlag.AlignTop
AlignVCenter = QtCore.Qt.AlignmentFlag.AlignVCenter
AlignBottom = QtCore.Qt.AlignmentFlag.AlignBottom
AlignLeft = QtCore.Qt.AlignmentFlag.AlignLeft
AlignHCenter = QtCore.Qt.AlignmentFlag.AlignHCenter
AlignRight = QtCore.Qt.AlignmentFlag.AlignRight
AlignCenter = QtCore.Qt.AlignmentFlag.AlignCenter
Horizontal = QtCore.Qt.Orientation.Horizontal
Vertical = QtCore.Qt.Orientation.Vertical
ScrollBarAsNeeded = QtCore.Qt.ScrollBarPolicy.ScrollBarAsNeeded
ReadOnly = QtCore.QFile.OpenModeFlag.ReadOnly
Text = QtCore.QFile.OpenModeFlag.Text
KeepAspectRatio = QtCore.Qt.AspectRatioMode.KeepAspectRatio
MenuButtonPopup = QtWidgets.QToolButton.ToolButtonPopupMode.MenuButtonPopup
InstantPopup = QtWidgets.QToolButton.ToolButtonPopupMode.InstantPopup
Checked = QtCore.Qt.CheckState.Checked
Unchecked = QtCore.Qt.CheckState.Unchecked
PartiallyChecked = QtCore.Qt.CheckState.PartiallyChecked
ItemIsUserCheckable = QtCore.Qt.ItemFlag.ItemIsUserCheckable
ItemIsUserTristate = QtCore.Qt.ItemFlag.ItemIsUserTristate
LeftArrow = QtCore.Qt.ArrowType.LeftArrow
RightArrow = QtCore.Qt.ArrowType.RightArrow
UpArrow = QtCore.Qt.ArrowType.UpArrow
DownArrow = QtCore.Qt.ArrowType.DownArrow
TopToolBarArea = QtCore.Qt.ToolBarArea.TopToolBarArea
LeftToolBarArea = QtCore.Qt.ToolBarArea.LeftToolBarArea
LeftDockWidgetArea = QtCore.Qt.DockWidgetArea.LeftDockWidgetArea
North = QtWidgets.QTabWidget.TabPosition.North
West = QtWidgets.QTabWidget.TabPosition.West
East = QtWidgets.QTabWidget.TabPosition.East
South = QtWidgets.QTabWidget.TabPosition.South
Ok = QtWidgets.QMessageBox.StandardButton.Ok
Cancel = QtWidgets.QMessageBox.StandardButton.Cancel
Close = QtWidgets.QMessageBox.StandardButton.Close
Open = QtWidgets.QMessageBox.StandardButton.Open
Reset = QtWidgets.QMessageBox.StandardButton.Reset
Save = QtWidgets.QMessageBox.StandardButton.Save
SaveAll = QtWidgets.QMessageBox.StandardButton.SaveAll
RestoreDefaults = QtWidgets.QMessageBox.StandardButton.RestoreDefaults
Yes = QtWidgets.QMessageBox.StandardButton.Yes
Help = QtWidgets.QMessageBox.StandardButton.Help
No = QtWidgets.QMessageBox.StandardButton.No
Apply = QtWidgets.QMessageBox.StandardButton.Apply
Discard = QtWidgets.QMessageBox.StandardButton.Discard
Critical = QtWidgets.QMessageBox.Icon.Critical
Information = QtWidgets.QMessageBox.Icon.Information
NoIcon = QtWidgets.QMessageBox.Icon.NoIcon
Question = QtWidgets.QMessageBox.Icon.Question
Warning = QtWidgets.QMessageBox.Icon.Warning
YesRole = QtWidgets.QDialogButtonBox.ButtonRole.YesRole
DialogOk = QtWidgets.QDialogButtonBox.StandardButton.Ok
DialogCancel = QtWidgets.QDialogButtonBox.StandardButton.Cancel
DockWidgetClosable = QtWidgets.QDockWidget.DockWidgetFeature.DockWidgetClosable
DockWidgetFloatable = QtWidgets.QDockWidget.DockWidgetFeature.DockWidgetFloatable
DockWidgetMovable = QtWidgets.QDockWidget.DockWidgetFeature.DockWidgetMovable
AllDockWidgetFeatures = DockWidgetClosable | DockWidgetFloatable | DockWidgetMovable
else:
QAction = QtWidgets.QAction
AlignTop = QtCore.Qt.AlignTop
AlignVCenter = QtCore.Qt.AlignVCenter
AlignBottom = QtCore.Qt.AlignBottom
AlignLeft = QtCore.Qt.AlignLeft
AlignHCenter = QtCore.Qt.AlignHCenter
AlignRight = QtCore.Qt.AlignRight
AlignCenter = QtCore.Qt.AlignCenter
Horizontal = QtCore.Qt.Horizontal
Vertical = QtCore.Qt.Vertical
ScrollBarAsNeeded = QtCore.Qt.ScrollBarAsNeeded
ReadOnly = QtCore.QFile.ReadOnly
Text = QtCore.QFile.Text
KeepAspectRatio = QtCore.Qt.KeepAspectRatio
MenuButtonPopup = QtWidgets.QToolButton.MenuButtonPopup
InstantPopup = QtWidgets.QToolButton.InstantPopup
Checked = QtCore.Qt.Checked
Unchecked = QtCore.Qt.Unchecked
PartiallyChecked = QtCore.Qt.PartiallyChecked
ItemIsUserCheckable = QtCore.Qt.ItemIsUserCheckable
ItemIsUserTristate = QtCore.Qt.ItemIsUserTristate
LeftArrow = QtCore.Qt.LeftArrow
RightArrow = QtCore.Qt.RightArrow
UpArrow = QtCore.Qt.UpArrow
DownArrow = QtCore.Qt.DownArrow
TopToolBarArea = QtCore.Qt.TopToolBarArea
LeftToolBarArea = QtCore.Qt.LeftToolBarArea
LeftDockWidgetArea = QtCore.Qt.LeftDockWidgetArea
North = QtWidgets.QTabWidget.North
West = QtWidgets.QTabWidget.West
East = QtWidgets.QTabWidget.East
South = QtWidgets.QTabWidget.South
Ok = QtWidgets.QMessageBox.Ok
Cancel = QtWidgets.QMessageBox.Cancel
Close = QtWidgets.QMessageBox.Close
Open = QtWidgets.QMessageBox.Open
Reset = QtWidgets.QMessageBox.Reset
Save = QtWidgets.QMessageBox.Save
SaveAll = QtWidgets.QMessageBox.SaveAll
RestoreDefaults = QtWidgets.QMessageBox.RestoreDefaults
Yes = QtWidgets.QMessageBox.Yes
Help = QtWidgets.QMessageBox.Help
No = QtWidgets.QMessageBox.No
Apply = QtWidgets.QMessageBox.Apply
Discard = QtWidgets.QMessageBox.Discard
Critical = QtWidgets.QMessageBox.Critical
Information = QtWidgets.QMessageBox.Information
NoIcon = QtWidgets.QMessageBox.NoIcon
Question = QtWidgets.QMessageBox.Question
Warning = QtWidgets.QMessageBox.Warning
YesRole = QtWidgets.QDialogButtonBox.YesRole
DialogOk = QtWidgets.QDialogButtonBox.Ok
DialogCancel = QtWidgets.QDialogButtonBox.Cancel
AllDockWidgetFeatures = QtWidgets.QDockWidget.AllDockWidgetFeatures
# Need to fix an issue on Wayland on Linux:
# conda-forge does not support Wayland, for who knows what reason.
if sys.platform.lower().startswith('linux') and 'CONDA_PREFIX' in os.environ:
args.use_x11 = True
if args.use_x11:
os.environ['XDG_SESSION_TYPE'] = 'x11'
layout = {
'vertical': QtWidgets.QVBoxLayout,
'horizontal': QtWidgets.QHBoxLayout,
}
alignment = {
'top': AlignTop,
'vcenter': AlignVCenter,
'bottom': AlignBottom,
'left': AlignLeft,
'hcenter': AlignHCenter,
'right': AlignRight,
'center': AlignCenter,
}
def add_widgets(layout, children):
'''Add 1 or more widgets to the layout.'''
if isinstance(children, list):
for child in children:
layout.addWidget(child)
else:
layout.addWidget(children)
def abstract_button(
cls,
parent=None,
*args,
exlusive=False,
checked=False,
checkable=True,
enabled=True,
):
'''Helper to simplify creating abstract buttons.'''
inst = cls(*args, parent)
inst.setAutoExclusive(exlusive)
inst.setCheckable(checkable)
if isinstance(checked, bool):
inst.setChecked(checked)
else:
inst.setTristate(True)
inst.setCheckState(checked)
inst.setEnabled(enabled)
return inst
def splash_timer(splash, window):
'''Non-block timer for a splashscreen.'''
splash.finish(window)
window.show()
def test_progressbar_horizontal(widget, *_):
child = []
bar1 = QtWidgets.QProgressBar(widget)
bar1.setProperty('value', 0)
child.append(bar1)
bar2 = QtWidgets.QProgressBar(widget)
bar2.setProperty('value', 24)
child.append(bar2)
bar3 = QtWidgets.QProgressBar(widget)
bar3.setProperty('value', 99)
child.append(bar3)
bar4 = QtWidgets.QProgressBar(widget)
bar4.setProperty('value', 100)
child.append(bar4)
return child
def test_progressbar_vertical(widget, *_):
layout_type = 'horizontal'
child = []
bar1 = QtWidgets.QProgressBar(widget)
bar1.setOrientation(Vertical)
bar1.setProperty('value', 0)
child.append(bar1)
bar2 = QtWidgets.QProgressBar(widget)
bar2.setOrientation(Vertical)
bar2.setProperty('value', 24)
child.append(bar2)
bar3 = QtWidgets.QProgressBar(widget)
bar3.setOrientation(Vertical)
bar3.setProperty('value', 99)
child.append(bar3)
bar4 = QtWidgets.QProgressBar(widget)
bar4.setOrientation(Vertical)
bar4.setProperty('value', 100)
child.append(bar4)
return child, layout_type
def test_slider_horizontal(widget, *_):
child = QtWidgets.QSlider(widget)
child.setOrientation(Horizontal)
return child
def test_slider_vertical(widget, *_):
layout_type = 'horizontal'
child = QtWidgets.QSlider(widget)
child.setOrientation(Vertical)
return child, layout_type
def test_splitter_horizontal(widget, *_):
child = QtWidgets.QSplitter(widget)
child.addWidget(QtWidgets.QListWidget())
child.addWidget(QtWidgets.QTreeWidget())
child.addWidget(QtWidgets.QTextEdit())
return child
def test_splitter_vertical(widget, *_):
layout_type = 'horizontal'
child = QtWidgets.QSplitter(widget)
child.setOrientation(Vertical)
child.addWidget(QtWidgets.QListWidget())
child.addWidget(QtWidgets.QTreeWidget())
child.addWidget(QtWidgets.QTextEdit())
return child, layout_type
def test_menu(widget, window, font, width, *_):
child = QtWidgets.QMenuBar(window)
child.setGeometry(QtCore.QRect(0, 0, width, int(1.5 * font.pointSize())))
menu = QtWidgets.QMenu('Main Menu', child)
menu.addAction(QAction('&Action 1', window))
menu.addAction(QAction('&Action 2', window))
submenu = QtWidgets.QMenu('Sub Menu', menu)
submenu.addAction(QAction('&Action 3', window))
action1 = QAction('&Action 4', window)
action1.setCheckable(True)
submenu.addAction(action1)
menu.addAction(submenu.menuAction())
action2 = QAction('&Action 5', window)
action2.setCheckable(True)
action2.setChecked(True)
menu.addSeparator()
menu.addAction(action2)
action3 = QAction('&Action 6', window)
action3.setCheckable(True)
menu.addAction(action3)
icon = QtGui.QIcon(f'{style_prefix}close.svg')
menu.addAction(QAction(icon, '&Action 7', window))
menu.addAction(QAction(icon, '&Action 8', window))
submenu.addAction(QAction(icon, '&Action 9', window))
child.addAction(menu.menuAction())
window.setMenuBar(child)
return child
def test_statusbar(_, window, *__):
child = QtWidgets.QStatusBar(window)
window.setStatusBar(child)
return child
def test_spinbox(widget, *_):
layout_type = 'horizontal'
child = []
spin1 = QtWidgets.QSpinBox(widget)
spin1.setValue(10)
child.append(spin1)
spin2 = QtWidgets.QSpinBox(widget)
spin2.setValue(10)
spin2.setEnabled(False)
child.append(spin2)
return child, layout_type
def test_double_spinbox(widget, *_):
layout_type = 'horizontal'
child = []
spin1 = QtWidgets.QDoubleSpinBox(widget)
spin1.setValue(10.5)
child.append(spin1)
spin2 = QtWidgets.QDoubleSpinBox(widget)
spin2.setValue(10.5)
spin2.setEnabled(False)
child.append(spin2)
return child, layout_type
def test_combobox(widget, *_):
layout_type = 'horizontal'
child = []
combo1 = QtWidgets.QComboBox(widget)
combo1.addItem('Item 1')
combo1.addItem('Item 2')
child.append(combo1)
combo2 = QtWidgets.QComboBox(widget)
combo2.addItem('Very Very Long Item 1')
combo2.addItem('Very Very Long Item 2')
child.append(combo2)
combo3 = QtWidgets.QComboBox(widget)
combo3.setEditable(True)
combo3.addItem('Edit 1')
combo3.addItem('Edit 2')
child.append(combo3)
return child, layout_type
def _test_tabwidget(widget, position):
child = QtWidgets.QTabWidget(widget)
child.setTabPosition(position)
child.addTab(QtWidgets.QWidget(), 'Tab 1')
child.addTab(QtWidgets.QWidget(), 'Tab 2')
child.addTab(QtWidgets.QWidget(), 'Tab 3')
return child
def test_tabwidget_top(widget, *_):
return _test_tabwidget(widget, North)
def test_tabwidget_left(widget, *_):
return _test_tabwidget(widget, West)
def test_tabwidget_right(widget, *_):
return _test_tabwidget(widget, East)
def test_tabwidget_bottom(widget, *_):
return _test_tabwidget(widget, South)
def test_closable_tabwidget_top(widget, *_):
child = _test_tabwidget(widget, North)
child.setTabsClosable(True)
return child
def test_closable_tabwidget_right(widget, *_):
child = _test_tabwidget(widget, East)
child.setTabsClosable(True)
return child
def test_dock(_, window, *__):
dock1 = QtWidgets.QDockWidget('&Dock widget 1', window)
dock1.setFeatures(AllDockWidgetFeatures)
dock2 = QtWidgets.QDockWidget('&Dock widget 2', window)
dock2.setFeatures(AllDockWidgetFeatures)
window.addDockWidget(LeftDockWidgetArea, dock1)
window.addDockWidget(LeftDockWidgetArea, dock2)
window.tabifyDockWidget(dock1, dock2)
def test_radio(widget, *_):
child = []
widget_type = QtWidgets.QRadioButton
child.append(abstract_button(widget_type, widget))
child.append(abstract_button(widget_type, widget, checked=True))
child.append(abstract_button(widget_type, widget, enabled=False))
child.append(abstract_button(widget_type, widget, checked=True, enabled=False))
child.append(abstract_button(widget_type, widget, 'With Text'))
return child
def test_checkbox(widget, _, __, ___, ____, app):
child = []
widget_type = QtWidgets.QCheckBox
child.append(abstract_button(widget_type, widget))
child.append(abstract_button(widget_type, widget, checked=True))
child.append(abstract_button(widget_type, widget, checked=PartiallyChecked))
child.append(abstract_button(widget_type, widget, enabled=False))
child.append(abstract_button(widget_type, widget, checked=True, enabled=False))
child.append(abstract_button(widget_type, widget, checked=PartiallyChecked, enabled=False))
child.append(abstract_button(widget_type, widget, 'With Text'))
child.append(abstract_button(widget_type, widget, 'With Large Text'))
checkbox_font = app.font()
checkbox_font.setPointSizeF(50.0)
child[-1].setFont(checkbox_font)
return child
def test_table(widget, *_):
child = QtWidgets.QTableWidget(widget)
child.setColumnCount(2)
child.setRowCount(4)
item = QtWidgets.QTableWidgetItem('Row 1')
child.setVerticalHeaderItem(0, item)
item = QtWidgets.QTableWidgetItem('Row 2')
child.setVerticalHeaderItem(1, item)
item = QtWidgets.QTableWidgetItem('Row 3')
child.setVerticalHeaderItem(2, item)
item = QtWidgets.QTableWidgetItem('Row 4')
child.setVerticalHeaderItem(3, item)
item = QtWidgets.QTableWidgetItem('Column 1')
child.setHorizontalHeaderItem(0, item)
item = QtWidgets.QTableWidgetItem('Column 2')
child.setHorizontalHeaderItem(1, item)
return child
def test_sortable_table(widget, *_):
child = test_table(widget)
child.setSortingEnabled(True)
return child
def test_list(widget, *_):
alignments = [AlignLeft, AlignRight, AlignHCenter]
child = QtWidgets.QListWidget(widget)
for index in range(10):
item = QtWidgets.QListWidgetItem(f'Item {index + 1}')
item.setTextAlignment(random.choice(alignments))
child.addItem(item)
icon = QtGui.QIcon(':/dark/close.svg')
for index in range(10):
item = QtWidgets.QListWidgetItem(icon, f'Item {index + 1}')
item.setTextAlignment(random.choice(alignments))
child.addItem(item)
return child
def test_scrollbar_vertical(widget, *_):
child = QtWidgets.QListWidget(widget)
for index in range(100):
child.addItem(QtWidgets.QListWidgetItem(f'Item {index + 1}'))
return child
def test_scrollbar_horizontal(widget, *_):
child = QtWidgets.QTableWidget(widget)
child.setColumnCount(100)
child.setRowCount(1)
item = QtWidgets.QTableWidgetItem(f'Row 1')
child.setVerticalHeaderItem(0, item)
for index in range(100):
item = QtWidgets.QTableWidgetItem(f'Column {index + 1}')
child.setHorizontalHeaderItem(index, item)
return child
def test_toolbar(_, window, *__):
toolbar1 = QtWidgets.QToolBar('Toolbar')
toolbar1.addAction('&Action 1')
toolbar1.addAction('&Action 2')
toolbar1.addSeparator()
toolbar1.addAction('&Action 3')
toolbar1.addAction('&Action 3 Really Long Name')
icon = QtGui.QIcon(':/dark/close.svg')
toolbar1.addAction(icon, '&Action 4')
window.addToolBar(TopToolBarArea, toolbar1)
toolbar2 = QtWidgets.QToolBar('Toolbar')
toolbar2.setOrientation(Vertical)
toolbar2.addAction('&Action 1')
toolbar2.addAction('&Action 2')
toolbar2.addSeparator()
toolbar2.addAction('&Action 3')
toolbar2.addAction('&Action 3 Really Long Name')
icon = QtGui.QIcon(':/dark/close.svg')
toolbar2.addAction(icon, '&Action 4')
window.addToolBar(LeftToolBarArea, toolbar2)
return None, None
def test_toolbutton(widget, window, *_):
layout_type = 'horizontal'
child = [
QtWidgets.QToolButton(widget),
QtWidgets.QToolButton(widget),
QtWidgets.QToolButton(widget),
QtWidgets.QToolButton(widget),
QtWidgets.QToolButton(widget),
QtWidgets.QToolButton(widget),
QtWidgets.QToolButton(widget),
QtWidgets.QToolButton(widget),
QtWidgets.QToolButton(widget),
]
window.setTabOrder(child[0], child[1])
window.setTabOrder(child[1], child[2])
window.setTabOrder(child[2], child[3])
window.setTabOrder(child[3], child[4])
window.setTabOrder(child[4], child[5])
window.setTabOrder(child[5], child[6])
window.setTabOrder(child[6], child[7])
child[0].setText('Simple ToolButton')
child[1].setText('Action Toolbutton')
child[2].setText('Menu Toolbutton')
child[3].setText('Instant Toolbutton')
child[1].addActions([
QAction('&Action 5', window),
QAction('&Action 6', window),
])
child[2].setPopupMode(MenuButtonPopup)
child[2].addActions([
QAction('&Action 9', window),
QAction('&Action 10', window),
])
child[3].setPopupMode(InstantPopup)
child[3].addActions([
QAction('&Action 11', window),
QAction('&Action 12', window),
])
child[4].setArrowType(LeftArrow)
child[5].setArrowType(RightArrow)
child[6].setArrowType(UpArrow)
child[7].setArrowType(DownArrow)
icon = QtGui.QIcon(':/dark/close.svg')
child[8].setIcon(icon)
return child, layout_type
def test_pushbutton(widget, *_):
layout_type = 'horizontal'
child = []
widget_type = QtWidgets.QPushButton
child.append(abstract_button(widget_type, widget, 'Button 1', checked=True))
child.append(abstract_button(widget_type, widget, 'Button 2', enabled=False))
child.append(abstract_button(widget_type, widget, 'Button 3', checkable=False))
icon = QtGui.QIcon(':/dark/close.svg')
child.append(abstract_button(widget_type, widget, icon, 'Button 4', checkable=False))
return child, layout_type
def test_tree(widget, *_):
child = []
tree1 = QtWidgets.QTreeWidget(widget)
tree1.setHeaderLabel('Tree 1')
item1 = QtWidgets.QTreeWidgetItem(tree1, ['Row 1'])
item2 = QtWidgets.QTreeWidgetItem(tree1, ['Row 2'])
item3 = QtWidgets.QTreeWidgetItem(item2, ['Row 2.1'])
item3.setFlags(item3.flags() | ItemIsUserCheckable)
item3.setCheckState(0, Unchecked)
item4 = QtWidgets.QTreeWidgetItem(item2, ['Row 2.2'])
item5 = QtWidgets.QTreeWidgetItem(item4, ['Row 2.2.1'])
item6 = QtWidgets.QTreeWidgetItem(item5, ['Row 2.2.1.1'])
item7 = QtWidgets.QTreeWidgetItem(item5, ['Row 2.2.1.2'])
item7.setFlags(item7.flags() | ItemIsUserCheckable)
item7.setCheckState(0, Checked)
item8 = QtWidgets.QTreeWidgetItem(item2, ['Row 2.3'])
item8.setFlags(item8.flags() | ItemIsUserTristate)
item8.setCheckState(0, PartiallyChecked)
item9 = QtWidgets.QTreeWidgetItem(tree1, ['Row 3'])
item10 = QtWidgets.QTreeWidgetItem(item9, ['Row 3.1'])
item11 = QtWidgets.QTreeWidgetItem(tree1, ['Row 4'])
child.append(tree1)
tree2 = QtWidgets.QTreeWidget(widget)
tree2.setHeaderLabel('Tree 2')
tree2.header().setSectionsClickable(True)
item12 = QtWidgets.QTreeWidgetItem(tree2, ['Row 1', 'Column 2', 'Column 3'])
child.append(tree2)
return child
def test_view_scrollarea(widget, *_):
# For us to have both scrollbars visible.
child = QtWidgets.QTableWidget(widget)
child.setColumnCount(100)
child.setRowCount(100)
for index in range(100):
row = QtWidgets.QTableWidgetItem(f'Row {index + 1}')
child.setVerticalHeaderItem(0, row)
column = QtWidgets.QTableWidgetItem(f'Column {index + 1}')
child.setHorizontalHeaderItem(index, column)
return child
def test_widget_scrollarea(widget, window, *_):
child = QtWidgets.QProgressBar(widget)
child.setProperty('value', 24)
window.resize(30, 30)
return child
def test_frame(widget, *_):
child = []
text = QtWidgets.QTextEdit()
text.setPlainText('Hello world\nTesting lines')
child.append(text)
table = QtWidgets.QTableWidget()
table.setColumnCount(5)
table.setRowCount(5)
child.append(table)
return child
def test_groupbox(widget, *_):
child = []
child.append(QtWidgets.QGroupBox('Groupbox 1', widget))
checkable = QtWidgets.QGroupBox('Groupbox 2', widget)
checkable.setCheckable(True)
child.append(checkable)
vbox = QtWidgets.QVBoxLayout(checkable)
vbox.setAlignment(AlignHCenter)
vbox.addWidget(QtWidgets.QLineEdit('Sample Label'))
return child
def test_toolbox(widget, *_):
# Test alignment with another item, in a vertical layout.
child = []
child.append(QtWidgets.QGroupBox('Groupbox', widget))
child.append(QtWidgets.QGroupBox('Really, really long groupbox', widget))
toolbox = QtWidgets.QToolBox(widget)
child.append(toolbox)
page1 = QtWidgets.QWidget()
toolbox.addItem(page1, 'Page 1')
page2 = QtWidgets.QWidget()
vbox = QtWidgets.QVBoxLayout(page2)
vbox.addWidget(QtWidgets.QLabel('Sample Label'))
toolbox.addItem(page2, 'Page 2')
page3 = QtWidgets.QWidget()
toolbox.addItem(page3, 'Really, really long page 3')
return child
def test_menubutton(widget, window, *_):
child = QtWidgets.QToolButton(widget)
child.setText('Menu Toolbutton')
child.setPopupMode(MenuButtonPopup)
child.addActions([
QAction('&Action 9', window),
QAction('&Action 10', window),
])
return child
def test_tooltip(widget, *_):
child = QtWidgets.QPushButton('Sample Label')
child.setToolTip('Sample Tooltip')
return child
def test_splashscreen(_, window, __, ___, ____, app):
pixmap = QtGui.QPixmap('assets/Yellowstone.jpg')
size = app.screens()[0].size()
scaled = pixmap.scaled(size, KeepAspectRatio)
splash = QtWidgets.QSplashScreen(scaled)
splash.show()
QtCore.QTimer.singleShot(2000, lambda: splash_timer(splash, window))
return None, None, False
def test_calendar(widget, *_):
child = QtWidgets.QCalendarWidget(widget)
child.setGridVisible(True)
return child
def _test_standard_button(window, app, button):
message = QtWidgets.QMessageBox(window)
message.addButton(button)
if args.pyqt6:
message.exec()
else:
message.exec_()
def test_ok_button(_, window, __, ___, ____, app):
_test_standard_button(window, app, Ok)
return None, None, False, True
def test_cancel_button(_, window, __, ___, ____, app):
_test_standard_button(window, app, Cancel)
return None, None, False, True
def test_close_button(_, window, __, ___, ____, app):
_test_standard_button(window, app, Close)
return None, None, False, True
def test_open_button(_, window, __, ___, ____, app):
_test_standard_button(window, app, Open)
return None, None, False, True
def test_reset_button(_, window, __, ___, ____, app):
_test_standard_button(window, app, Reset)
return None, None, False, True
def test_save_button(_, window, __, ___, ____, app):
_test_standard_button(window, app, Save)
return None, None, False, True
def test_saveall_button(_, window, __, ___, ____, app):
_test_standard_button(window, app, SaveAll)
return None, None, False, True
def test_restoredefaults_button(_, window, __, ___, ____, app):
_test_standard_button(window, app, RestoreDefaults)
return None, None, False, True
def test_yes_button(_, window, __, ___, ____, app):
_test_standard_button(window, app, Yes)
return None, None, False, True
def test_help_button(_, window, __, ___, ____, app):
_test_standard_button(window, app, Help)
return None, None, False, True
def test_no_button(_, window, __, ___, ____, app):
_test_standard_button(window, app, No)
return None, None, False, True
def test_apply_button(_, window, __, ___, ____, app):
_test_standard_button(window, app, Apply)
return None, None, False, True
def test_discard_button(_, window, __, ___, ____, app):
_test_standard_button(window, app, Discard)
return None, None, False, True
def _test_standard_icon(window, app, icon):
message = QtWidgets.QMessageBox(window)
message.setIcon(icon)
if args.pyqt6:
message.exec()
else:
message.exec_()
def test_critical_icon(_, window, __, ___, ____, app):
_test_standard_icon(window, app, Critical)
return None, None, False, True
def test_info_icon(_, window, __, ___, ____, app):
_test_standard_icon(window, app, Information)
return None, None, False, True
def test_no_icon(_, window, __, ___, ____, app):
_test_standard_icon(window, app, NoIcon)
return None, None, False, True
def test_question_icon(_, window, __, ___, ____, app):
_test_standard_icon(window, app, Question)
return None, None, False, True
def test_warning_icon(_, window, __, ___, ____, app):
_test_standard_icon(window, app, Warning)
return None, None, False, True
def test_multiple_buttons(widget, *_):
child = []
child.append(QtWidgets.QTextEdit(widget))
container = QtWidgets.QWidget(widget)
hbox = QtWidgets.QHBoxLayout(container)
hbox.addWidget(QtWidgets.QPushButton('Delete'))
hbox.addWidget(QtWidgets.QPushButton('Complete'))
child.append(container)
child.append(QtWidgets.QLineEdit(widget))
dialog = QtWidgets.QDialogButtonBox(Horizontal, widget)
dialog.addButton('Yes', YesRole)
dialog.addButton('Really really really long', YesRole)
dialog.addButton(DialogOk)
dialog.addButton(DialogCancel)
child.append(dialog)
return child
def test_disabled_menu(widget, window, font, width, *_):
child = QtWidgets.QMenuBar(window)
child.setGeometry(QtCore.QRect(0, 0, width, int(1.5 * font.pointSize())))
menu = QtWidgets.QMenu('Main Menu', child)
menu.addAction(QAction('&Action 1', window))
menu.addAction(QAction('&Action 2', window))
submenu = QtWidgets.QMenu('Sub Menu', menu)
submenu.addAction(QAction('&Action 3', window))
action1 = QAction('&Action 4', window)
action1.setCheckable(True)
action1.setEnabled(False)
submenu.addAction(action1)
menu.addAction(submenu.menuAction())
action2 = QAction('&Action 5', window)
action2.setCheckable(True)
action2.setChecked(True)
menu.addSeparator()
menu.addAction(action2)
action3 = QAction('&Action 6', window)
action3.setCheckable(True)
menu.addAction(action3)
icon = QtGui.QIcon(':/dark/close.svg')
menu.addAction(QAction(icon, '&Action 7', window))
menu.addAction(QAction(icon, '&Action 8', window))
menu.actions()[2].setEnabled(False)
submenu.addAction(QAction(icon, '&Action 9', window))
child.addAction(menu.menuAction())
window.setMenuBar(child)
return child
def test_disabled_menubar(widget, window, font, width, *_):
child = QtWidgets.QMenuBar(window)
child.setGeometry(QtCore.QRect(0, 0, width, int(1.5 * font.pointSize())))
menu = QtWidgets.QMenu('Main Menu', child)
child.addAction(menu.menuAction())
window.setMenuBar(child)
menu.setEnabled(False)
return child
def test(args, qtargv, test_widget):
'''Test a single widget.'''
app = QtWidgets.QApplication(qtargv)
# use the default font size
font = app.font()
if args.font_size > 0:
font.setPointSizeF(args.font_size)
if args.font_family:
font.setFamily(args.font_family)
app.setFont(font)
# setup stylesheet
if args.stylesheet != 'native':
file = QtCore.QFile(stylesheet)
file.open(ReadOnly | Text)
stream = QtCore.QTextStream(file)
app.setStyleSheet(stream.readAll())
# Setup the main window.
window = QtWidgets.QMainWindow()
window.setWindowTitle('Sample single widget application.')
window.resize(args.width, args.height)
widget = QtWidgets.QWidget()
scroll = QtWidgets.QScrollArea()
scroll.setHorizontalScrollBarPolicy(ScrollBarAsNeeded)
scroll.setVerticalScrollBarPolicy(ScrollBarAsNeeded)
scroll.setWidgetResizable(True)
# Get the correct parameters for our test widget.
try:
function = globals()[f'test_{test_widget}']
except KeyError:
raise NotImplementedError(f'test for {test_widget} not implemented')
result = function(widget, window, font, args.width, args.height, app)
child = []
layout_type = 'vertical'
show_window = True
quit = False
if result and isinstance(result, list):
# Have a single value passed as a list
child = result
elif isinstance(result, tuple):
child = result[0]
else:
child = result
if isinstance(result, tuple) and len(result) >= 2:
layout_type = result[1]
if isinstance(result, tuple) and len(result) >= 3:
show_window = result[2]
if isinstance(result, tuple) and len(result) >= 4:
quit = result[3]
# Add the widgets to the layout.
if layout_type is not None and child is not None:
widget_layout = layout[layout_type]()
if args.compress:
widget_layout.addStretch(1)
add_widgets(widget_layout, child)
widget_layout.addStretch(1)
else:
add_widgets(widget_layout, child)
if args.alignment is not None:
widget_layout.setAlignment(alignment[args.alignment])
widget.setLayout(widget_layout)
scroll.setWidget(widget)
window.setCentralWidget(scroll)
# run
if show_window:
window.show()
if quit:
return app.quit()
if args.pyqt6:
return app.exec()
else:
return app.exec_()
def main():
'Application entry point'
# Disable garbage collection to avoid runtime errors.
gc.disable()
os.environ['QT_SCALE_FACTOR'] = str(args.scale)
if args.style != 'native':
style = QtWidgets.QStyleFactory.create(args.style)
QtWidgets.QApplication.setStyle(style)
if args.widget == 'all':
all_tests = [i for i in globals().keys() if i.startswith('test_')]
all_widgets = [i[len('test_'):] for i in all_tests]
for widget in all_widgets:
test(args, sys.argv[:1] + unknown, widget)
gc.collect()
else:
test(args, sys.argv[:1] + unknown, args.widget)
return 0
if __name__ == '__main__':
sys.exit(main())