def testAllowEmpty(self):
l1 = create_layer('l1')
l2 = create_layer('l2')
QgsProject.instance().addMapLayers([l1, l2])
m = QgsMapLayerModel()
self.assertEqual(m.rowCount(QModelIndex()), 2)
m.setAllowEmptyLayer(True)
self.assertEqual(m.rowCount(QModelIndex()), 3)
self.assertFalse(m.data(m.index(0, 0), Qt.DisplayRole))
m.setAllowEmptyLayer(False)
self.assertEqual(m.rowCount(QModelIndex()), 2)
self.assertTrue(m.data(m.index(0, 0), Qt.DisplayRole))
# add layers after allow empty is true
m.setAllowEmptyLayer(True)
l3 = create_layer('l3')
QgsProject.instance().addMapLayers([l3])
self.assertEqual(m.rowCount(QModelIndex()), 4)
self.assertFalse(m.data(m.index(0, 0), Qt.DisplayRole))
self.assertEqual(m.data(m.index(1, 0), Qt.DisplayRole), 'l1')
self.assertEqual(m.data(m.index(2, 0), Qt.DisplayRole), 'l2')
self.assertEqual(m.data(m.index(3, 0), Qt.DisplayRole), 'l3')
QgsProject.instance().removeMapLayers([l1.id(), l2.id(), l3.id()])
def testDisplayRole(self):
l1 = create_layer('l1')
l2 = create_layer('l2')
QgsProject.instance().addMapLayers([l1, l2])
m = QgsMapLayerModel()
self.assertEqual(m.data(m.index(0, 0), Qt.DisplayRole), 'l1')
self.assertEqual(m.data(m.index(1, 0), Qt.DisplayRole), 'l2')
m.setAllowEmptyLayer(True)
self.assertFalse(m.data(m.index(0, 0), Qt.DisplayRole))
self.assertEqual(m.data(m.index(1, 0), Qt.DisplayRole), 'l1')
self.assertEqual(m.data(m.index(2, 0), Qt.DisplayRole), 'l2')
QgsProject.instance().removeMapLayers([l1.id(), l2.id()])
def testIsEmptyRole(self):
l1 = create_layer('l1')
l2 = create_layer('l2')
QgsProject.instance().addMapLayers([l1, l2])
m = QgsMapLayerModel()
self.assertFalse(m.data(m.index(0, 0), QgsMapLayerModel.EmptyRole))
self.assertFalse(m.data(m.index(1, 0), QgsMapLayerModel.EmptyRole))
m.setAllowEmptyLayer(True)
self.assertTrue(m.data(m.index(0, 0), QgsMapLayerModel.EmptyRole))
self.assertFalse(m.data(m.index(1, 0), QgsMapLayerModel.EmptyRole))
self.assertFalse(m.data(m.index(2, 0), QgsMapLayerModel.EmptyRole))
m.setAdditionalItems(['a'])
self.assertFalse(m.data(m.index(3, 0), QgsMapLayerModel.EmptyRole))
QgsProject.instance().removeMapLayers([l1.id(), l2.id()])
def testDisplayRoleShowCrs(self):
l1 = create_layer('l1')
l2 = create_layer('l2')
QgsProject.instance().addMapLayers([l1, l2])
m = QgsMapLayerModel()
m.setShowCrs(True)
self.assertEqual(m.data(m.index(0, 0), Qt.DisplayRole), 'l1 [EPSG:3111]')
self.assertEqual(m.data(m.index(1, 0), Qt.DisplayRole), 'l2 [EPSG:3111]')
m.setAllowEmptyLayer(True)
self.assertFalse(m.data(m.index(0, 0), Qt.DisplayRole))
self.assertEqual(m.data(m.index(1, 0), Qt.DisplayRole), 'l1 [EPSG:3111]')
self.assertEqual(m.data(m.index(2, 0), Qt.DisplayRole), 'l2 [EPSG:3111]')
m.setAdditionalItems(['a'])
self.assertEqual(m.data(m.index(3, 0), Qt.DisplayRole), 'a')
QgsProject.instance().removeMapLayers([l1.id(), l2.id()])
def testCheckStateRole(self):
l1 = create_layer('l1')
l2 = create_layer('l2')
QgsProject.instance().addMapLayers([l1, l2])
m = QgsMapLayerModel()
# not checkable
self.assertFalse(m.data(m.index(0, 0), Qt.CheckStateRole))
self.assertFalse(m.data(m.index(1, 0), Qt.CheckStateRole))
m.setAllowEmptyLayer(True)
self.assertFalse(m.data(m.index(0, 0), Qt.CheckStateRole))
self.assertFalse(m.data(m.index(1, 0), Qt.CheckStateRole))
self.assertFalse(m.data(m.index(2, 0), Qt.CheckStateRole))
m.setAllowEmptyLayer(False)
# checkable
m.setItemsCheckable(True)
m.checkAll(Qt.Checked)
self.assertTrue(m.data(m.index(0, 0), Qt.CheckStateRole))
self.assertTrue(m.data(m.index(1, 0), Qt.CheckStateRole))
m.setAllowEmptyLayer(True)
self.assertFalse(m.data(m.index(0, 0), Qt.CheckStateRole))
self.assertTrue(m.data(m.index(1, 0), Qt.CheckStateRole))
self.assertTrue(m.data(m.index(2, 0), Qt.CheckStateRole))
m.setAdditionalItems(['a'])
self.assertFalse(m.data(m.index(3, 0), Qt.CheckStateRole))
QgsProject.instance().removeMapLayers([l1.id(), l2.id()])
def testAdditionalItems(self):
l1 = create_layer('l1')
l2 = create_layer('l2')
QgsProject.instance().addMapLayers([l1, l2])
m = QgsMapLayerModel()
self.assertEqual(m.rowCount(QModelIndex()), 2)
m.setAdditionalItems(['a', 'b'])
self.assertEqual(m.rowCount(QModelIndex()), 4)
self.assertEqual(m.data(m.index(0, 0), Qt.DisplayRole), 'l1')
self.assertEqual(m.data(m.index(1, 0), Qt.DisplayRole), 'l2')
self.assertEqual(m.data(m.index(2, 0), Qt.DisplayRole), 'a')
self.assertEqual(m.data(m.index(3, 0), Qt.DisplayRole), 'b')
m.setAllowEmptyLayer(True)
self.assertEqual(m.rowCount(QModelIndex()), 5)
self.assertFalse(m.data(m.index(0, 0), Qt.DisplayRole))
self.assertEqual(m.data(m.index(1, 0), Qt.DisplayRole), 'l1')
self.assertEqual(m.data(m.index(2, 0), Qt.DisplayRole), 'l2')
self.assertEqual(m.data(m.index(3, 0), Qt.DisplayRole), 'a')
self.assertEqual(m.data(m.index(4, 0), Qt.DisplayRole), 'b')
QgsProject.instance().removeMapLayers([l1.id(), l2.id()])
self.assertEqual(m.rowCount(QModelIndex()), 3)
self.assertFalse(m.data(m.index(0, 0), Qt.DisplayRole))
self.assertEqual(m.data(m.index(1, 0), Qt.DisplayRole), 'a')
self.assertEqual(m.data(m.index(2, 0), Qt.DisplayRole), 'b')
def testDisplayRoleShowCrs(self):
l1 = create_layer('l1')
l2 = create_layer('l2')
l3 = QgsVectorLayer("NoGeometry?field=fldtxt:string&field=fldint:integer",
'no geom', "memory")
QgsProject.instance().addMapLayers([l1, l2, l3])
m = QgsMapLayerModel()
m.setShowCrs(True)
self.assertEqual(m.data(m.index(0, 0), Qt.DisplayRole), 'l1 [EPSG:3111]')
self.assertEqual(m.data(m.index(1, 0), Qt.DisplayRole), 'l2 [EPSG:3111]')
self.assertEqual(m.data(m.index(2, 0), Qt.DisplayRole), 'no geom')
m.setAllowEmptyLayer(True)
self.assertFalse(m.data(m.index(0, 0), Qt.DisplayRole))
self.assertEqual(m.data(m.index(1, 0), Qt.DisplayRole), 'l1 [EPSG:3111]')
self.assertEqual(m.data(m.index(2, 0), Qt.DisplayRole), 'l2 [EPSG:3111]')
self.assertEqual(m.data(m.index(3, 0), Qt.DisplayRole), 'no geom')
m.setAdditionalItems(['a'])
self.assertEqual(m.data(m.index(4, 0), Qt.DisplayRole), 'a')
QgsProject.instance().removeMapLayers([l1.id(), l2.id(), l3.id()])
def testSetData(self):
l1 = create_layer('l1')
l2 = create_layer('l2')
QgsProject.instance().addMapLayers([l1, l2])
m = QgsMapLayerModel()
# set checked
m.setItemsCheckable(True)
self.assertTrue(m.setData(m.index(0, 0), True, Qt.CheckStateRole))
self.assertTrue(m.data(m.index(0, 0), Qt.CheckStateRole))
self.assertFalse(m.data(m.index(1, 0), Qt.CheckStateRole))
self.assertTrue(m.setData(m.index(1, 0), True, Qt.CheckStateRole))
self.assertTrue(m.data(m.index(0, 0), Qt.CheckStateRole))
self.assertTrue(m.data(m.index(1, 0), Qt.CheckStateRole))
m.setAllowEmptyLayer(True)
self.assertFalse(m.setData(m.index(0, 0), True, Qt.CheckStateRole))
self.assertTrue(m.setData(m.index(1, 0), True, Qt.CheckStateRole))
m.setAdditionalItems(['a'])
self.assertFalse(m.setData(m.index(3, 0), True, Qt.CheckStateRole))
QgsProject.instance().removeMapLayers([l1.id(), l2.id()])
def testDisplayRoleShowCrs(self):
l1 = create_layer('l1')
l2 = create_layer('l2')
l3 = QgsVectorLayer("NoGeometry?field=fldtxt:string&field=fldint:integer",
'no geom', "memory")
QgsProject.instance().addMapLayers([l1, l2, l3])
m = QgsMapLayerModel()
m.setShowCrs(True)
self.assertEqual(m.data(m.index(0, 0), Qt.DisplayRole), 'l1 [EPSG:3111]')
self.assertEqual(m.data(m.index(1, 0), Qt.DisplayRole), 'l2 [EPSG:3111]')
self.assertEqual(m.data(m.index(2, 0), Qt.DisplayRole), 'no geom')
m.setAllowEmptyLayer(True)
self.assertFalse(m.data(m.index(0, 0), Qt.DisplayRole))
self.assertEqual(m.data(m.index(1, 0), Qt.DisplayRole), 'l1 [EPSG:3111]')
self.assertEqual(m.data(m.index(2, 0), Qt.DisplayRole), 'l2 [EPSG:3111]')
self.assertEqual(m.data(m.index(3, 0), Qt.DisplayRole), 'no geom')
m.setAdditionalItems(['a'])
self.assertEqual(m.data(m.index(4, 0), Qt.DisplayRole), 'a')
QgsProject.instance().removeMapLayers([l1.id(), l2.id(), l3.id()])
def testSetData(self):
l1 = create_layer('l1')
l2 = create_layer('l2')
QgsMapLayerRegistry.instance().addMapLayers([l1, l2])
m = QgsMapLayerModel()
# set checked
m.setItemsCheckable(True)
self.assertTrue(m.setData(m.index(0, 0), True, Qt.CheckStateRole))
self.assertTrue(m.data(m.index(0, 0), Qt.CheckStateRole))
self.assertFalse(m.data(m.index(1, 0), Qt.CheckStateRole))
self.assertTrue(m.setData(m.index(1, 0), True, Qt.CheckStateRole))
self.assertTrue(m.data(m.index(0, 0), Qt.CheckStateRole))
self.assertTrue(m.data(m.index(1, 0), Qt.CheckStateRole))
m.setAllowEmptyLayer(True)
self.assertFalse(m.setData(m.index(0, 0), True, Qt.CheckStateRole))
self.assertTrue(m.setData(m.index(1, 0), True, Qt.CheckStateRole))
m.setAdditionalItems(['a'])
self.assertFalse(m.setData(m.index(3, 0), True, Qt.CheckStateRole))
QgsMapLayerRegistry.instance().removeMapLayers([l1.id(), l2.id()])
def testRemoveRows(self):
l1 = create_layer('l1')
l2 = create_layer('l2')
QgsProject.instance().addMapLayers([l1, l2])
m = QgsMapLayerModel()
self.assertEqual(m.rowCount(), 2)
self.assertEqual(m.data(m.index(0, 0), QgsMapLayerModel.LayerIdRole),
l1.id())
self.assertEqual(m.data(m.index(1, 0), QgsMapLayerModel.LayerIdRole),
l2.id())
self.assertTrue(m.removeRows(0, 1))
self.assertEqual(m.rowCount(), 1)
self.assertEqual(m.data(m.index(0, 0), QgsMapLayerModel.LayerIdRole),
l2.id())
self.assertTrue(m.removeRows(0, 1))
self.assertEqual(m.rowCount(), 0)
self.assertFalse(m.removeRows(0, 1))
self.assertFalse(m.removeRows(-1, 1))
m = QgsMapLayerModel()
self.assertEqual(m.rowCount(), 2)
self.assertTrue(m.removeRows(0, 2))
self.assertEqual(m.rowCount(), 0)
m = QgsMapLayerModel()
m.setAllowEmptyLayer(True)
self.assertEqual(m.rowCount(), 3)
self.assertFalse(m.removeRows(0, 2))
self.assertEqual(m.rowCount(), 3)
self.assertTrue(m.removeRows(2, 1))
self.assertEqual(m.rowCount(), 2)
self.assertFalse(m.data(m.index(0, 0), QgsMapLayerModel.LayerIdRole))
self.assertEqual(m.data(m.index(1, 0), QgsMapLayerModel.LayerIdRole),
l1.id())
self.assertFalse(m.removeRows(2, 1))
self.assertEqual(m.rowCount(), 2)
self.assertTrue(m.removeRows(1, 1))
self.assertEqual(m.rowCount(), 1)
self.assertFalse(m.data(m.index(0, 0), QgsMapLayerModel.LayerIdRole))
m = QgsMapLayerModel()
m.setAllowEmptyLayer(True)
self.assertEqual(m.rowCount(), 3)
self.assertFalse(m.removeRows(3, 2))
self.assertTrue(m.removeRows(1, 2))
self.assertEqual(m.rowCount(), 1)
self.assertFalse(m.data(m.index(0, 0), QgsMapLayerModel.LayerIdRole))
self.assertFalse(m.removeRows(1, 1))
self.assertFalse(m.removeRows(0, 1))
QgsProject.instance().removeMapLayers([l1.id(), l2.id()])
def testIsAdditionalRole(self):
l1 = create_layer('l1')
l2 = create_layer('l2')
QgsProject.instance().addMapLayers([l1, l2])
m = QgsMapLayerModel()
self.assertFalse(m.data(m.index(0, 0), QgsMapLayerModel.AdditionalRole))
self.assertFalse(m.data(m.index(1, 0), QgsMapLayerModel.AdditionalRole))
m.setAllowEmptyLayer(True)
self.assertFalse(m.data(m.index(0, 0), QgsMapLayerModel.AdditionalRole))
self.assertFalse(m.data(m.index(1, 0), QgsMapLayerModel.AdditionalRole))
self.assertFalse(m.data(m.index(2, 0), QgsMapLayerModel.AdditionalRole))
m.setAdditionalItems(['a'])
self.assertFalse(m.data(m.index(0, 0), QgsMapLayerModel.AdditionalRole))
self.assertFalse(m.data(m.index(1, 0), QgsMapLayerModel.AdditionalRole))
self.assertFalse(m.data(m.index(2, 0), QgsMapLayerModel.AdditionalRole))
self.assertTrue(m.data(m.index(3, 0), QgsMapLayerModel.AdditionalRole))
QgsProject.instance().removeMapLayers([l1.id(), l2.id()])
def testAdditionalLayers(self):
l1 = create_layer('l1')
l2 = create_layer('l2')
QgsProject.instance().addMapLayers([l1, l2])
m = QgsMapLayerModel()
self.assertEqual(m.rowCount(QModelIndex()), 2)
l3 = create_layer('l3')
l4 = create_layer('l4')
m.setAdditionalLayers([l3, l4])
self.assertEqual(m.rowCount(QModelIndex()), 4)
m.setAdditionalItems(['a', 'b'])
self.assertEqual(m.rowCount(QModelIndex()), 6)
self.assertEqual(m.data(m.index(0, 0), Qt.DisplayRole), 'l1')
self.assertEqual(m.data(m.index(1, 0), Qt.DisplayRole), 'l2')
self.assertEqual(m.data(m.index(2, 0), Qt.DisplayRole), 'l3')
self.assertEqual(m.data(m.index(3, 0), Qt.DisplayRole), 'l4')
self.assertEqual(m.data(m.index(4, 0), Qt.DisplayRole), 'a')
self.assertEqual(m.data(m.index(5, 0), Qt.DisplayRole), 'b')
m.setAllowEmptyLayer(True)
self.assertEqual(m.rowCount(QModelIndex()), 7)
self.assertFalse(m.data(m.index(0, 0), Qt.DisplayRole))
self.assertEqual(m.data(m.index(1, 0), Qt.DisplayRole), 'l1')
self.assertEqual(m.data(m.index(2, 0), Qt.DisplayRole), 'l2')
self.assertEqual(m.data(m.index(3, 0), Qt.DisplayRole), 'l3')
self.assertEqual(m.data(m.index(4, 0), Qt.DisplayRole), 'l4')
self.assertEqual(m.data(m.index(5, 0), Qt.DisplayRole), 'a')
self.assertEqual(m.data(m.index(6, 0), Qt.DisplayRole), 'b')
l3.deleteLater()
QCoreApplication.sendPostedEvents(None, QEvent.DeferredDelete)
self.assertEqual(m.rowCount(QModelIndex()), 6)
self.assertFalse(m.data(m.index(0, 0), Qt.DisplayRole))
self.assertEqual(m.data(m.index(1, 0), Qt.DisplayRole), 'l1')
self.assertEqual(m.data(m.index(2, 0), Qt.DisplayRole), 'l2')
self.assertEqual(m.data(m.index(3, 0), Qt.DisplayRole), 'l4')
self.assertEqual(m.data(m.index(4, 0), Qt.DisplayRole), 'a')
self.assertEqual(m.data(m.index(5, 0), Qt.DisplayRole), 'b')
l5 = create_layer('l5')
l6 = create_layer('l6')
m.setAdditionalLayers([l5, l6, l4])
self.assertEqual(m.rowCount(QModelIndex()), 8)
self.assertFalse(m.data(m.index(0, 0), Qt.DisplayRole))
self.assertEqual(m.data(m.index(1, 0), Qt.DisplayRole), 'l1')
self.assertEqual(m.data(m.index(2, 0), Qt.DisplayRole), 'l2')
self.assertEqual(m.data(m.index(3, 0), Qt.DisplayRole), 'l5')
self.assertEqual(m.data(m.index(4, 0), Qt.DisplayRole), 'l6')
self.assertEqual(m.data(m.index(5, 0), Qt.DisplayRole), 'l4')
self.assertEqual(m.data(m.index(6, 0), Qt.DisplayRole), 'a')
self.assertEqual(m.data(m.index(7, 0), Qt.DisplayRole), 'b')
m.setAdditionalLayers([l5, l4])
self.assertEqual(m.rowCount(QModelIndex()), 7)
self.assertFalse(m.data(m.index(0, 0), Qt.DisplayRole))
self.assertEqual(m.data(m.index(1, 0), Qt.DisplayRole), 'l1')
self.assertEqual(m.data(m.index(2, 0), Qt.DisplayRole), 'l2')
self.assertEqual(m.data(m.index(3, 0), Qt.DisplayRole), 'l5')
self.assertEqual(m.data(m.index(4, 0), Qt.DisplayRole), 'l4')
self.assertEqual(m.data(m.index(5, 0), Qt.DisplayRole), 'a')
self.assertEqual(m.data(m.index(6, 0), Qt.DisplayRole), 'b')
QgsProject.instance().removeMapLayers([l1.id(), l2.id()])
self.assertEqual(m.rowCount(QModelIndex()), 5)
self.assertFalse(m.data(m.index(0, 0), Qt.DisplayRole))
self.assertEqual(m.data(m.index(1, 0), Qt.DisplayRole), 'l5')
self.assertEqual(m.data(m.index(2, 0), Qt.DisplayRole), 'l4')
self.assertEqual(m.data(m.index(3, 0), Qt.DisplayRole), 'a')
self.assertEqual(m.data(m.index(4, 0), Qt.DisplayRole), 'b')
