Expand test

qgis · Nov 28, 2020 · 672b593 · 672b593
1 parent 0c7df50
commit 672b593
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/tests/src/python/test_qgspointcloudrgbrenderer.py b/tests/src/python/test_qgspointcloudrgbrenderer.py
@@ -26,7 +26,8 @@
     QgsContrastEnhancement,
     QgsUnitTypes,
     QgsMapUnitScale,
-    QgsCoordinateReferenceSystem
+    QgsCoordinateReferenceSystem,
+    QgsDoubleRange
 )
 
 from qgis.PyQt.QtCore import QDir, QSize
@@ -148,6 +149,11 @@ def testUsedAttributes(self):
 
         self.assertEqual(renderer.usedAttributes(prc), {'r', 'g', 'b'})
 
+        # if context is filtering by z, we also need the z attribute
+        rc.setZRange(QgsDoubleRange(1, 10))
+        prc = QgsPointCloudRenderContext(rc, QgsVector3D(), QgsVector3D())
+        self.assertEqual(renderer.usedAttributes(prc), {'r', 'g', 'b', 'Z'})
+
     @unittest.skipIf('ept' not in QgsProviderRegistry.instance().providerList(), 'EPT provider not available')
     def testRender(self):
         layer = QgsPointCloudLayer(unitTestDataPath() + '/point_clouds/ept/rgb/ept.json', 'test', 'ept')