PyQt.QTableView 内容的单元测试

Question

我的 GUI（PyQt5、Python3.6）显示来自 SQLite 数据库的数据，主要使用 QTableViews。

我正在使用定义的输入文件对应用程序进行单元测试，并且我想检查是否所有内容都显示在 GUI 中的正确位置。现在，所有数据都在附加到 QTableView 的模型中，而不是 QTableView 本身。我有时会使用QTableView.hideRow() 来隐藏原始数据中存在但不希望在此特定视图中显示的某些列。

如何测试所有内容都显示在 QTableView 中的预期位置？

这是一个我想为其编写单元测试的迷你示例类：

#!/usr/bin/python3
# -*- coding: utf-8 -*-

from PyQt5 import QtSql
from PyQt5.QtWidgets import (QWidget, QTableView, QApplication, QHBoxLayout)
import sys

class Example(QWidget):
    def __init__(self):
        super().__init__()
        self.resize(400, 150)

        self.createConnection()
        self.fillTable()
        self.createModel()
        self.initUI()

    def createConnection(self):
        self.db = QtSql.QSqlDatabase.addDatabase("QSQLITE")
        self.db.setDatabaseName("test.db")
        if not self.db.open():
            print("Cannot establish a database connection")
            return False

    def fillTable(self):
        self.db.transaction()
        q = QtSql.QSqlQuery()

        q.exec_("DROP TABLE IF EXISTS Cars;")
        q.exec_("CREATE TABLE Cars (Company TEXT, Model TEXT, Year NUMBER);")
        q.exec_("INSERT INTO Cars VALUES ('Honda', 'Civic', 2009);")
        q.exec_("INSERT INTO Cars VALUES ('VW', 'Golf', 2013);")
        q.exec_("INSERT INTO Cars VALUES ('VW', 'Polo', 1999);")

        self.db.commit()

    def createModel(self):
        self.model = QtSql.QSqlTableModel()
        self.model.setTable("Cars")
        self.model.select()

    def initUI(self):
        layout = QHBoxLayout()
        self.setLayout(layout)

        view = QTableView()
        layout.addWidget(view)

        view.setModel(self.model)
        view.hideRow(1)

    def closeEvent(self, e):
        if (self.db.open()):
            self.db.close()


def main():
    app = QApplication(sys.argv)
    ex = Example()
    ex.show()
    sys.exit(app.exec_())


if __name__ == '__main__':
    main()

我当然可以测试模型中的数据：

self.assertEqual(model.headerData(0, Qt.Horizontal, Qt.DisplayRole), "Company")
self.assertEqual(model.data(model.index(0, 0), Qt.DisplayRole), "Honda")

等等

但是 model 包含 3 行，而我的 QTableView 只有 2 行（因为隐藏了 VW Golf 行）。

那么，有没有办法测试 QTableView 中显示的数据而不是模型中的数据？（在示例类中，我希望测试断言视图仅包含 2行，而 Polo 行是第二行。）

Answer 1

如果您想验证是否有任何索引可见，这不依赖于模型，因为模型是数据的表示，例如，模型可以由多个视图共享，因此每个视图选择要显示的项目，以知道哪些索引在 QTableView 中可见，您必须使用 isIndexHidden() 方法：

from PyQt5 import QtCore, QtWidgets, QtSql
import sys

class Example(QtWidgets.QWidget):
    def __init__(self):
        super().__init__()
        ...

    def initUI(self):
        layout = QtWidgets.QHBoxLayout()
        self.setLayout(layout)

        self.view = QtWidgets.QTableView()
        layout.addWidget(self.view)

        self.view.setModel(self.model)
        self.view.hideRow(1)
        self.items_visible()

    def items_visible(self):
        for i in range(self.model.rowCount()):
            for j in range(self.model.columnCount()):
                ix = self.model.index(i, j)
                if not self.view.isIndexHidden(ix):
                    print("{}-{}: {}".format(ix.row(), ix.column(), ix.data()))


def main():
    app = QtWidgets.QApplication(sys.argv)
    ex = Example()
    ex.show()
    sys.exit(app.exec_())


if __name__ == '__main__':
    main()