@@ -67,7 +67,7 @@ def runGetFeatureTests(self, provider):
67
67
self .assert_query (provider , 'true' , [1 , 2 , 3 , 4 , 5 ])
68
68
self .assert_query (provider , 'false' , [])
69
69
70
- #Three value logic
70
+ # Three value logic
71
71
self .assert_query (provider , 'false and false' , [])
72
72
self .assert_query (provider , 'false and true' , [])
73
73
self .assert_query (provider , 'false and NULL' , [])
@@ -90,7 +90,7 @@ def runGetFeatureTests(self, provider):
90
90
self .assert_query (provider , 'not false' , [1 , 2 , 3 , 4 , 5 ])
91
91
self .assert_query (provider , 'not null' , [])
92
92
93
- #not
93
+ # not
94
94
self .assert_query (provider , 'not name = \' Apple\' ' , [1 , 3 , 4 ])
95
95
self .assert_query (provider , 'not name IS NULL' , [1 , 2 , 3 , 4 ])
96
96
self .assert_query (provider , 'not name = \' Apple\' or name = \' Apple\' ' , [1 , 2 , 3 , 4 ])
@@ -121,17 +121,15 @@ def testGetFeaturesFilterRectTests(self):
121
121
def testRectAndExpression (self ):
122
122
extent = QgsRectangle (- 70 , 67 , - 60 , 80 )
123
123
result = set ([f ['pk' ] for f in self .provider .getFeatures (
124
- QgsFeatureRequest ()
125
- .setFilterExpression ('"cnt">200' )
126
- .setFilterRect (extent ))])
124
+ QgsFeatureRequest ().setFilterExpression ('"cnt">200' ).setFilterRect (extent ))])
127
125
expected = [4 ]
128
126
assert set (expected ) == result , 'Expected {} and got {} when testing for combination of filterRect and expression' .format (set (expected ), result )
129
127
130
128
def testGetFeaturesLimit (self ):
131
129
it = self .provider .getFeatures (QgsFeatureRequest ().setLimit (2 ))
132
130
features = [f ['pk' ] for f in it ]
133
131
assert len (features ) == 2 , 'Expected two features, got {} instead' .format (len (features ))
134
- #fetch one feature
132
+ # fetch one feature
135
133
feature = QgsFeature ()
136
134
assert not it .nextFeature (feature ), 'Expected no feature after limit, got one'
137
135
it .rewind ()
@@ -142,7 +140,7 @@ def testGetFeaturesLimit(self):
142
140
it .rewind ()
143
141
features = [f ['pk' ] for f in it ]
144
142
assert len (features ) == 2 , 'Expected two features after rewind, got {} instead' .format (len (features ))
145
- #test with expression, both with and without compilation
143
+ # test with expression, both with and without compilation
146
144
try :
147
145
self .disableCompiler ()
148
146
except AttributeError :
@@ -157,32 +155,32 @@ def testGetFeaturesLimit(self):
157
155
it = self .provider .getFeatures (QgsFeatureRequest ().setLimit (2 ).setFilterExpression ('cnt <= 100' ))
158
156
features = [f ['pk' ] for f in it ]
159
157
assert set (features ) == set ([1 , 5 ]), 'Expected [1,5] for expression and feature limit, Got {} instead' .format (features )
160
- #limit to more features than exist
158
+ # limit to more features than exist
161
159
it = self .provider .getFeatures (QgsFeatureRequest ().setLimit (3 ).setFilterExpression ('cnt <= 100' ))
162
160
features = [f ['pk' ] for f in it ]
163
161
assert set (features ) == set ([1 , 5 ]), 'Expected [1,5] for expression and feature limit, Got {} instead' .format (features )
164
- #limit to less features than possible
162
+ # limit to less features than possible
165
163
it = self .provider .getFeatures (QgsFeatureRequest ().setLimit (1 ).setFilterExpression ('cnt <= 100' ))
166
164
features = [f ['pk' ] for f in it ]
167
165
assert 1 in features or 5 in features , 'Expected either 1 or 5 for expression and feature limit, Got {} instead' .format (features )
168
166
169
167
def testMinValue (self ):
170
- assert self .provider .minimumValue (1 ) == - 200
171
- assert self .provider .minimumValue (2 ) == 'Apple'
168
+ self .assertEqual ( self . provider .minimumValue (1 ), - 200 )
169
+ self .assertEqual ( self . provider .minimumValue (2 ), 'Apple' )
172
170
173
171
def testMaxValue (self ):
174
- assert self .provider .maximumValue (1 ) == 400
175
- assert self .provider .maximumValue (2 ) == 'Pear'
172
+ self .assertEqual ( self . provider .maximumValue (1 ), 400 )
173
+ self .assertEqual ( self . provider .maximumValue (2 ), 'Pear' )
176
174
177
175
def testExtent (self ):
178
176
reference = QgsGeometry .fromRect (
179
177
QgsRectangle (- 71.123 , 66.33 , - 65.32 , 78.3 ))
180
178
provider_extent = QgsGeometry .fromRect (self .provider .extent ())
181
179
182
- assert QgsGeometry .compare (provider_extent .asPolygon (), reference .asPolygon (), 0.000001 )
180
+ assert QgsGeometry .compare (provider_extent .asPolygon (), reference .asPolygon (), 0.00001 ), 'Expected {}, got {}' . format ( reference . exportToWkt (), provider_extent . exportToWkt () )
183
181
184
182
def testUnique (self ):
185
- assert set (self .provider .uniqueValues (1 )) == set ([- 200 , 100 , 200 , 300 , 400 ])
183
+ self . assertEqual ( set (self .provider .uniqueValues (1 )), set ([- 200 , 100 , 200 , 300 , 400 ]) )
186
184
assert set ([u'Apple' , u'Honey' , u'Orange' , u'Pear' , NULL ]) == set (self .provider .uniqueValues (2 )), 'Got {}' .format (set (self .provider .uniqueValues (2 )))
187
185
188
186
def testFeatureCount (self ):
0 commit comments