def test_report_table(self):
rep = OWReport.get_instance()
model = PyTableModel([["x", 1, 2], ["y", 2, 2]])
model.setHorizontalHeaderLabels(["a", "b", "c"])
model.setData(model.index(0, 0), Qt.AlignHCenter | Qt.AlignTop,
Qt.TextAlignmentRole)
model.setData(model.index(1, 0), QFont("", -1, QFont.Bold),
Qt.FontRole)
model.setData(model.index(1, 2), QBrush(Qt.red), Qt.BackgroundRole)
view = gui.TableView()
view.show()
view.setModel(model)
rep.report_table("Name", view)
self.maxDiff = None
self.assertEqual(
rep.report_html,
"<h2>Name</h2><table>\n"
"<tr>"
'<th style="color:black;border:0;background:transparent;'
'text-align:left;vertical-align:middle;">a</th>'
'<th style="color:black;border:0;background:transparent;'
'text-align:left;vertical-align:middle;">b</th>'
'<th style="color:black;border:0;background:transparent;'
'text-align:left;vertical-align:middle;">c</th>'
"</tr>"
"<tr>"
'<td style="color:black;border:0;background:transparent;'
'text-align:center;vertical-align:top;">x</td>'
'<td style="color:black;border:0;background:transparent;'
'text-align:right;vertical-align:middle;">1</td>'
'<td style="color:black;border:0;background:transparent;'
'text-align:right;vertical-align:middle;">2</td>'
"</tr>"
"<tr>"
'<td style="color:black;border:0;background:transparent;'
'font-weight: bold;text-align:left;vertical-align:middle;">y</td>'
'<td style="color:black;border:0;background:transparent;'
'text-align:right;vertical-align:middle;">2</td>'
'<td style="color:black;border:0;background:#ff0000;'
'text-align:right;vertical-align:middle;">2</td>'
"</tr></table>",
)
def test_report_table(self):
rep = OWReport.get_instance()
model = PyTableModel([['x', 1, 2],
['y', 2, 2]])
model.setHorizontalHeaderLabels(['a', 'b', 'c'])
model.setData(model.index(0, 0), Qt.AlignHCenter | Qt.AlignTop, Qt.TextAlignmentRole)
model.setData(model.index(1, 0), QFont('', -1, QFont.Bold), Qt.FontRole)
model.setData(model.index(1, 2), QBrush(Qt.red), Qt.BackgroundRole)
view = gui.TableView()
view.show()
view.setModel(model)
rep.report_table('Name', view)
self.maxDiff = None
self.assertEqual(
rep.report_html,
'<h2>Name</h2><table>\n'
'<tr>'
'<th style="color:black;border:0;background:transparent;'
'font-weight:normal;text-align:left;vertical-align:middle;">a</th>'
'<th style="color:black;border:0;background:transparent;'
'font-weight:normal;text-align:left;vertical-align:middle;">b</th>'
'<th style="color:black;border:0;background:transparent;'
'font-weight:normal;text-align:left;vertical-align:middle;">c</th>'
'</tr>'
'<tr>'
'<td style="color:black;border:0;background:transparent;'
'font-weight:normal;text-align:center;vertical-align:top;">x</td>'
'<td style="color:black;border:0;background:transparent;'
'font-weight:normal;text-align:right;vertical-align:middle;">1</td>'
'<td style="color:black;border:0;background:transparent;'
'font-weight:normal;text-align:right;vertical-align:middle;">2</td>'
'</tr>'
'<tr>'
'<td style="color:black;border:0;background:transparent;'
'font-weight:bold;text-align:left;vertical-align:middle;">y</td>'
'<td style="color:black;border:0;background:transparent;'
'font-weight:normal;text-align:right;vertical-align:middle;">2</td>'
'<td style="color:black;border:0;background:#ff0000;'
'font-weight:normal;text-align:right;vertical-align:middle;">2</td>'
'</tr></table>')
class TestPyTableModel(TestCase):
def setUp(self):
self.model = PyTableModel([[1, 4], [2, 3]])
def test_init(self):
self.model = PyTableModel()
self.assertEqual(self.model.rowCount(), 0)
def test_rowCount(self):
self.assertEqual(self.model.rowCount(), 2)
self.assertEqual(len(self.model), 2)
def test_columnCount(self):
self.assertEqual(self.model.columnCount(), 2)
def test_data(self):
mi = self.model.index(0, 0)
self.assertEqual(self.model.data(mi), '1')
self.assertEqual(self.model.data(mi, Qt.EditRole), 1)
def test_editable(self):
editable_model = PyTableModel([[0]], editable=True)
self.assertFalse(
int(self.model.flags(self.model.index(0, 0)) & Qt.ItemIsEditable))
self.assertTrue(
int(
editable_model.flags(editable_model.index(0, 0))
& Qt.ItemIsEditable))
def test_sort(self):
self.model.sort(1)
self.assertEqual(self.model[0][0], 2)
def test_setHeaderLabels(self):
self.model.setHorizontalHeaderLabels(['Col 1', 'Col 2'])
self.assertEqual(self.model.headerData(1, Qt.Horizontal), 'Col 2')
self.assertEqual(self.model.headerData(1, Qt.Vertical), '1')
def test_removeRows(self):
self.model.removeRows(0, 1)
self.assertEqual(len(self.model), 1)
self.assertEqual(self.model[0][1], 3)
def test_removeColumns(self):
self.model.removeColumns(0, 1)
self.assertEqual(self.model.columnCount(), 1)
self.assertEqual(self.model[1][0], 3)
def test_insertRows(self):
self.model.insertRows(0, 1)
self.assertEqual(self.model[1][0], 1)
def test_insertColumns(self):
self.model.insertColumns(0, 1)
self.assertEqual(self.model[0], ['', 1, 4])
def test_wrap(self):
self.model.wrap([[0]])
self.assertEqual(self.model.rowCount(), 1)
self.assertEqual(self.model.columnCount(), 1)
def test_clear(self):
self.model.clear()
self.assertEqual(self.model.rowCount(), 0)
def test_append(self):
self.model.append([5, 6])
self.assertEqual(self.model[2][1], 6)
self.assertEqual(self.model.rowCount(), 3)
def test_extend(self):
self.model.extend([[5, 6]])
self.assertEqual(self.model[2][1], 6)
self.assertEqual(self.model.rowCount(), 3)
def test_insert(self):
self.model.insert(0, [5, 6])
self.assertEqual(self.model[0][1], 6)
self.assertEqual(self.model.rowCount(), 3)
def test_remove(self):
self.model.remove([2, 3])
self.assertEqual(self.model.rowCount(), 1)
def test_other_roles(self):
self.model.append([2, 3])
self.model.setData(self.model.index(2, 0), Qt.AlignCenter,
Qt.TextAlignmentRole)
del self.model[1]
self.assertTrue(
Qt.AlignCenter
& self.model.data(self.model.index(1, 0), Qt.TextAlignmentRole))
class TestPyTableModel(unittest.TestCase):
def setUp(self):
self.model = PyTableModel([[1, 4], [2, 3]])
def test_init(self):
self.model = PyTableModel()
self.assertEqual(self.model.rowCount(), 0)
def test_rowCount(self):
self.assertEqual(self.model.rowCount(), 2)
self.assertEqual(len(self.model), 2)
def test_columnCount(self):
self.assertEqual(self.model.columnCount(), 2)
def test_data(self):
mi = self.model.index(0, 0)
self.assertEqual(self.model.data(mi), '1')
self.assertEqual(self.model.data(mi, Qt.EditRole), 1)
def test_editable(self):
editable_model = PyTableModel([[0]], editable=True)
self.assertFalse(
int(self.model.flags(self.model.index(0, 0)) & Qt.ItemIsEditable))
self.assertTrue(
int(
editable_model.flags(editable_model.index(0, 0))
& Qt.ItemIsEditable))
def test_sort(self):
self.model.sort(1)
self.assertEqual(self.model.index(0, 0).data(Qt.EditRole), 2)
def test_setHeaderLabels(self):
self.model.setHorizontalHeaderLabels(['Col 1', 'Col 2'])
self.assertEqual(self.model.headerData(1, Qt.Horizontal), 'Col 2')
self.assertEqual(self.model.headerData(1, Qt.Vertical), 2)
def test_removeRows(self):
self.model.removeRows(0, 1)
self.assertEqual(len(self.model), 1)
self.assertEqual(self.model[0][1], 3)
def test_removeColumns(self):
self.model.removeColumns(0, 1)
self.assertEqual(self.model.columnCount(), 1)
self.assertEqual(self.model[1][0], 3)
def test_insertRows(self):
self.model.insertRows(0, 1)
self.assertEqual(self.model[1][0], 1)
def test_insertColumns(self):
self.model.insertColumns(0, 1)
self.assertEqual(self.model[0], ['', 1, 4])
def test_wrap(self):
self.model.wrap([[0]])
self.assertEqual(self.model.rowCount(), 1)
self.assertEqual(self.model.columnCount(), 1)
def test_clear(self):
self.model.clear()
self.assertEqual(self.model.rowCount(), 0)
def test_append(self):
self.model.append([5, 6])
self.assertEqual(self.model[2][1], 6)
self.assertEqual(self.model.rowCount(), 3)
def test_extend(self):
self.model.extend([[5, 6]])
self.assertEqual(self.model[2][1], 6)
self.assertEqual(self.model.rowCount(), 3)
def test_insert(self):
self.model.insert(0, [5, 6])
self.assertEqual(self.model[0][1], 6)
self.assertEqual(self.model.rowCount(), 3)
def test_remove(self):
self.model.remove([2, 3])
self.assertEqual(self.model.rowCount(), 1)
def test_other_roles(self):
self.model.append([2, 3])
self.model.setData(self.model.index(2, 0), Qt.AlignCenter,
Qt.TextAlignmentRole)
del self.model[1]
self.assertTrue(
Qt.AlignCenter
& self.model.data(self.model.index(1, 0), Qt.TextAlignmentRole))
def test_set_iten_slice(self):
self.model[:1] = [[10, 11], [12, 13], [14, 15]]
self.assertEqual(list(self.model),
[[10, 11], [12, 13], [14, 15], [2, 3]])
self.model[1:3] = []
self.assertEqual(list(self.model), [[10, 11], [2, 3]])
self.model[:] = [[20, 21]]
self.assertEqual(list(self.model), [[20, 21]])
self.model[1:] = [[10, 11], [2, 3]]
self.assertEqual(list(self.model), [[20, 21], [10, 11], [2, 3]])
def test_emits_column_changes_on_row_insert(self):
inserted = []
removed = []
model = PyTableModel()
model.columnsInserted.connect(inserted.append)
model.columnsRemoved.connect(removed.append)
inserted = QSignalSpy(model.columnsInserted)
removed = QSignalSpy(model.columnsRemoved)
model.append([2])
self.assertEqual(list(inserted)[-1][1:], [0, 0])
model.append([2, 3])
self.assertEqual(list(inserted)[-1][1:], [1, 1])
del model[:]
self.assertEqual(list(removed)[0][1:], [0, 1])
model.extend([[0, 1], [0, 2]])
self.assertEqual(list(inserted)[-1][1:], [0, 1])
model.clear()
self.assertEqual(list(removed)[0][1:], [0, 1])
model[:] = [[1], [2]]
self.assertEqual(list(inserted)[-1][1:], [0, 0])
class TestPyTableModel(unittest.TestCase):
def setUp(self):
self.model = PyTableModel([[1, 4],
[2, 3]])
def test_init(self):
self.model = PyTableModel()
self.assertEqual(self.model.rowCount(), 0)
def test_rowCount(self):
self.assertEqual(self.model.rowCount(), 2)
self.assertEqual(len(self.model), 2)
def test_columnCount(self):
self.assertEqual(self.model.columnCount(), 2)
def test_data(self):
mi = self.model.index(0, 0)
self.assertEqual(self.model.data(mi), '1')
self.assertEqual(self.model.data(mi, Qt.EditRole), 1)
def test_editable(self):
editable_model = PyTableModel([[0]], editable=True)
self.assertFalse(int(self.model.flags(self.model.index(0, 0)) & Qt.ItemIsEditable))
self.assertTrue(int(editable_model.flags(editable_model.index(0, 0)) & Qt.ItemIsEditable))
def test_sort(self):
self.model.sort(1)
self.assertEqual(self.model.index(0, 0).data(Qt.EditRole), 2)
def test_setHeaderLabels(self):
self.model.setHorizontalHeaderLabels(['Col 1', 'Col 2'])
self.assertEqual(self.model.headerData(1, Qt.Horizontal), 'Col 2')
self.assertEqual(self.model.headerData(1, Qt.Vertical), 2)
def test_removeRows(self):
self.model.removeRows(0, 1)
self.assertEqual(len(self.model), 1)
self.assertEqual(self.model[0][1], 3)
def test_removeColumns(self):
self.model.removeColumns(0, 1)
self.assertEqual(self.model.columnCount(), 1)
self.assertEqual(self.model[1][0], 3)
def test_insertRows(self):
self.model.insertRows(0, 1)
self.assertEqual(self.model[1][0], 1)
def test_insertColumns(self):
self.model.insertColumns(0, 1)
self.assertEqual(self.model[0], ['', 1, 4])
def test_wrap(self):
self.model.wrap([[0]])
self.assertEqual(self.model.rowCount(), 1)
self.assertEqual(self.model.columnCount(), 1)
def test_clear(self):
self.model.clear()
self.assertEqual(self.model.rowCount(), 0)
def test_append(self):
self.model.append([5, 6])
self.assertEqual(self.model[2][1], 6)
self.assertEqual(self.model.rowCount(), 3)
def test_extend(self):
self.model.extend([[5, 6]])
self.assertEqual(self.model[2][1], 6)
self.assertEqual(self.model.rowCount(), 3)
def test_insert(self):
self.model.insert(0, [5, 6])
self.assertEqual(self.model[0][1], 6)
self.assertEqual(self.model.rowCount(), 3)
def test_remove(self):
self.model.remove([2, 3])
self.assertEqual(self.model.rowCount(), 1)
def test_other_roles(self):
self.model.append([2, 3])
self.model.setData(self.model.index(2, 0),
Qt.AlignCenter,
Qt.TextAlignmentRole)
del self.model[1]
self.assertTrue(Qt.AlignCenter &
self.model.data(self.model.index(1, 0),
Qt.TextAlignmentRole))
