def testBasic(self):
renderer = QgsPointCloudClassifiedRenderer()
renderer.setAttribute('attr')
self.assertEqual(renderer.attribute(), 'attr')
renderer.setCategories([
QgsPointCloudCategory(3, QColor(255, 0, 0), 'cat 3'),
QgsPointCloudCategory(7, QColor(0, 255, 0), 'cat 7')
])
self.assertEqual(len(renderer.categories()), 2)
self.assertEqual(renderer.categories()[0].label(), 'cat 3')
self.assertEqual(renderer.categories()[1].label(), 'cat 7')
renderer.setMaximumScreenError(18)
renderer.setMaximumScreenErrorUnit(QgsUnitTypes.RenderInches)
renderer.setPointSize(13)
renderer.setPointSizeUnit(QgsUnitTypes.RenderPoints)
renderer.setPointSizeMapUnitScale(QgsMapUnitScale(1000, 2000))
rr = renderer.clone()
self.assertEqual(rr.maximumScreenError(), 18)
self.assertEqual(rr.maximumScreenErrorUnit(),
QgsUnitTypes.RenderInches)
self.assertEqual(rr.pointSize(), 13)
self.assertEqual(rr.pointSizeUnit(), QgsUnitTypes.RenderPoints)
self.assertEqual(rr.pointSizeMapUnitScale().minScale, 1000)
self.assertEqual(rr.pointSizeMapUnitScale().maxScale, 2000)
self.assertEqual(rr.attribute(), 'attr')
self.assertEqual(len(rr.categories()), 2)
self.assertEqual(rr.categories()[0].label(), 'cat 3')
self.assertEqual(rr.categories()[1].label(), 'cat 7')
doc = QDomDocument("testdoc")
elem = renderer.save(doc, QgsReadWriteContext())
r2 = QgsPointCloudClassifiedRenderer.create(elem,
QgsReadWriteContext())
self.assertEqual(r2.maximumScreenError(), 18)
self.assertEqual(r2.maximumScreenErrorUnit(),
QgsUnitTypes.RenderInches)
self.assertEqual(r2.pointSize(), 13)
self.assertEqual(r2.pointSizeUnit(), QgsUnitTypes.RenderPoints)
self.assertEqual(r2.pointSizeMapUnitScale().minScale, 1000)
self.assertEqual(r2.pointSizeMapUnitScale().maxScale, 2000)
self.assertEqual(r2.attribute(), 'attr')
self.assertEqual(len(r2.categories()), 2)
self.assertEqual(r2.categories()[0].label(), 'cat 3')
self.assertEqual(r2.categories()[1].label(), 'cat 7')
def testLegend(self):
renderer = QgsPointCloudClassifiedRenderer()
renderer.setAttribute('Classification')
renderer.setCategories([QgsPointCloudCategory(3, QColor(255, 0, 0), 'cat 3'),
QgsPointCloudCategory(7, QColor(0, 255, 0), 'cat 7')])
layer = QgsPointCloudLayer(unitTestDataPath() + '/point_clouds/ept/sunshine-coast/ept.json', 'test', 'ept')
layer_tree_layer = QgsLayerTreeLayer(layer)
nodes = renderer.createLegendNodes(layer_tree_layer)
self.assertEqual(len(nodes), 2)
self.assertEqual(nodes[0].data(Qt.DisplayRole), 'cat 3')
self.assertEqual(nodes[0].data(QgsLayerTreeModelLegendNode.RuleKeyRole), '3')
self.assertEqual(nodes[1].data(Qt.DisplayRole), 'cat 7')
self.assertEqual(nodes[1].data(QgsLayerTreeModelLegendNode.RuleKeyRole), '7')