def test_data(self):
model = ContinuousPalettesModel()
model.items = ["Palettes", self.palette1, self.palette2]
data = model.data
index = model.index
self.assertEqual(data(index(0, 0), Qt.EditRole), "Palettes")
self.assertEqual(data(index(1, 0), Qt.EditRole),
self.palette1.friendly_name)
self.assertEqual(data(index(2, 0), Qt.EditRole),
self.palette2.friendly_name)
self.assertEqual(data(index(0, 0), Qt.DisplayRole), "Palettes")
self.assertEqual(data(index(1, 0), Qt.DisplayRole),
self.palette1.friendly_name)
self.assertEqual(data(index(2, 0), Qt.DisplayRole),
self.palette2.friendly_name)
self.assertIsNone(data(index(0, 0), Qt.DecorationRole))
with patch.object(self.palette1, "color_strip") as color_strip:
self.assertIs(data(index(1, 0), Qt.DecorationRole),
color_strip.return_value)
with patch.object(self.palette2, "color_strip") as color_strip:
self.assertIs(data(index(2, 0), Qt.DecorationRole),
color_strip.return_value)
self.assertIsNone(data(index(0, 0), Qt.UserRole))
self.assertIs(data(index(1, 0), Qt.UserRole), self.palette1)
self.assertIs(data(index(2, 0), Qt.UserRole), self.palette2)
self.assertIsNone(data(index(2, 0), Qt.FontRole))
def testIndexOf(self):
model = ContinuousPalettesModel()
model.items = ["Palettes", self.palette1, self.palette2]
self.assertEqual(model.indexOf(self.palette1), 1)
self.assertEqual(model.indexOf(self.palette1.name), 1)
self.assertEqual(model.indexOf(self.palette1.friendly_name), 1)
self.assertEqual(model.indexOf(self.palette2), 2)
self.assertEqual(model.indexOf(self.palette2.name), 2)
self.assertEqual(model.indexOf(self.palette2.friendly_name), 2)
self.assertIsNone(model.indexOf(42))
def test_select_flags(self):
model = ContinuousPalettesModel()
model.items = ["Palettes", self.palette1, self.palette2]
self.assertFalse(model.flags(model.index(0, 0)) & Qt.ItemIsSelectable)
self.assertTrue(model.flags(model.index(1, 0)) & Qt.ItemIsSelectable)
self.assertTrue(model.flags(model.index(2, 0)) & Qt.ItemIsSelectable)
def test_count(self):
model = ContinuousPalettesModel()
model.items = [self.palette1, self.palette1]
self.assertEqual(model.rowCount(QModelIndex()), 2)
self.assertEqual(model.columnCount(QModelIndex()), 1)