def testAddingRemovingLayers(self):
# test model handles layer addition and removal
m = QgsMapLayerModel()
self.assertEqual(m.rowCount(QModelIndex()), 0)
l1 = create_layer('l1')
QgsProject.instance().addMapLayer(l1)
self.assertEqual(m.rowCount(QModelIndex()), 1)
l2 = create_layer('l2')
QgsProject.instance().addMapLayer(l2)
self.assertEqual(m.rowCount(QModelIndex()), 2)
QgsProject.instance().removeMapLayer(l1)
self.assertEqual(m.rowCount(QModelIndex()), 1)
QgsProject.instance().removeMapLayer(l2)
self.assertEqual(m.rowCount(QModelIndex()), 0)
# try creating a model when layers already exist in registry
l1 = create_layer('l1')
l2 = create_layer('l2')
QgsProject.instance().addMapLayers([l1, l2])
m = QgsMapLayerModel()
self.assertEqual(m.rowCount(QModelIndex()), 2)
QgsProject.instance().removeMapLayers([l1.id(), l2.id()])
self.assertEqual(m.rowCount(QModelIndex()), 0)
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 testIndexFromLayer(self):
l1 = create_layer('l1')
l2 = create_layer('l2')
QgsProject.instance().addMapLayers([l1, l2])
m = QgsMapLayerModel()
l3 = create_layer('l3') # not in registry
self.assertEqual(m.indexFromLayer(l1).row(), 0)
self.assertEqual(m.indexFromLayer(l2).row(), 1)
self.assertFalse(m.indexFromLayer(l3).isValid())
m.setAllowEmptyLayer(True)
self.assertEqual(m.indexFromLayer(l1).row(), 1)
self.assertEqual(m.indexFromLayer(l2).row(), 2)
QgsProject.instance().removeMapLayers([l1.id(), l2.id()])
def testGettersSetters(self):
""" test model getters/setters """
m = QgsMapLayerModel()
m.setItemsCheckable(True)
self.assertTrue(m.itemsCheckable())
m.setItemsCheckable(False)
self.assertFalse(m.itemsCheckable())
m.setAllowEmptyLayer(True)
self.assertTrue(m.allowEmptyLayer())
m.setAllowEmptyLayer(False)
self.assertFalse(m.allowEmptyLayer())
m.setShowCrs(True)
self.assertTrue(m.showCrs())
m.setShowCrs(False)
self.assertFalse(m.showCrs())
m.setAdditionalItems(['a', 'b'])
self.assertEqual(m.additionalItems(), ['a', 'b'])
m.setAdditionalItems([])
self.assertFalse(m.additionalItems())
def testCheckStateRole(self):
l1 = create_layer('l1')
l2 = create_layer('l2')
QgsMapLayerRegistry.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))
QgsMapLayerRegistry.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 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')
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 testFlags(self):
l1 = create_layer('l1')
l2 = create_layer('l2')
QgsProject.instance().addMapLayers([l1, l2])
m = QgsMapLayerModel()
# not checkable
self.assertFalse(m.flags(m.index(0, 0)) & Qt.ItemIsUserCheckable)
self.assertFalse(m.flags(m.index(1, 0)) & Qt.ItemIsUserCheckable)
m.setAllowEmptyLayer(True)
self.assertFalse(m.flags(m.index(0, 0)) & Qt.ItemIsUserCheckable)
self.assertFalse(m.flags(m.index(1, 0)) & Qt.ItemIsUserCheckable)
self.assertFalse(m.flags(m.index(2, 0)) & Qt.ItemIsUserCheckable)
m.setAllowEmptyLayer(False)
# checkable
m.setItemsCheckable(True)
self.assertTrue(m.flags(m.index(0, 0)) & Qt.ItemIsUserCheckable)
self.assertTrue(m.flags(m.index(1, 0)) & Qt.ItemIsUserCheckable)
m.setAllowEmptyLayer(True)
self.assertFalse(m.flags(m.index(0, 0)) & Qt.ItemIsUserCheckable)
self.assertTrue(m.flags(m.index(1, 0)) & Qt.ItemIsUserCheckable)
self.assertTrue(m.flags(m.index(2, 0)) & Qt.ItemIsUserCheckable)
m.setAdditionalItems(['a'])
self.assertFalse(m.flags(m.index(3, 0)) & Qt.ItemIsUserCheckable)
QgsProject.instance().removeMapLayers([l1.id(), l2.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 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')
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 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 testProject(self):
lA = create_layer('lA')
lB = create_layer('lB')
projectA = QgsProject.instance()
projectB = QgsProject()
projectA.addMapLayer(lA)
projectB.addMapLayer(lB)
m = QgsMapLayerModel()
self.assertEqual(m.data(m.index(0, 0), Qt.DisplayRole), lA.name())
m.setProject(projectB)
self.assertEqual(m.data(m.index(0, 0), Qt.DisplayRole), lB.name())
m.setProject(projectA)
self.assertEqual(m.data(m.index(0, 0), Qt.DisplayRole), lA.name())
QgsProject.instance().removeAllMapLayers()
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 testIndexFromLayer(self):
l1 = create_layer('l1')
l2 = create_layer('l2')
QgsProject.instance().addMapLayers([l1, l2])
m = QgsMapLayerModel()
l3 = create_layer('l3') # not in registry
self.assertEqual(m.indexFromLayer(l1).row(), 0)
self.assertEqual(m.layerFromIndex(m.indexFromLayer(l1)), l1)
self.assertEqual(m.indexFromLayer(l2).row(), 1)
self.assertEqual(m.layerFromIndex(m.indexFromLayer(l2)), l2)
self.assertFalse(m.indexFromLayer(l3).isValid())
m.setAllowEmptyLayer(True)
self.assertEqual(m.indexFromLayer(l1).row(), 1)
self.assertEqual(m.indexFromLayer(l2).row(), 2)
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 testMime(self):
l1 = create_layer('l1')
l2 = create_layer('l2')
QgsProject.instance().addMapLayers([l1, l2])
m = QgsMapLayerModel()
self.assertEqual(m.mimeTypes(), ['application/qgis.layermodeldata'])
data = m.mimeData([m.index(0, 0)])
self.assertTrue(data)
self.assertFalse(
m.canDropMimeData(data, Qt.MoveAction, 0, 0, QModelIndex()))
m.setItemsCanBeReordered(True)
self.assertTrue(
m.canDropMimeData(data, Qt.MoveAction, 0, 0, QModelIndex()))
self.assertTrue(
m.dropMimeData(data, Qt.MoveAction, 2, 0, QModelIndex()))
self.assertEqual(m.rowCount(), 3)
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.assertEqual(m.data(m.index(2, 0), QgsMapLayerModel.LayerIdRole),
l1.id())
QgsProject.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 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 testFlags(self):
l1 = create_layer('l1')
l2 = create_layer('l2')
QgsProject.instance().addMapLayers([l1, l2])
m = QgsMapLayerModel()
# not checkable
self.assertFalse(m.flags(m.index(0, 0)) & Qt.ItemIsUserCheckable)
self.assertFalse(m.flags(m.index(1, 0)) & Qt.ItemIsUserCheckable)
m.setAllowEmptyLayer(True)
self.assertFalse(m.flags(m.index(0, 0)) & Qt.ItemIsUserCheckable)
self.assertFalse(m.flags(m.index(1, 0)) & Qt.ItemIsUserCheckable)
self.assertFalse(m.flags(m.index(2, 0)) & Qt.ItemIsUserCheckable)
m.setAllowEmptyLayer(False)
# checkable
m.setItemsCheckable(True)
self.assertTrue(m.flags(m.index(0, 0)) & Qt.ItemIsUserCheckable)
self.assertTrue(m.flags(m.index(1, 0)) & Qt.ItemIsUserCheckable)
m.setAllowEmptyLayer(True)
self.assertFalse(m.flags(m.index(0, 0)) & Qt.ItemIsUserCheckable)
self.assertTrue(m.flags(m.index(1, 0)) & Qt.ItemIsUserCheckable)
self.assertTrue(m.flags(m.index(2, 0)) & Qt.ItemIsUserCheckable)
m.setAdditionalItems(['a'])
self.assertFalse(m.flags(m.index(3, 0)) & Qt.ItemIsUserCheckable)
QgsProject.instance().removeMapLayers([l1.id(), l2.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 testAddingRemovingLayers(self):
# test model handles layer addition and removal
m = QgsMapLayerModel()
self.assertEqual(m.rowCount(QModelIndex()), 0)
l1 = create_layer('l1')
QgsProject.instance().addMapLayer(l1)
self.assertEqual(m.rowCount(QModelIndex()), 1)
self.assertEqual(m.layerFromIndex(m.index(0, 0)), l1)
l2 = create_layer('l2')
QgsProject.instance().addMapLayer(l2)
self.assertEqual(m.rowCount(QModelIndex()), 2)
self.assertEqual(m.layerFromIndex(m.index(0, 0)), l1)
self.assertEqual(m.layerFromIndex(m.index(1, 0)), l2)
QgsProject.instance().removeMapLayer(l1)
self.assertEqual(m.rowCount(QModelIndex()), 1)
self.assertEqual(m.layerFromIndex(m.index(0, 0)), l2)
QgsProject.instance().removeMapLayer(l2)
self.assertEqual(m.rowCount(QModelIndex()), 0)
# try creating a model when layers already exist in registry
l1 = create_layer('l1')
l2 = create_layer('l2')
QgsProject.instance().addMapLayers([l1, l2])
m = QgsMapLayerModel()
self.assertEqual(m.rowCount(QModelIndex()), 2)
QgsProject.instance().removeMapLayers([l1.id(), l2.id()])
self.assertEqual(m.rowCount(QModelIndex()), 0)
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 testCheckAll(self):
l1 = create_layer('l1')
l2 = create_layer('l2')
QgsProject.instance().addMapLayers([l1, l2])
m = QgsMapLayerModel()
m.setItemsCheckable(True)
self.assertFalse(m.layersChecked())
self.assertEqual(set(m.layersChecked(Qt.Unchecked)), set([l1, l2]))
m.checkAll(Qt.Checked)
self.assertEqual(set(m.layersChecked()), set([l1, l2]))
self.assertFalse(set(m.layersChecked(Qt.Unchecked)))
m.checkAll(Qt.Unchecked)
self.assertFalse(m.layersChecked())
self.assertEqual(set(m.layersChecked(Qt.Unchecked)), set([l1, l2]))
QgsProject.instance().removeMapLayers([l1.id(), l2.id()])
def testGettersSetters(self):
""" test model getters/setters """
m = QgsMapLayerModel()
m.setItemsCheckable(True)
self.assertTrue(m.itemsCheckable())
m.setItemsCheckable(False)
self.assertFalse(m.itemsCheckable())
m.setAllowEmptyLayer(True)
self.assertTrue(m.allowEmptyLayer())
m.setAllowEmptyLayer(False)
self.assertFalse(m.allowEmptyLayer())
m.setShowCrs(True)
self.assertTrue(m.showCrs())
m.setShowCrs(False)
self.assertFalse(m.showCrs())
m.setAdditionalItems(['a', 'b'])
self.assertEqual(m.additionalItems(), ['a', 'b'])
m.setAdditionalItems([])
self.assertFalse(m.additionalItems())
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 testCheckAll(self):
l1 = create_layer('l1')
l2 = create_layer('l2')
QgsProject.instance().addMapLayers([l1, l2])
m = QgsMapLayerModel()
m.setItemsCheckable(True)
self.assertFalse(m.layersChecked())
self.assertEqual(set(m.layersChecked(Qt.Unchecked)), set([l1, l2]))
m.checkAll(Qt.Checked)
self.assertEqual(set(m.layersChecked()), set([l1, l2]))
self.assertFalse(set(m.layersChecked(Qt.Unchecked)))
m.checkAll(Qt.Unchecked)
self.assertFalse(m.layersChecked())
self.assertEqual(set(m.layersChecked(Qt.Unchecked)), set([l1, l2]))
QgsProject.instance().removeMapLayers([l1.id(), l2.id()])
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')
self.assertIsNone(m.data(m.index(0, 0), Qt.DecorationRole))
# set icon and text for empty item
m.setAllowEmptyLayer(True, 'empty',
QgsApplication.getThemeIcon('/mItemBookmark.svg'))
self.assertEqual(m.data(m.index(0, 0), Qt.DisplayRole), 'empty')
self.assertFalse(m.data(m.index(0, 0), Qt.DecorationRole).isNull())
QgsProject.instance().removeMapLayers([l1.id(), l2.id(), l3.id()])
