[composer] Add some python tests for QgsComposerMapGrid

(sponsored by NIWA, New Zealand)
qgis · Sep 3, 2014 · c9e5341 · c9e5341
1 parent 1b77316
commit c9e5341
Show file tree

Hide file tree

Showing 3 changed files with 239 additions and 55 deletions.
diff --git a/tests/src/python/CMakeLists.txt b/tests/src/python/CMakeLists.txt
@@ -23,6 +23,7 @@ ADD_PYTHON_TEST(PyQgsAnalysis test_qgsanalysis.py)
 ADD_PYTHON_TEST(PyQgsColorScheme test_qgscolorscheme.py)
 ADD_PYTHON_TEST(PyQgsColorSchemeRegistry test_qgscolorschemeregistry.py)
 ADD_PYTHON_TEST(PyQgsComposerMap  test_qgscomposermap.py)
+ADD_PYTHON_TEST(PyQgsComposerMapGrid  test_qgscomposermapgrid.py)
 ADD_PYTHON_TEST(PyQgsComposerEffects  test_qgscomposereffects.py)
 ADD_PYTHON_TEST(PyQgsComposerShapes  test_qgscomposershapes.py)
 ADD_PYTHON_TEST(PyQgsSymbolLayerV2 test_qgssymbollayerv2.py)

diff --git a/tests/src/python/test_qgscomposermap.py b/tests/src/python/test_qgscomposermap.py
@@ -68,42 +68,6 @@ def __init__(self, methodName):
         self.mComposerMap.setFrameEnabled(True)
         self.mComposition.addComposerMap(self.mComposerMap)
 
-    def testGrid(self):
-        """Test that we can create a grid for a map."""
-        myRectangle = QgsRectangle(781662.375, 3339523.125,
-                                   793062.375, 3345223.125)
-        self.mComposerMap.setNewExtent(myRectangle)
-        self.mComposerMap.setGridEnabled(True)
-        self.mComposerMap.setGridIntervalX(2000)
-        self.mComposerMap.setGridIntervalY(2000)
-        self.mComposerMap.setShowGridAnnotation(True)
-        self.mComposerMap.setGridPenWidth(0.5)
-        self.mComposerMap.setGridPenColor(QColor(0,255,0))
-        self.mComposerMap.setGridAnnotationFont( QgsFontUtils.getStandardTestFont() )
-        self.mComposerMap.setGridAnnotationPrecision(0)
-        self.mComposerMap.setGridAnnotationPosition(QgsComposerMap.Disabled,
-                                                    QgsComposerMap.Left)
-        self.mComposerMap.setGridAnnotationPosition(
-            QgsComposerMap.OutsideMapFrame,
-            QgsComposerMap.Right)
-        self.mComposerMap.setGridAnnotationPosition(QgsComposerMap.Disabled,
-                                                    QgsComposerMap.Top)
-        self.mComposerMap.setGridAnnotationPosition(
-            QgsComposerMap.OutsideMapFrame,
-            QgsComposerMap.Bottom)
-        self.mComposerMap.setGridAnnotationDirection(QgsComposerMap.Horizontal,
-                                                     QgsComposerMap.Right)
-        self.mComposerMap.setGridAnnotationDirection(QgsComposerMap.Horizontal,
-                                                     QgsComposerMap.Bottom)
-        self.mComposerMap.setAnnotationFontColor(QColor(255,0,0,150))
-        self.mComposerMap.setGridBlendMode(QPainter.CompositionMode_Overlay)
-        checker = QgsCompositionChecker('composermap_grid', self.mComposition)
-        myTestResult, myMessage = checker.testComposition()
-        self.mComposerMap.setGridEnabled(False)
-        self.mComposerMap.setShowGridAnnotation(False)
-
-        assert myTestResult == True, myMessage
-
     def testOverviewMap(self):
         overviewMap = QgsComposerMap(self.mComposition, 20, 130, 70, 70)
         overviewMap.setFrameEnabled(True)
@@ -200,25 +164,6 @@ def testuniqueId(self):
         myMessage = 'old: %s new: %s'  % (oldId, newId)
         assert oldId != newId, myMessage
 
-    def testZebraStyle(self):
-        self.mComposerMap.setGridFrameStyle(QgsComposerMap.Zebra)
-        myRectangle = QgsRectangle(785462.375, 3341423.125,
-                                   789262.375, 3343323.125)
-        self.mComposerMap.setNewExtent( myRectangle )
-        self.mComposerMap.setGridEnabled(True)
-        self.mComposerMap.setGridIntervalX(2000)
-        self.mComposerMap.setGridIntervalY(2000)
-        self.mComposerMap.setGridFrameWidth( 10 )
-        self.mComposerMap.setGridFramePenSize( 1 )
-        self.mComposerMap.setGridPenWidth( 0.5 )
-        self.mComposerMap.setGridFramePenColor( QColor( 255, 100, 0, 200 ) )
-        self.mComposerMap.setGridFrameFillColor1( QColor( 50, 90, 50, 100 ) )
-        self.mComposerMap.setGridFrameFillColor2( QColor( 200, 220, 100, 60 ) )
-
-        checker = QgsCompositionChecker('composermap_zebrastyle', self.mComposition)
-        myTestResult, myMessage = checker.testComposition()
-        assert myTestResult == True, myMessage
-
     def testWorldFileGeneration( self ):
         myRectangle = QgsRectangle(781662.375, 3339523.125, 793062.375, 3345223.125)
         self.mComposerMap.setNewExtent( myRectangle )

diff --git a/tests/src/python/test_qgscomposermapgrid.py b/tests/src/python/test_qgscomposermapgrid.py
@@ -0,0 +1,238 @@
+# -*- coding: utf-8 -*-
+"""QGIS Unit tests for QgsComposerMap.
+
+.. note. 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__ = '(C) 2012 by Dr. Horst Düster / Dr. Marco Hugentobler'
+__date__ = '20/08/2012'
+__copyright__ = 'Copyright 2012, The QGIS Project'
+# This will get replaced with a git SHA1 when you do a git archive
+__revision__ = '$Format:%H$'
+
+import os
+import qgis
+from PyQt4.QtCore import QFileInfo
+from PyQt4.QtXml import QDomDocument
+from PyQt4.QtGui import (QPainter,
+                          QColor)
+
+from qgis.core import (QgsComposerMap,
+                       QgsComposerMapGrid,
+                       QgsRectangle,
+                       QgsComposition,
+                       QgsMapSettings,
+                       QgsMapLayerRegistry,
+                       QgsCoordinateReferenceSystem,
+                       QgsFontUtils
+                     )
+from utilities import (unitTestDataPath,
+                       getQgisTestApp,
+                       TestCase,
+                       unittest,
+                       expectedFailure
+                      )
+from qgscompositionchecker import QgsCompositionChecker
+
+QGISAPP, CANVAS, IFACE, PARENT = getQgisTestApp()
+TEST_DATA_DIR = unitTestDataPath()
+
+
+class TestQgsComposerMap(TestCase):
+
+    def __init__(self, methodName):
+        """Run once on class initialisation."""
+        unittest.TestCase.__init__(self, methodName)
+
+        # create composition with composer map
+        self.mMapSettings= QgsMapSettings()
+        crs = QgsCoordinateReferenceSystem( 32633 )
+        self.mMapSettings.setDestinationCrs( crs )
+        self.mMapSettings.setCrsTransformEnabled(False)
+        self.mComposition = QgsComposition(self.mMapSettings)
+        self.mComposition.setPaperSize(297, 210)
+        self.mComposerMap = QgsComposerMap(self.mComposition, 20, 20, 200, 100)
+        self.mComposerMap.setFrameEnabled(True)
+        self.mComposerMap.setBackgroundColor( QColor( 150, 100, 100 ) )
+        self.mComposition.addComposerMap(self.mComposerMap)
+
+    def testGrid(self):
+        """Test that we can create a grid for a map."""
+        myRectangle = QgsRectangle(781662.375, 3339523.125,
+                                   793062.375, 3345223.125)
+        self.mComposerMap.setNewExtent(myRectangle)
+        self.mComposerMap.grid().setEnabled( True )
+        self.mComposerMap.grid().setIntervalX( 2000 )
+        self.mComposerMap.grid().setIntervalY( 2000 )
+        self.mComposerMap.grid().setAnnotationEnabled( True )
+        self.mComposerMap.grid().setGridLineColor( QColor( 0, 255, 0 ) )
+        self.mComposerMap.grid().setGridLineWidth( 0.5 )
+        self.mComposerMap.grid().setAnnotationFont( QgsFontUtils.getStandardTestFont() )
+        self.mComposerMap.grid().setAnnotationPrecision( 0 )
+        self.mComposerMap.grid().setAnnotationPosition( QgsComposerMapGrid.Disabled, QgsComposerMapGrid.Left )
+        self.mComposerMap.grid().setAnnotationPosition( QgsComposerMapGrid.OutsideMapFrame, QgsComposerMapGrid.Right )
+        self.mComposerMap.grid().setAnnotationPosition( QgsComposerMapGrid.Disabled, QgsComposerMapGrid.Top )
+        self.mComposerMap.grid().setAnnotationPosition( QgsComposerMapGrid.OutsideMapFrame, QgsComposerMapGrid.Bottom )
+        self.mComposerMap.grid().setAnnotationDirection( QgsComposerMapGrid.Horizontal, QgsComposerMapGrid.Right )
+        self.mComposerMap.grid().setAnnotationDirection( QgsComposerMapGrid.Horizontal, QgsComposerMapGrid.Bottom )
+        self.mComposerMap.grid().setAnnotationFontColor( QColor( 255, 0, 0, 150 ) )
+        self.mComposerMap.grid().setBlendMode( QPainter.CompositionMode_Overlay )
+
+        checker = QgsCompositionChecker('composermap_grid', self.mComposition)
+        myTestResult, myMessage = checker.testComposition()
+        self.mComposerMap.setGridEnabled(False)
+        self.mComposerMap.setShowGridAnnotation(False)
+
+        assert myTestResult == True, myMessage
+
+    def testCrossGrid(self):
+        myRectangle = QgsRectangle( 781662.375, 3339523.125, 793062.375, 3345223.125 )
+        self.mComposerMap.setNewExtent( myRectangle )
+        self.mComposerMap.grid().setEnabled( True )
+        self.mComposerMap.grid().setStyle( QgsComposerMapGrid.Cross )
+        self.mComposerMap.grid().setCrossLength( 2.0 )
+        self.mComposerMap.grid().setIntervalX( 2000 )
+        self.mComposerMap.grid().setIntervalY( 2000 )
+        self.mComposerMap.grid().setAnnotationEnabled( False )
+        self.mComposerMap.grid().setGridLineColor( QColor( 0, 255, 0 ) )
+        self.mComposerMap.grid().setGridLineWidth( 0.5 )
+        self.mComposerMap.grid().setBlendMode( QPainter.CompositionMode_SourceOver )
+
+        checker = QgsCompositionChecker('composermap_crossgrid', self.mComposition)
+        myTestResult, myMessage = checker.testComposition()
+
+        self.mComposerMap.grid().setStyle( QgsComposerMapGrid.Solid )
+        self.mComposerMap.grid().setEnabled( False )
+        self.mComposerMap.grid().setAnnotationEnabled( False )
+
+        assert myTestResult == True, myMessage
+
+    def testMarkerGrid(self):
+        myRectangle = QgsRectangle( 781662.375, 3339523.125, 793062.375, 3345223.125 )
+        self.mComposerMap.setNewExtent( myRectangle )
+        self.mComposerMap.grid().setEnabled( True )
+        self.mComposerMap.grid().setStyle( QgsComposerMapGrid.Markers )
+        self.mComposerMap.grid().setCrossLength( 2.0 )
+        self.mComposerMap.grid().setIntervalX( 2000 )
+        self.mComposerMap.grid().setIntervalY( 2000 )
+        self.mComposerMap.grid().setAnnotationEnabled( False )
+        self.mComposerMap.grid().setBlendMode( QPainter.CompositionMode_SourceOver )
+
+        checker = QgsCompositionChecker('composermap_markergrid', self.mComposition)
+        myTestResult, myMessage = checker.testComposition()
+
+        self.mComposerMap.grid().setStyle( QgsComposerMapGrid.Solid )
+        self.mComposerMap.grid().setEnabled( False )
+        self.mComposerMap.grid().setAnnotationEnabled( False )
+
+        assert myTestResult == True, myMessage
+
+    def testFrameOnly(self):
+        myRectangle = QgsRectangle( 781662.375, 3339523.125, 793062.375, 3345223.125 )
+        self.mComposerMap.setNewExtent( myRectangle )
+        self.mComposerMap.grid().setEnabled( True )
+        self.mComposerMap.grid().setStyle( QgsComposerMapGrid.FrameAnnotationsOnly )
+        self.mComposerMap.grid().setIntervalX( 2000 )
+        self.mComposerMap.grid().setIntervalY( 2000 )
+        self.mComposerMap.grid().setAnnotationEnabled( False )
+        self.mComposerMap.grid().setFrameStyle( QgsComposerMapGrid.Zebra )
+        self.mComposerMap.grid().setBlendMode( QPainter.CompositionMode_SourceOver )
+
+        checker = QgsCompositionChecker('composermap_gridframeonly', self.mComposition)
+        myTestResult, myMessage = checker.testComposition()
+
+        self.mComposerMap.grid().setStyle( QgsComposerMapGrid.Solid )
+        self.mComposerMap.grid().setEnabled( False )
+        self.mComposerMap.grid().setAnnotationEnabled( False )
+        self.mComposerMap.grid().setFrameStyle( QgsComposerMapGrid.NoFrame )
+
+        assert myTestResult == True, myMessage
+
+    def testZebraStyle(self):
+        self.mComposerMap.setGridFrameStyle(QgsComposerMap.Zebra)
+        myRectangle = QgsRectangle(785462.375, 3341423.125,
+                                   789262.375, 3343323.125)
+        self.mComposerMap.setNewExtent( myRectangle )
+        self.mComposerMap.grid().setIntervalX( 2000 )
+        self.mComposerMap.grid().setIntervalY( 2000 )
+        self.mComposerMap.grid().setGridLineColor( QColor( 0, 0, 0 ) )
+        self.mComposerMap.grid().setAnnotationFontColor( QColor( 0, 0, 0 ) )
+        self.mComposerMap.grid().setBlendMode( QPainter.CompositionMode_SourceOver )
+        self.mComposerMap.grid().setFrameStyle( QgsComposerMapGrid.Zebra )
+        self.mComposerMap.grid().setFrameWidth( 10 )
+        self.mComposerMap.grid().setFramePenSize( 1 )
+        self.mComposerMap.grid().setGridLineWidth( 0.5 )
+        self.mComposerMap.grid().setFramePenColor( QColor( 255, 100, 0, 200 ) )
+        self.mComposerMap.grid().setFrameFillColor1( QColor( 50, 90, 50, 100 ) )
+        self.mComposerMap.grid().setFrameFillColor2( QColor( 200, 220, 100, 60 ) )
+        self.mComposerMap.grid().setEnabled( True )
+
+        checker = QgsCompositionChecker('composermap_zebrastyle', self.mComposition)
+        myTestResult, myMessage = checker.testComposition( 0, 100 )
+        assert myTestResult == True, myMessage
+
+    def testZebraStyleSides(self):
+        self.mComposerMap.setGridFrameStyle(QgsComposerMap.Zebra)
+        myRectangle = QgsRectangle(781662.375, 3339523.125, 793062.375, 3345223.125)
+        self.mComposerMap.setNewExtent( myRectangle )
+        self.mComposerMap.grid().setIntervalX( 2000 )
+        self.mComposerMap.grid().setIntervalY( 2000 )
+        self.mComposerMap.grid().setGridLineColor( QColor( 0, 0, 0 ) )
+        self.mComposerMap.grid().setAnnotationFontColor( QColor( 0, 0, 0 ) )
+        self.mComposerMap.grid().setBlendMode( QPainter.CompositionMode_SourceOver )
+        self.mComposerMap.grid().setFrameStyle( QgsComposerMapGrid.Zebra )
+        self.mComposerMap.grid().setFrameWidth( 10 )
+        self.mComposerMap.grid().setFramePenSize( 1 )
+        self.mComposerMap.grid().setGridLineWidth( 0.5 )
+        self.mComposerMap.grid().setFramePenColor( QColor( 0, 0, 0 ) )
+        self.mComposerMap.grid().setFrameFillColor1( QColor( 0, 0, 0 ) )
+        self.mComposerMap.grid().setFrameFillColor2( QColor( 255, 255, 255 ) )
+        self.mComposerMap.grid().setEnabled( True )
+
+        self.mComposerMap.grid().setFrameSideFlag( QgsComposerMapGrid.FrameLeft, True )
+        self.mComposerMap.grid().setFrameSideFlag( QgsComposerMapGrid.FrameRight, False )
+        self.mComposerMap.grid().setFrameSideFlag( QgsComposerMapGrid.FrameTop, False )
+        self.mComposerMap.grid().setFrameSideFlag( QgsComposerMapGrid.FrameBottom, False )
+
+        checker = QgsCompositionChecker('composermap_zebrastyle_left', self.mComposition)
+        myTestResult, myMessage = checker.testComposition( 0, 100 )
+        assert myTestResult == True, myMessage
+
+        self.mComposerMap.grid().setFrameSideFlag( QgsComposerMapGrid.FrameTop, True )
+        checker = QgsCompositionChecker('composermap_zebrastyle_lefttop', self.mComposition)
+        myTestResult, myMessage = checker.testComposition( 0, 100 )
+        assert myTestResult == True, myMessage
+
+        self.mComposerMap.grid().setFrameSideFlag( QgsComposerMapGrid.FrameRight, True )
+        checker = QgsCompositionChecker('composermap_zebrastyle_lefttopright', self.mComposition)
+        myTestResult, myMessage = checker.testComposition( 0, 100 )
+        assert myTestResult == True, myMessage
+
+        self.mComposerMap.grid().setFrameSideFlag( QgsComposerMapGrid.FrameBottom, True )
+        self.mComposerMap.grid().setFrameStyle( QgsComposerMapGrid.NoFrame )
+
+    def testInteriorTicks(self):
+        self.mComposerMap.setGridFrameStyle(QgsComposerMap.Zebra)
+        myRectangle = QgsRectangle( 781662.375, 3339523.125, 793062.375, 3345223.125)
+        self.mComposerMap.setNewExtent( myRectangle )
+        self.mComposerMap.grid().setIntervalX( 2000 )
+        self.mComposerMap.grid().setIntervalY( 2000 )
+        self.mComposerMap.grid().setAnnotationFontColor( QColor( 0, 0, 0 ) )
+        self.mComposerMap.grid().setBlendMode( QPainter.CompositionMode_SourceOver )
+        self.mComposerMap.grid().setFrameStyle( QgsComposerMapGrid.InteriorTicks )
+        self.mComposerMap.grid().setFrameWidth( 10 )
+        self.mComposerMap.grid().setFramePenSize( 1 )
+        self.mComposerMap.grid().setFramePenColor( QColor( 0, 0, 0 ) )
+        self.mComposerMap.grid().setEnabled( True )
+        self.mComposerMap.grid().setStyle( QgsComposerMapGrid.FrameAnnotationsOnly )
+
+        checker = QgsCompositionChecker('composermap_interiorticks', self.mComposition)
+        myTestResult, myMessage = checker.testComposition( 0, 100 )
+        assert myTestResult == True, myMessage
+
+
+if __name__ == '__main__':
+    unittest.main()
+