A test for atlas feature extent after rotation

qgis · Jun 18, 2017 · 391f76b · 391f76b
1 parent 1aa2a4e
commit 391f76b
Showing 1 changed file with 67 additions and 3 deletions.
diff --git a/tests/src/python/test_qgsatlascomposition.py b/tests/src/python/test_qgsatlascomposition.py
@@ -22,9 +22,25 @@
 from qgis.testing import start_app, unittest
 from utilities import unitTestDataPath
 from qgis.PyQt.QtCore import QFileInfo, QRectF, qWarning
-from qgis.core import QgsVectorLayer, QgsProject, QgsCoordinateReferenceSystem, \
-    QgsComposition, QgsFillSymbol, QgsSingleSymbolRenderer, QgsComposerLabel, QgsComposerMap, QgsFontUtils, \
-    QgsRectangle, QgsComposerLegend, QgsFeature, QgsGeometry, QgsPointXY, QgsRendererCategory, QgsCategorizedSymbolRenderer, QgsMarkerSymbol
+from qgis.core import (
+    QgsCategorizedSymbolRenderer,
+    QgsComposerLabel,
+    QgsComposerLegend,
+    QgsComposerMap,
+    QgsComposition,
+    QgsCoordinateReferenceSystem,
+    QgsFeature,
+    QgsFillSymbol,
+    QgsFontUtils,
+    QgsGeometry,
+    QgsMarkerSymbol,
+    QgsPointXY,
+    QgsProject,
+    QgsRectangle,
+    QgsRendererCategory,
+    QgsSingleSymbolRenderer,
+    QgsVectorLayer,
+)
 from qgscompositionchecker import QgsCompositionChecker
 
 start_app()
@@ -113,6 +129,7 @@ def testCase(self):
         self.predefinedscales_render_test()
         self.hidden_render_test()
         self.legend_test()
+        self.rotation_test()
 
         shutil.rmtree(tmppath, True)
 
@@ -311,6 +328,53 @@ def legend_test(self):
         self.mComposition.removeComposerItem(legend)
         QgsProject.instance().removeMapLayer(ptLayer.id())
 
+    def rotation_test(self):
+        # We will create a polygon layer with a rotated rectangle.
+        # Then we will make it the object layer for the atlas,
+        # rotate the map and test that the bounding rectangle
+        # is smaller than the bounds without rotation.
+        polygonLayer = QgsVectorLayer('Polygon', 'test_polygon', 'memory')
+        poly = QgsFeature(polygonLayer.pendingFields())
+        points = [(10, 15), (15, 10), (45, 40), (40, 45)]
+        poly.setGeometry(QgsGeometry.fromPolygon([[QgsPointXY(x[0], x[1]) for x in points]]))
+        polygonLayer.dataProvider().addFeatures([poly])
+        QgsProject.instance().addMapLayer(polygonLayer)
+
+        # Recreating the composer locally
+        composition = QgsComposition(QgsProject.instance())
+        composition.setPaperSize(297, 210)
+
+        # the atlas map
+        atlasMap = QgsComposerMap(composition, 20, 20, 130, 130)
+        atlasMap.setFrameEnabled(True)
+        atlasMap.setLayers([polygonLayer])
+        atlasMap.setNewExtent(QgsRectangle(0, 0, 100, 50))
+        composition.addComposerMap(atlasMap)
+
+        # the atlas
+        atlas = composition.atlasComposition()
+        atlas.setCoverageLayer(polygonLayer)
+        atlas.setEnabled(True)
+        composition.setAtlasMode(QgsComposition.ExportAtlas)
+
+        atlasMap.setAtlasDriven(True)
+        atlasMap.setAtlasScalingMode(QgsComposerMap.Auto)
+        atlasMap.setAtlasMargin(0.0)
+
+        # Testing
+        atlasMap.setMapRotation(0.0)
+        atlas.firstFeature()
+        nonRotatedExtent = QgsRectangle(atlasMap.currentMapExtent())
+
+        atlasMap.setMapRotation(45.0)
+        atlas.firstFeature()
+        rotatedExtent = QgsRectangle(atlasMap.currentMapExtent())
+
+        assert rotatedExtent.width() < nonRotatedExtent.width() * 0.9
+        assert rotatedExtent.height() < nonRotatedExtent.height() * 0.9
+
+        QgsProject.instance().removeMapLayer(polygonLayer)
+
 
 if __name__ == '__main__':
     unittest.main()