def testIsEmpty(self):
""" test isEmpty method """
config = QgsAttributeTableConfig()
self.assertTrue(config.isEmpty())
c = QgsAttributeTableConfig.ColumnConfig()
c.name = 'test'
config.setColumns([c])
self.assertFalse(config.isEmpty())
def testSetColumns(self):
""" test setting columns """
config = QgsAttributeTableConfig()
self.assertEqual(config.columns(), [])
c1 = QgsAttributeTableConfig.ColumnConfig()
c1.name = 'test'
c1.hidden = False
c1.width = 9
c2 = QgsAttributeTableConfig.ColumnConfig()
c2.name = 'test2'
c2.hidden = True
c2.width = 11
config.setColumns([c1, c2])
result = config.columns()
self.assertEqual(result[0].name, 'test')
self.assertEqual(result[1].name, 'test2')
self.assertEqual(result[0].hidden, False)
self.assertEqual(result[1].hidden, True)
self.assertEqual(result[0].width, 9)
self.assertEqual(result[1].width, 11)
def testSetColumns(self):
""" test setting columns """
config = QgsAttributeTableConfig()
self.assertEqual(config.columns(), [])
c1 = QgsAttributeTableConfig.ColumnConfig()
c1.name = 'test'
c1.hidden = False
c1.width = 9
c2 = QgsAttributeTableConfig.ColumnConfig()
c2.name = 'test2'
c2.hidden = True
c2.width = 11
config.setColumns([c1, c2])
result = config.columns()
self.assertEqual(result[0].name, 'test')
self.assertEqual(result[1].name, 'test2')
self.assertEqual(result[0].hidden, False)
self.assertEqual(result[1].hidden, True)
self.assertEqual(result[0].width, 9)
self.assertEqual(result[1].width, 11)
def testColumnHidden(self):
""" test hiding columns """
config = QgsAttributeTableConfig()
c1 = QgsAttributeTableConfig.ColumnConfig()
c1.name = 'test'
c1.hidden = False
c2 = QgsAttributeTableConfig.ColumnConfig()
c2.name = 'test2'
c2.hidden = False
config.setColumns([c1, c2])
self.assertFalse(config.columnHidden(0))
self.assertFalse(config.columnHidden(1))
config.setColumnHidden(1, True)
self.assertFalse(config.columnHidden(0))
self.assertTrue(config.columnHidden(1))
self.assertFalse(config.columns()[0].hidden)
self.assertTrue(config.columns()[1].hidden)
config.setColumnHidden(0, True)
self.assertTrue(config.columnHidden(0))
self.assertTrue(config.columnHidden(1))
self.assertTrue(config.columns()[0].hidden)
self.assertTrue(config.columns()[1].hidden)
c2.hidden = True
config.setColumns([c1, c2])
self.assertFalse(config.columnHidden(0))
self.assertTrue(config.columnHidden(1))
def testSameColumns(self):
""" test hasSameColumns() check """
config = QgsAttributeTableConfig()
c1 = QgsAttributeTableConfig.ColumnConfig()
c1.name = 'test'
c1.hidden = False
c1.width = 100
c2 = QgsAttributeTableConfig.ColumnConfig()
c2.name = 'test2'
c2.hidden = False
c2.width = 120
config.setColumns([c1, c2])
config2 = QgsAttributeTableConfig()
config2.setColumns([c1, c2])
self.assertTrue(config.hasSameColumns(config2))
c1.width = 200
config2.setColumns([c1, c2])
self.assertTrue(config.hasSameColumns(config2))
c1.hidden = True
config2.setColumns([c1, c2])
self.assertFalse(config.hasSameColumns(config2))
config2.setColumns([c2, c1])
self.assertFalse(config.hasSameColumns(config2))
c2.name = 'test3'
config2.setColumns([c1, c2])
self.assertFalse(config.hasSameColumns(config2))
def testColumnHidden(self):
""" test hiding columns """
config = QgsAttributeTableConfig()
c1 = QgsAttributeTableConfig.ColumnConfig()
c1.name = 'test'
c1.hidden = False
c2 = QgsAttributeTableConfig.ColumnConfig()
c2.name = 'test2'
c2.hidden = False
config.setColumns([c1, c2])
self.assertFalse(config.columnHidden(0))
self.assertFalse(config.columnHidden(1))
config.setColumnHidden(1, True)
self.assertFalse(config.columnHidden(0))
self.assertTrue(config.columnHidden(1))
self.assertFalse(config.columns()[0].hidden)
self.assertTrue(config.columns()[1].hidden)
config.setColumnHidden(0, True)
self.assertTrue(config.columnHidden(0))
self.assertTrue(config.columnHidden(1))
self.assertTrue(config.columns()[0].hidden)
self.assertTrue(config.columns()[1].hidden)
c2.hidden = True
config.setColumns([c1, c2])
self.assertFalse(config.columnHidden(0))
self.assertTrue(config.columnHidden(1))
def testSameColumns(self):
""" test hasSameColumns() check """
config = QgsAttributeTableConfig()
c1 = QgsAttributeTableConfig.ColumnConfig()
c1.name = 'test'
c1.hidden = False
c1.width = 100
c2 = QgsAttributeTableConfig.ColumnConfig()
c2.name = 'test2'
c2.hidden = False
c2.width = 120
config.setColumns([c1, c2])
config2 = QgsAttributeTableConfig()
config2.setColumns([c1, c2])
self.assertTrue(config.hasSameColumns(config2))
c1.width = 200
config2.setColumns([c1, c2])
self.assertTrue(config.hasSameColumns(config2))
c1.hidden = True
config2.setColumns([c1, c2])
self.assertFalse(config.hasSameColumns(config2))
config2.setColumns([c2, c1])
self.assertFalse(config.hasSameColumns(config2))
c2.name = 'test3'
config2.setColumns([c1, c2])
self.assertFalse(config.hasSameColumns(config2))
def layer_has_custom_attribute_table(layer) -> bool:
# Do not use the isEmpty() on the QgsVectorLayer. It's automatically populated if empty with the fields.
config = QgsAttributeTableConfig()
config.update(layer.fields())
return not config.hasSameColumns(layer.attributeTableConfig())
def testColumnWidth(self):
""" test setting column widths """
config = QgsAttributeTableConfig()
c1 = QgsAttributeTableConfig.ColumnConfig()
c1.name = 'test'
c1.width = -1
c2 = QgsAttributeTableConfig.ColumnConfig()
c2.name = 'test2'
c2.width = 27
config.setColumns([c1, c2])
self.assertEqual(config.columnWidth(0), -1)
self.assertEqual(config.columnWidth(1), 27)
with self.assertRaises(IndexError):
config.columnWidth(-1)
with self.assertRaises(IndexError):
config.columnWidth(2)
config.setColumnWidth(1, -1)
self.assertEqual(config.columnWidth(0), -1)
self.assertEqual(config.columnWidth(1), -1)
self.assertEqual(config.columns()[0].width, -1)
self.assertEqual(config.columns()[1].width, -1)
config.setColumnWidth(0, 34)
self.assertEqual(config.columnWidth(0), 34)
self.assertEqual(config.columnWidth(1), -1)
self.assertEqual(config.columns()[0].width, 34)
self.assertEqual(config.columns()[1].width, -1)
with self.assertRaises(IndexError):
config.setColumnWidth(-1, 11)
with self.assertRaises(IndexError):
config.setColumnWidth(2, 11)
c2.width = 12
config.setColumns([c1, c2])
self.assertEqual(config.columnWidth(0), -1)
self.assertEqual(config.columnWidth(1), 12)