14
14
15
15
import os
16
16
from PyQt4 .QtCore import (QStringList ,
17
- QFileInfo ,
18
-
19
- )
17
+ QFileInfo )
18
+ from PyQt4 .QtXml import QDomDocument
20
19
21
20
from qgis .core import (QgsComposerMap ,
22
21
QgsRectangle ,
25
24
QgsMapRenderer ,
26
25
QgsMapLayerRegistry ,
27
26
QgsMultiBandColorRenderer
28
- )
27
+ )
29
28
from utilities import (unitTestDataPath ,
30
29
getQgisTestApp ,
31
30
TestCase ,
32
31
unittest
33
32
#expectedFailure
34
- )
33
+ )
35
34
from qgscompositionchecker import QgsCompositionChecker
36
35
37
36
QGISAPP , CANVAS , IFACE , PARENT = getQgisTestApp ()
@@ -43,15 +42,15 @@ class TestQgsComposerMap(TestCase):
43
42
def __init__ (self , methodName ):
44
43
"""Run once on class initialisation."""
45
44
unittest .TestCase .__init__ (self , methodName )
46
- myPath = os .path .join (TEST_DATA_DIR , " landsat.tif" )
45
+ myPath = os .path .join (TEST_DATA_DIR , ' landsat.tif' )
47
46
rasterFileInfo = QFileInfo (myPath )
48
47
mRasterLayer = QgsRasterLayer (rasterFileInfo .filePath (),
49
48
rasterFileInfo .completeBaseName ())
50
49
rasterRenderer = QgsMultiBandColorRenderer (
51
50
mRasterLayer .dataProvider (), 2 , 3 , 4 )
52
- mRasterLayer .setRenderer ( rasterRenderer )
51
+ mRasterLayer .setRenderer (rasterRenderer )
53
52
#pipe = mRasterLayer.pipe()
54
- #assert pipe.set(rasterRenderer), " Cannot set pipe renderer"
53
+ #assert pipe.set(rasterRenderer), ' Cannot set pipe renderer'
55
54
QgsMapLayerRegistry .instance ().addMapLayer (mRasterLayer )
56
55
57
56
# create composition with composer map
@@ -93,16 +92,15 @@ def testGrid(self):
93
92
QgsComposerMap .Bottom )
94
93
checker = QgsCompositionChecker ()
95
94
myPath = os .path .join (TEST_DATA_DIR ,
96
- " control_images" ,
97
- " expected_composermap" ,
98
- " composermap_landsat_grid.png" )
99
- testResult = checker .testComposition (" Composer map grid" ,
95
+ ' control_images' ,
96
+ ' expected_composermap' ,
97
+ ' composermap_landsat_grid.png' )
98
+ myTestResult , myMessage = checker .testComposition (' Composer map grid' ,
100
99
self .mComposition , myPath )
101
100
self .mComposerMap .setGridEnabled (False )
102
101
self .mComposerMap .setShowGridAnnotation (False )
103
102
104
- print testResult
105
- assert testResult [0 ] == True
103
+ assert myTestResult [0 ] == True , myMessage
106
104
107
105
def testOverviewMap (self ):
108
106
overviewMap = QgsComposerMap (self .mComposition , 20 , 130 , 70 , 70 )
@@ -118,21 +116,22 @@ def testOverviewMap(self):
118
116
overviewMap .setOverviewFrameMap (self .mComposerMap .id ())
119
117
checker = QgsCompositionChecker ()
120
118
myPngPath = os .path .join (TEST_DATA_DIR ,
121
- "control_images" ,
122
- "expected_composermap" ,
123
- "composermap_landsat_overview.png" )
124
- testResult = checker .testComposition ("Composer map overview" ,
125
- self .mComposition ,
126
- myPngPath )
119
+ 'control_images' ,
120
+ 'expected_composermap' ,
121
+ 'composermap_landsat_overview.png' )
122
+ myTestResult , myMessage = checker .testComposition (
123
+ 'Composer map overview' ,
124
+ self .mComposition ,
125
+ myPngPath )
127
126
self .mComposition .removeComposerItem (overviewMap )
128
- assert testResult [ 0 ] == True
127
+ assert myTestResult == True , myMessage
129
128
130
129
131
- def uniqueId (self , mComposerMap , mComposition ):
130
+ def testuniqueId (self , mComposerMap , mComposition ):
132
131
doc = QDomDocument ()
133
- documentElement = doc .createElement ( " ComposerItemClipboard" )
134
- mComposerMap .writeXML ( documentElement , doc )
135
- mComposition .addItemsFromXML ( documentElement , doc , 0 , false )
132
+ documentElement = doc .createElement (' ComposerItemClipboard' )
133
+ mComposerMap .writeXML (documentElement , doc )
134
+ mComposition .addItemsFromXML (documentElement , doc , 0 , False )
136
135
137
136
#test if both composer maps have different ids
138
137
newMap = QgsComposerMap ()
@@ -146,22 +145,22 @@ def uniqueId(self, mComposerMap, mComposition):
146
145
oldId = mComposerMap .id ()
147
146
newId = newMap .id ()
148
147
149
- mComposition .removeComposerItem ( newMap );
150
- print "old: " + str (oldId )
151
- print "new " + str (newId )
152
- assert oldId != newId
148
+ mComposition .removeComposerItem (newMap )
149
+ myMessage = 'old: %s new: %s' % (oldId , newId )
150
+ assert oldId != newId , myMessage
153
151
154
- def zebraStyle (self ):
155
- mComposerMap .setGridFrameStyle ( QgsComposerMap .Zebra )
156
- mComposerMap .setGridEnabled ( True )
152
+ def testZebraStyle (self ):
153
+ self .mComposerMap .setGridFrameStyle (QgsComposerMap .Zebra )
154
+ self .mComposerMap .setGridEnabled (True )
155
+ checker = QgsCompositionChecker ()
157
156
myPngPath = os .path .join (TEST_DATA_DIR ,
158
- " control_images" ,
159
- " expected_composermap" ,
160
- " composermap_zebra_style.png" )
161
- testResult = checker .testComposition (" Composer map zebra" ,
157
+ ' control_images' ,
158
+ ' expected_composermap' ,
159
+ ' composermap_zebra_style.png' )
160
+ testResult , myMessage = checker .testComposition (' Composer map zebra' ,
162
161
self .mComposition ,
163
162
myPngPath )
164
- assert testResult [ 0 ] == True
163
+ assert testResult == True , myMessage
165
164
166
165
if __name__ == '__main__' :
167
166
unittest .main ()
0 commit comments