def _test(raster_layer):
rat = get_rat(self.raster_layer_color, 1)
model = RATModel(rat)
tester = QAbstractItemModelTester(
model, QAbstractItemModelTester.FailureReportingMode.Warning)
row_count = model.rowCount(QModelIndex())
value_index = 1 if model.has_color else 0
value_0 = model.data(model.index(0, value_index, QModelIndex()))
value_last = model.data(
model.index(row_count - 1, value_index, QModelIndex()))
# Insert first
self.assertTrue(model.insert_row(0))
self.assertEqual(model.rowCount(QModelIndex()), row_count + 1)
self.assertNotEqual(
model.data(model.index(0, value_index, QModelIndex())),
value_0)
self.assertEqual(
model.data(model.index(0, value_index, QModelIndex())), 0)
self.assertTrue(model.remove_row(0))
self.assertEqual(model.rowCount(QModelIndex()), row_count)
self.assertEqual(
model.data(model.index(0, value_index, QModelIndex())),
value_0)
# Insert last
self.assertTrue(model.insert_row(row_count))
self.assertEqual(model.rowCount(QModelIndex()), row_count + 1)
self.assertNotEqual(
model.data(model.index(row_count, value_index, QModelIndex())),
value_last)
self.assertEqual(
model.data(model.index(row_count, value_index, QModelIndex())),
0)
self.assertEqual(
model.data(
model.index(row_count - 1, value_index, QModelIndex())),
value_last)
self.assertTrue(model.remove_row(row_count))
self.assertEqual(model.rowCount(QModelIndex()), row_count)
self.assertEqual(
model.data(
model.index(row_count - 1, value_index, QModelIndex())),
value_last)
def test_edit_color(self):
rat = get_rat(self.raster_layer_color, 1)
model = RATModel(rat)
tester = QAbstractItemModelTester(
model, QAbstractItemModelTester.FailureReportingMode.Warning)
index = model.index(0, 0)
value = QColor(Qt.magenta)
model.setData(index, value)
self.assertEqual(model.data(index, Qt.BackgroundColorRole), value)