More provider check handling of extra fields

qgis · Feb 20, 2018 · 4af24bf · 4af24bf
1 parent 7711f13
commit 4af24bf
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/src/python/featuresourcetestbase.py b/tests/src/python/featuresourcetestbase.py
@@ -75,9 +75,9 @@ def testGetFeatures(self, source=None, extra_features=[], skip_features=[], chan
         while it.nextFeature(f):
             # expect feature to be valid
             self.assertTrue(f.isValid())
-            # split off the first 5 attributes only - some source test datasets will include
-            # additional attributes which we ignore
-            attrs = f.attributes()[0:5]
+            # some source test datasets will include additional attributes which we ignore,
+            # so cherry pick desired attributes
+            attrs = [f['pk'], f['cnt'], f['name'], f['name2'], f['num_char']]
             # force the num_char attribute to be text - some sources (e.g., delimited text) will
             # automatically detect that this attribute contains numbers and set it as a numeric
             # field