def testGettersSetters(self):
""" test model getters/setters """
m = QgsMapLayerProxyModel()
l1 = create_layer('l1')
QgsProject.instance().addMapLayer(l1)
l2 = create_layer('l2')
QgsProject.instance().addMapLayer(l2)
m.setFilters(QgsMapLayerProxyModel.LineLayer
| QgsMapLayerProxyModel.WritableLayer)
self.assertEqual(
m.filters(), QgsMapLayerProxyModel.LineLayer
| QgsMapLayerProxyModel.WritableLayer)
m.setExceptedLayerIds([l2.id()])
self.assertEqual(m.exceptedLayerIds(), [l2.id()])
m.setExceptedLayerList([l2])
self.assertEqual(m.exceptedLayerList(), [l2])
m.setExcludedProviders(['a', 'b'])
self.assertEqual(m.excludedProviders(), ['a', 'b'])
m.setFilterString('c')
self.assertEqual(m.filterString(), 'c')
def testGettersSetters(self):
""" test model getters/setters """
m = QgsMapLayerProxyModel()
l1 = create_layer('l1')
QgsProject.instance().addMapLayer(l1)
l2 = create_layer('l2')
QgsProject.instance().addMapLayer(l2)
m.setFilters(QgsMapLayerProxyModel.LineLayer | QgsMapLayerProxyModel.WritableLayer)
self.assertEqual(m.filters(), QgsMapLayerProxyModel.LineLayer | QgsMapLayerProxyModel.WritableLayer)
m.setExceptedLayerIds([l2.id()])
self.assertEqual(m.exceptedLayerIds(), [l2.id()])
m.setExceptedLayerList([l2])
self.assertEqual(m.exceptedLayerList(), [l2])
m.setLayerWhitelist([l2])
self.assertEqual(m.layerWhitelist(), [l2])
m.setExcludedProviders(['a', 'b'])
self.assertEqual(m.excludedProviders(), ['a', 'b'])
m.setFilterString('c')
self.assertEqual(m.filterString(), 'c')
def testMeshLayer(self):
m = QgsMapLayerProxyModel()
l1 = create_mesh_layer("l1")
QgsProject.instance().addMapLayer(l1)
l2 = create_layer('l2')
QgsProject.instance().addMapLayer(l2)
m.setFilters(QgsMapLayerProxyModel.MeshLayer)
self.assertEqual(m.filters(), QgsMapLayerProxyModel.MeshLayer)
self.assertEqual(m.rowCount(), 1)
self.assertEqual(m.data(m.index(0, 0)), 'l1')
def testMeshLayer(self):
m = QgsMapLayerProxyModel()
l1 = create_mesh_layer("l1")
QgsProject.instance().addMapLayer(l1)
l2 = create_layer('l2')
QgsProject.instance().addMapLayer(l2)
m.setFilters(QgsMapLayerProxyModel.MeshLayer)
self.assertEqual(m.filters(), QgsMapLayerProxyModel.MeshLayer)
self.assertEqual(m.rowCount(), 1)
self.assertEqual(m.data(m.index(0, 0)), 'l1')
def testAnnotationLayer(self):
"""
Test filtering annotation layers
"""
QgsProject.instance().clear()
m = QgsMapLayerProxyModel()
options = QgsAnnotationLayer.LayerOptions(
QgsProject.instance().transformContext())
l1 = QgsAnnotationLayer('annotation 1', options)
QgsProject.instance().addMapLayer(l1)
l2 = create_layer('l2')
QgsProject.instance().addMapLayer(l2)
m.setFilters(QgsMapLayerProxyModel.AnnotationLayer)
self.assertEqual(m.filters(), QgsMapLayerProxyModel.AnnotationLayer)
self.assertEqual(m.rowCount(), 1)
self.assertEqual(m.data(m.index(0, 0)), 'annotation 1')
self.assertTrue(m.acceptsLayer(l1))
self.assertFalse(m.acceptsLayer(l2))
m.setFilters(QgsMapLayerProxyModel.VectorLayer)
self.assertEqual(m.rowCount(), 1)
self.assertEqual(m.data(m.index(0, 0)), 'l2')
self.assertFalse(m.acceptsLayer(l1))
self.assertTrue(m.acceptsLayer(l2))
m.setFilters(QgsMapLayerProxyModel.All)
self.assertEqual(m.rowCount(), 2)
self.assertEqual(m.data(m.index(0, 0)), 'annotation 1')
self.assertEqual(m.data(m.index(1, 0)), 'l2')
self.assertTrue(m.acceptsLayer(l1))
self.assertTrue(m.acceptsLayer(l2))
