Add more tests for point cloud rgb renderer

qgis · Nov 16, 2020 · d09a99d · d09a99d
1 parent f2135b3
commit d09a99d
Show file tree

Hide file tree

Showing 2 changed files with 76 additions and 1 deletion.
diff --git a/tests/src/python/test_qgspointcloudrgbrenderer.py b/tests/src/python/test_qgspointcloudrgbrenderer.py
@@ -15,9 +15,19 @@
 from qgis.core import (
     QgsProviderRegistry,
     QgsPointCloudLayer,
-    QgsPointCloudRgbRenderer
+    QgsPointCloudRgbRenderer,
+    QgsReadWriteContext,
+    QgsRenderContext,
+    QgsPointCloudRenderContext,
+    QgsVector3D,
+    QgsMultiRenderChecker,
+    QgsMapSettings,
+    QgsRectangle
 )
 
+from qgis.PyQt.QtCore import QDir, QSize
+from qgis.PyQt.QtXml import QDomDocument
+
 from qgis.testing import start_app, unittest
 from utilities import unitTestDataPath
 
@@ -26,6 +36,16 @@
 
 class TestQgsPointCloudRgbRenderer(unittest.TestCase):
 
+    @classmethod
+    def setUpClass(cls):
+        cls.report = "<h1>Python QgsPointCloudRgbRenderer Tests</h1>\n"
+
+    @classmethod
+    def tearDownClass(cls):
+        report_file_path = "%s/qgistest.html" % QDir.tempPath()
+        with open(report_file_path, 'a') as report_file:
+            report_file.write(cls.report)
+
     @unittest.skipIf('ept' not in QgsProviderRegistry.instance().providerList(), 'EPT provider not available')
     def testSetLayer(self):
         layer = QgsPointCloudLayer(unitTestDataPath() + '/point_clouds/ept/rgb/ept.json', 'test', 'ept')
@@ -34,6 +54,61 @@ def testSetLayer(self):
         # test that a point cloud with RGB attributes is automatically assigned the RGB renderer by default
         self.assertIsInstance(layer.renderer(), QgsPointCloudRgbRenderer)
 
+    def testBasic(self):
+        renderer = QgsPointCloudRgbRenderer()
+        renderer.setBlueAttribute('b')
+        self.assertEqual(renderer.blueAttribute(), 'b')
+        renderer.setGreenAttribute('g')
+        self.assertEqual(renderer.greenAttribute(), 'g')
+        renderer.setRedAttribute('r')
+        self.assertEqual(renderer.redAttribute(), 'r')
+
+        rr = renderer.clone()
+        self.assertEqual(rr.blueAttribute(), 'b')
+        self.assertEqual(rr.greenAttribute(), 'g')
+        self.assertEqual(rr.redAttribute(), 'r')
+
+        doc = QDomDocument("testdoc")
+        elem = renderer.save(doc, QgsReadWriteContext())
+
+        r2 = QgsPointCloudRgbRenderer.create(elem, QgsReadWriteContext())
+        self.assertEqual(r2.blueAttribute(), 'b')
+        self.assertEqual(r2.greenAttribute(), 'g')
+        self.assertEqual(r2.redAttribute(), 'r')
+
+    def testUsedAttributes(self):
+        renderer = QgsPointCloudRgbRenderer()
+        renderer.setBlueAttribute('b')
+        renderer.setGreenAttribute('g')
+        renderer.setRedAttribute('r')
+
+        rc = QgsRenderContext()
+        prc = QgsPointCloudRenderContext(rc, QgsVector3D(), QgsVector3D())
+
+        self.assertEqual(renderer.usedAttributes(prc), {'r', 'g', 'b'})
+
+    @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')
+        self.assertTrue(layer.isValid())
+
+        layer.renderer().setPenWidth(2)
+
+        mapsettings = QgsMapSettings()
+        mapsettings.setOutputSize(QSize(400, 400))
+        mapsettings.setOutputDpi(96)
+        mapsettings.setDestinationCrs(layer.crs())
+        mapsettings.setExtent(QgsRectangle(497753.5, 7050887.5, 497754.6, 7050888.6))
+        mapsettings.setLayers([layer])
+
+        renderchecker = QgsMultiRenderChecker()
+        renderchecker.setMapSettings(mapsettings)
+        renderchecker.setControlPathPrefix('pointcloudrenderer')
+        renderchecker.setControlName('expected_rgb_render')
+        result = renderchecker.runTest('expected_rgb_render')
+        TestQgsPointCloudRgbRenderer.report += renderchecker.report()
+        self.assertTrue(result)
+
 
 if __name__ == '__main__':
     unittest.main()
diff --git a/...a/control_images/pointcloudrenderer/expected_rgb_render/expected_rgb_render.png b/...a/control_images/pointcloudrenderer/expected_rgb_render/expected_rgb_render.png