Add TestQgsSymbolV2 with tests for rendering different geometry types

qgis · Jan 21, 2016 · 1a7dede · 1a7dede
1 parent 0063381
commit 1a7dede
Show file tree

Hide file tree

Showing 11 changed files with 186 additions and 0 deletions.
diff --git a/tests/src/python/CMakeLists.txt b/tests/src/python/CMakeLists.txt
@@ -60,6 +60,7 @@ ADD_PYTHON_TEST(PyQgsSpatialiteProvider test_provider_spatialite.py)
 ADD_PYTHON_TEST(PyQgsSymbolLayerV2 test_qgssymbollayerv2.py)
 ADD_PYTHON_TEST(PyQgsSymbolExpressionVariables test_qgssymbolexpressionvariables.py)
 ADD_PYTHON_TEST(PyQgsSyntacticSugar test_syntactic_sugar.py)
+ADD_PYTHON_TEST(PyQgsSymbolV2 test_qgssymbolv2.py)
 ADD_PYTHON_TEST(PyQgsVectorColorRamp test_qgsvectorcolorramp.py)
 ADD_PYTHON_TEST(PyQgsVectorFileWriter test_qgsvectorfilewriter.py)
 ADD_PYTHON_TEST(PyQgsVectorLayer test_qgsvectorlayer.py)

diff --git a/tests/src/python/test_qgssymbolv2.py b/tests/src/python/test_qgssymbolv2.py
@@ -0,0 +1,185 @@
+# -*- coding: utf-8 -*-
+
+"""
+***************************************************************************
+    test_qgssymbolv2.py
+    ---------------------
+    Date                 : January 2016
+    Copyright            : (C) 2016 by Nyall Dawson
+    Email                : nyall dot dawson at gmail dot com
+***************************************************************************
+*                                                                         *
+*   This program is free software; you can redistribute it and/or modify  *
+*   it under the terms of the GNU General Public License as published by  *
+*   the Free Software Foundation; either version 2 of the License, or     *
+*   (at your option) any later version.                                   *
+*                                                                         *
+***************************************************************************
+"""
+
+__author__ = 'Nyall Dawson'
+__date__ = 'January 2016'
+__copyright__ = '(C) 2016, Nyall Dawson'
+# This will get replaced with a git SHA1 when you do a git archive
+__revision__ = '$Format:%H$'
+
+import os
+import csv
+
+from utilities import (unitTestDataPath,
+                       getQgisTestApp,
+                       TestCase,
+                       unittest,
+                       expectedFailure
+                       )
+
+from PyQt4.QtCore import QSize, QDir
+from PyQt4.QtGui import QImage, QColor, QPainter
+
+from qgis.core import (QgsGeometry,
+                       QgsMarkerSymbolV2,
+                       QgsFillSymbolV2,
+                       QgsLineSymbolV2,
+                       QgsRenderContext,
+                       QgsFeature,
+                       QGis,
+                       QgsMapSettings,
+                       QgsRectangle,
+                       QgsRenderChecker
+                       )
+
+QGISAPP, CANVAS, IFACE, PARENT = getQgisTestApp()
+TEST_DATA_DIR = unitTestDataPath()
+
+
+class TestQgsSymbolV2(TestCase):
+
+    def setUp(self):
+        #Create some simple symbols
+        self.fill_symbol = QgsFillSymbolV2.createSimple({'color': '#ffffff'})
+        self.line_symbol = QgsLineSymbolV2.createSimple({'color': '#ffffff', 'line_width': '3'})
+        self.marker_symbol = QgsMarkerSymbolV2.createSimple({'color': '#ffffff', 'size': '3'})
+        self.report = "<h1>Python QgsSymbolV2 Tests</h1>\n"
+
+    def tearDown(self):
+        report_file_path = "%s/qgistest.html" % QDir.tempPath()
+        with open(report_file_path, 'a') as report_file:
+            report_file.write(self.report)
+
+    def testGeometryRendering(self):
+        '''Tests rendering a bunch of different geometries, including bad/odd geometries.'''
+
+        tests = [{'name': 'Point',
+                  'wkt': 'Point (1 2)',
+                  'reference_image': 'point'},
+                 {'name': 'MultiPoint',
+                  'wkt': 'MultiPoint ((10 30),(40 20),(30 10),(20 10))',
+                  'reference_image': 'multipoint'},
+                 {'name': 'LineString',
+                  'wkt': 'LineString (0 0,3 4,4 3)',
+                  'reference_image': 'linestring'},
+                 {'name': 'MultiLineString',
+                  'wkt': 'MultiLineString ((0 0, 1 0, 1 1, 2 1, 2 0), (3 1, 5 1, 5 0, 6 0))',
+                  'reference_image': 'multilinestring'},
+                 {'name': 'Polygon',
+                  'wkt': 'Polygon ((0 0, 10 0, 10 10, 0 10, 0 0),(5 5, 7 5, 7 7 , 5 7, 5 5))',
+                  'reference_image': 'polygon'},
+                 {'name': 'MultiPolygon',
+                  'wkt': 'MultiPolygon (((0 0, 1 0, 1 1, 2 1, 2 2, 0 2, 0 0)),((4 0, 5 0, 5 2, 3 2, 3 1, 4 1, 4 0)))',
+                  'reference_image': 'multipolygon'},
+                 {'name': 'CircularString',
+                  'wkt': 'CIRCULARSTRING(268 415,227 505,227 406)',
+                  'reference_image': 'circular_string'},
+                 {'name': 'CompoundCurve',
+                  'wkt': 'COMPOUNDCURVE((5 3, 5 13), CIRCULARSTRING(5 13, 7 15, 9 13), (9 13, 9 3), CIRCULARSTRING(9 3, 7 1, 5 3))',
+                  'reference_image': 'compound_curve'},
+                 {'name': 'CurvePolygon',
+                  'wkt': 'CURVEPOLYGON(CIRCULARSTRING(1 3, 3 5, 4 7, 7 3, 1 3))',
+                  'reference_image': 'curve_polygon'}]
+
+        for test in tests:
+            geom = QgsGeometry.fromWkt(test['wkt'])
+            assert geom and not geom.isEmpty(), 'Could not create geometry'
+            rendered_image = self.renderGeometry(geom)
+            assert self.imageCheck(test['name'], test['reference_image'], rendered_image)
+
+            #Note - each test is repeated with the same geometry and reference image, but with added
+            #z and m dimensions. This tests that presence of the dimensions does not affect rendering
+
+            #test with Z
+            geom_z = QgsGeometry.fromWkt(test['wkt'])
+            geom_z.geometry().addZValue(5)
+            rendered_image = self.renderGeometry(geom_z)
+            assert self.imageCheck(test['name'] + ' z', test['reference_image'], rendered_image)
+
+            #test with ZM
+            geom_z.geometry().addMValue(15)
+            rendered_image = self.renderGeometry(geom_z)
+            assert self.imageCheck(test['name'] + ' zm', test['reference_image'], rendered_image)
+
+            #test with ZM
+            geom_m = QgsGeometry.fromWkt(test['wkt'])
+            geom_m.geometry().addMValue(15)
+            rendered_image = self.renderGeometry(geom_m)
+            assert self.imageCheck(test['name'] + ' m', test['reference_image'], rendered_image)
+
+    def renderGeometry(self, geom):
+        f = QgsFeature()
+        f.setGeometry(geom)
+
+        image = QImage(200, 200, QImage.Format_RGB32)
+
+        painter = QPainter()
+        ms = QgsMapSettings()
+        extent = geom.geometry().boundingBox()
+        # buffer extent by 10%
+        if extent.width() > 0:
+            extent = extent.buffer((extent.height() + extent.width()) / 20.0)
+        else:
+            extent = extent.buffer(10)
+
+        ms.setExtent(extent)
+        ms.setOutputSize(image.size())
+        context = QgsRenderContext.fromMapSettings(ms)
+        context.setPainter(painter)
+        context.setScaleFactor(96 / 25.4) # 96 DPI
+
+        painter.begin(image)
+        image.fill(QColor(0, 0, 0))
+
+        if geom.type() == QGis.Polygon:
+            self.fill_symbol.startRender(context)
+            self.fill_symbol.renderFeature(f, context)
+            self.fill_symbol.stopRender(context)
+
+        elif geom.type() == QGis.Line:
+            self.line_symbol.startRender(context)
+            self.line_symbol.renderFeature(f, context)
+            self.line_symbol.stopRender(context)
+
+        elif geom.type() == QGis.Point:
+            self.marker_symbol.startRender(context)
+            self.marker_symbol.renderFeature(f, context)
+            self.marker_symbol.stopRender(context)
+
+        painter.end()
+        return image
+
+    def imageCheck(self, name, reference_image, image):
+        self.report += "<h2>Render {}</h2>\n".format(name)
+        temp_dir = QDir.tempPath() + '/'
+        file_name = temp_dir + 'symbol_' + name + ".png"
+        image.save(file_name, "PNG")
+        checker = QgsRenderChecker()
+        checker.setControlPathPrefix("symbolv2")
+        checker.setControlName("expected_" + reference_image)
+        checker.setRenderedImage(file_name)
+        checker.setColorTolerance(2)
+        result = checker.compareImages(name, 20)
+        self.report += checker.report()
+        print self.report
+        return result
+
+
+if __name__ == '__main__':
+    unittest.main()# -*- coding: utf-8 -*-
diff --git a/...a/control_images/symbolv2/expected_circular_string/expected_circular_string.png b/...a/control_images/symbolv2/expected_circular_string/expected_circular_string.png
diff --git a/...ata/control_images/symbolv2/expected_compound_curve/expected_compound_curve.png b/...ata/control_images/symbolv2/expected_compound_curve/expected_compound_curve.png
diff --git a/...tdata/control_images/symbolv2/expected_curve_polygon/expected_curve_polygon.png b/...tdata/control_images/symbolv2/expected_curve_polygon/expected_curve_polygon.png
diff --git a/tests/testdata/control_images/symbolv2/expected_linestring/expected_linestring.png b/tests/testdata/control_images/symbolv2/expected_linestring/expected_linestring.png
diff --git a/...a/control_images/symbolv2/expected_multilinestring/expected_multilinestring.png b/...a/control_images/symbolv2/expected_multilinestring/expected_multilinestring.png
diff --git a/tests/testdata/control_images/symbolv2/expected_multipoint/expected_multipoint.png b/tests/testdata/control_images/symbolv2/expected_multipoint/expected_multipoint.png
diff --git a/...estdata/control_images/symbolv2/expected_multipolygon/expected_multipolygon.png b/...estdata/control_images/symbolv2/expected_multipolygon/expected_multipolygon.png
diff --git a/tests/testdata/control_images/symbolv2/expected_point/expected_point.png b/tests/testdata/control_images/symbolv2/expected_point/expected_point.png
diff --git a/tests/testdata/control_images/symbolv2/expected_polygon/expected_polygon.png b/tests/testdata/control_images/symbolv2/expected_polygon/expected_polygon.png