Skip to content

Commit 54776eb

Browse files
committed
Better debugging for provider tests
1 parent 0a84fbd commit 54776eb

File tree

1 file changed

+13
-15
lines changed

1 file changed

+13
-15
lines changed

tests/src/python/providertestbase.py

+13-15
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,7 @@ def runGetFeatureTests(self, provider):
6767
self.assert_query(provider, 'true', [1, 2, 3, 4, 5])
6868
self.assert_query(provider, 'false', [])
6969

70-
#Three value logic
70+
# Three value logic
7171
self.assert_query(provider, 'false and false', [])
7272
self.assert_query(provider, 'false and true', [])
7373
self.assert_query(provider, 'false and NULL', [])
@@ -90,7 +90,7 @@ def runGetFeatureTests(self, provider):
9090
self.assert_query(provider, 'not false', [1, 2, 3, 4, 5])
9191
self.assert_query(provider, 'not null', [])
9292

93-
#not
93+
# not
9494
self.assert_query(provider, 'not name = \'Apple\'', [1, 3, 4])
9595
self.assert_query(provider, 'not name IS NULL', [1, 2, 3, 4])
9696
self.assert_query(provider, 'not name = \'Apple\' or name = \'Apple\'', [1, 2, 3, 4])
@@ -121,17 +121,15 @@ def testGetFeaturesFilterRectTests(self):
121121
def testRectAndExpression(self):
122122
extent = QgsRectangle(-70, 67, -60, 80)
123123
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))])
127125
expected = [4]
128126
assert set(expected) == result, 'Expected {} and got {} when testing for combination of filterRect and expression'.format(set(expected), result)
129127

130128
def testGetFeaturesLimit(self):
131129
it = self.provider.getFeatures(QgsFeatureRequest().setLimit(2))
132130
features = [f['pk'] for f in it]
133131
assert len(features) == 2, 'Expected two features, got {} instead'.format(len(features))
134-
#fetch one feature
132+
# fetch one feature
135133
feature = QgsFeature()
136134
assert not it.nextFeature(feature), 'Expected no feature after limit, got one'
137135
it.rewind()
@@ -142,7 +140,7 @@ def testGetFeaturesLimit(self):
142140
it.rewind()
143141
features = [f['pk'] for f in it]
144142
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
146144
try:
147145
self.disableCompiler()
148146
except AttributeError:
@@ -157,32 +155,32 @@ def testGetFeaturesLimit(self):
157155
it = self.provider.getFeatures(QgsFeatureRequest().setLimit(2).setFilterExpression('cnt <= 100'))
158156
features = [f['pk'] for f in it]
159157
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
161159
it = self.provider.getFeatures(QgsFeatureRequest().setLimit(3).setFilterExpression('cnt <= 100'))
162160
features = [f['pk'] for f in it]
163161
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
165163
it = self.provider.getFeatures(QgsFeatureRequest().setLimit(1).setFilterExpression('cnt <= 100'))
166164
features = [f['pk'] for f in it]
167165
assert 1 in features or 5 in features, 'Expected either 1 or 5 for expression and feature limit, Got {} instead'.format(features)
168166

169167
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')
172170

173171
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')
176174

177175
def testExtent(self):
178176
reference = QgsGeometry.fromRect(
179177
QgsRectangle(-71.123, 66.33, -65.32, 78.3))
180178
provider_extent = QgsGeometry.fromRect(self.provider.extent())
181179

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())
183181

184182
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]))
186184
assert set([u'Apple', u'Honey', u'Orange', u'Pear', NULL]) == set(self.provider.uniqueValues(2)), 'Got {}'.format(set(self.provider.uniqueValues(2)))
187185

188186
def testFeatureCount(self):

0 commit comments

Comments
 (0)