def test_convert_group_to_group_layer(self):
"""
Test converting a QgsLayerTreeGroup to a QgsGroupLayer
"""
group_node = QgsLayerTreeGroup()
options = QgsGroupLayer.LayerOptions(QgsCoordinateTransformContext())
group_layer = group_node.convertToGroupLayer(options)
self.assertFalse(group_layer.childLayers())
self.assertEqual(group_node.groupLayer(), group_layer)
# if a group layer is already assigned, convertToGroupLayer should do nothing
self.assertIsNone(group_node.convertToGroupLayer(options))
group_node.setGroupLayer(None)
# add some child layers to node
layer = QgsVectorLayer("Point?field=fldtxt:string", "layer1", "memory")
group_node.addLayer(layer)
layer2 = QgsVectorLayer("Point?field=fldtxt:string", "layer2",
"memory")
group_node.addLayer(layer2)
group_layer = group_node.convertToGroupLayer(options)
self.assertEqual(group_layer.childLayers(), [layer2, layer])
self.assertEqual(group_node.groupLayer(), group_layer)
def test_group_layer_updates_from_node(self):
"""
Test that group layer child layers are synced correctly from the group node
"""
group_node = QgsLayerTreeGroup('my group')
options = QgsGroupLayer.LayerOptions(QgsCoordinateTransformContext())
group_layer = group_node.convertToGroupLayer(options)
self.assertFalse(group_layer.childLayers())
layer = QgsVectorLayer("Point?field=fldtxt:string", "layer1", "memory")
group_node.addLayer(layer)
self.assertEqual(group_layer.childLayers(), [layer])
layer2 = QgsVectorLayer("Point?field=fldtxt:string", "layer2",
"memory")
group_node.insertLayer(0, layer2)
self.assertEqual(group_layer.childLayers(), [layer, layer2])
layer3 = QgsVectorLayer("Point?field=fldtxt:string", "layer3",
"memory")
layer4 = QgsVectorLayer("Point?field=fldtxt:string", "layer4",
"memory")
layer3_node = QgsLayerTreeLayer(layer3)
layer4_node = QgsLayerTreeLayer(layer4)
group_node.insertChildNodes(1, [layer3_node, layer4_node])
self.assertEqual(group_layer.childLayers(),
[layer, layer4, layer3, layer2])
layer5 = QgsVectorLayer("Point?field=fldtxt:string", "layer5",
"memory")
layer5_node = QgsLayerTreeLayer(layer5)
group_node.addChildNode(layer5_node)
self.assertEqual(group_layer.childLayers(),
[layer5, layer, layer4, layer3, layer2])
group_node.removeChildNode(layer3_node)
self.assertEqual(group_layer.childLayers(),
[layer5, layer, layer4, layer2])
group_node.removeLayer(layer)
self.assertEqual(group_layer.childLayers(), [layer5, layer4, layer2])
group_node.removeChildren(0, 2)
self.assertEqual(group_layer.childLayers(), [layer5])
group_node.removeAllChildren()
self.assertEqual(group_layer.childLayers(), [])
def test_group_layer_nested(self):
"""
Test group node with child nodes converted to group layer
"""
group_node = QgsLayerTreeGroup('my group')
options = QgsGroupLayer.LayerOptions(QgsCoordinateTransformContext())
group_layer = group_node.convertToGroupLayer(options)
self.assertFalse(group_layer.childLayers())
layer = QgsVectorLayer("Point?field=fldtxt:string", "layer1", "memory")
group_node.addLayer(layer)
group2 = group_node.addGroup('child group 1')
layer2 = QgsVectorLayer("Point?field=fldtxt:string", "layer2",
"memory")
group_node.addLayer(layer2)
layer3 = QgsVectorLayer("Point?field=fldtxt:string", "layer3",
"memory")
layer3_node = group2.addLayer(layer3)
group3 = group2.addGroup('grand child group 1')
group4 = group2.addGroup('grand child group 2')
layer4 = QgsVectorLayer("Point?field=fldtxt:string", "layer4",
"memory")
layer4_node = group3.addLayer(layer4)
layer5 = QgsVectorLayer("Point?field=fldtxt:string", "layer5",
"memory")
layer5_node = group4.addLayer(layer5)
self.assertEqual(group_layer.childLayers(),
[layer2, layer5, layer4, layer3, layer])
layer5_node.setItemVisibilityChecked(False)
self.assertEqual(group_layer.childLayers(),
[layer2, layer4, layer3, layer])
group2.setItemVisibilityChecked(False)
self.assertEqual(group_layer.childLayers(), [layer2, layer])
group2.setItemVisibilityCheckedRecursive(True)
self.assertEqual(group_layer.childLayers(),
[layer2, layer5, layer4, layer3, layer])
def test_group_layer_updates_from_node_visibility(self):
"""
Test that group layer child layers are synced correctly from the group node when layer visibility is changed
"""
group_node = QgsLayerTreeGroup('my group')
options = QgsGroupLayer.LayerOptions(QgsCoordinateTransformContext())
group_layer = group_node.convertToGroupLayer(options)
self.assertFalse(group_layer.childLayers())
layer = QgsVectorLayer("Point?field=fldtxt:string", "layer1", "memory")
group_node.addLayer(layer)
layer2 = QgsVectorLayer("Point?field=fldtxt:string", "layer2",
"memory")
layer2_node = group_node.addLayer(layer2)
layer3 = QgsVectorLayer("Point?field=fldtxt:string", "layer3",
"memory")
group_node.addLayer(layer3)
self.assertEqual(group_layer.childLayers(), [layer3, layer2, layer])
layer2_node.setItemVisibilityChecked(False)
self.assertEqual(group_layer.childLayers(), [layer3, layer])
layer2_node.setItemVisibilityChecked(True)
self.assertEqual(group_layer.childLayers(), [layer3, layer2, layer])
