Unit tests for QgsVectorLayerUtils.matchAttributesToFields

qgis · Sep 24, 2018 · 95e95d0 · 95e95d0
1 parent a879400
commit 95e95d0
Showing 1 changed file with 88 additions and 0 deletions.
diff --git a/tests/src/python/test_qgsvectorlayerutils.py b/tests/src/python/test_qgsvectorlayerutils.py
@@ -14,6 +14,7 @@
 
 import qgis  # NOQA
 
+from qgis.PyQt.QtCore import QVariant
 from qgis.core import (QgsProject,
                        QgsVectorLayer,
                        QgsVectorLayerUtils,
@@ -24,6 +25,11 @@
                        QgsPointXY,
                        QgsDefaultValue,
                        QgsRelation,
+                       QgsFields,
+                       QgsField,
+                       QgsMemoryProviderUtils,
+                       QgsWkbTypes,
+                       QgsCoordinateReferenceSystem,
                        NULL
                        )
 from qgis.testing import start_app, unittest
@@ -423,6 +429,88 @@ def testDuplicateFeature(self):
         # - check if the ids are still the same
         self.assertEqual(copyValueList, origValueList)
 
+    def test_make_features_compatible_attributes(self):
+        """Test corner cases for attributes"""
+
+        # Test feature with attributes
+        fields = QgsFields()
+        fields.append(QgsField('int_f', QVariant.Int))
+        fields.append(QgsField('str_f', QVariant.String))
+        f1 = QgsFeature(fields)
+        f1['int_f'] = 1
+        f1['str_f'] = 'str'
+        f1.setGeometry(QgsGeometry.fromWkt('Point(9 45)'))
+        f2 = f1
+        QgsVectorLayerUtils.matchAttributesToFields(f2, fields)
+        self.assertEqual(f1.attributes(), f2.attributes())
+        self.assertTrue(f1.geometry().asWkt(), f2.geometry().asWkt())
+
+        # Test pad with 0 with fields
+        f1.setAttributes([])
+        QgsVectorLayerUtils.matchAttributesToFields(f1, fields)
+        self.assertEqual(len(f1.attributes()), 2)
+        self.assertEqual(f1.attributes()[0], QVariant())
+        self.assertEqual(f1.attributes()[1], QVariant())
+
+        # Test pad with 0 without fields
+        f1 = QgsFeature()
+        QgsVectorLayerUtils.matchAttributesToFields(f1, fields)
+        self.assertEqual(len(f1.attributes()), 2)
+        self.assertEqual(f1.attributes()[0], QVariant())
+        self.assertEqual(f1.attributes()[1], QVariant())
+
+        # Test drop extra attrs
+        f1 = QgsFeature(fields)
+        f1.setAttributes([1, 'foo', 'extra'])
+        QgsVectorLayerUtils.matchAttributesToFields(f1, fields)
+        self.assertEqual(len(f1.attributes()), 2)
+        self.assertEqual(f1.attributes()[0], 1)
+        self.assertEqual(f1.attributes()[1], 'foo')
+
+        # Rearranged fields
+        fields2 = QgsFields()
+        fields2.append(QgsField('str_f', QVariant.String))
+        fields2.append(QgsField('int_f', QVariant.Int))
+        f1 = QgsFeature(fields2)
+        f1.setAttributes([1, 'foo', 'extra'])
+        QgsVectorLayerUtils.matchAttributesToFields(f1, fields)
+        self.assertEqual(len(f1.attributes()), 2)
+        self.assertEqual(f1.attributes()[0], 'foo')
+        self.assertEqual(f1.attributes()[1], 1)
+
+        # mixed
+        fields2.append(QgsField('extra', QVariant.String))
+        fields.append(QgsField('extra2', QVariant.Int))
+        f1.setFields(fields2)
+        f1.setAttributes([1, 'foo', 'blah'])
+        QgsVectorLayerUtils.matchAttributesToFields(f1, fields)
+        self.assertEqual(len(f1.attributes()), 3)
+        self.assertEqual(f1.attributes()[0], 'foo')
+        self.assertEqual(f1.attributes()[1], 1)
+        self.assertEqual(f1.attributes()[2], QVariant())
+
+        fields.append(QgsField('extra', QVariant.Int))
+        f1.setAttributes([1, 'foo', 'blah'])
+        QgsVectorLayerUtils.matchAttributesToFields(f1, fields)
+        self.assertEqual(len(f1.attributes()), 4)
+        self.assertEqual(f1.attributes()[0], 'foo')
+        self.assertEqual(f1.attributes()[1], 1)
+        self.assertEqual(f1.attributes()[2], QVariant())
+        self.assertEqual(f1.attributes()[3], 'blah')
+
+        # case insensitive
+        fields2.append(QgsField('extra3', QVariant.String))
+        fields.append(QgsField('EXTRA3', QVariant.Int))
+        f1.setFields(fields2)
+        f1.setAttributes([1, 'foo', 'blah', 'blergh'])
+        QgsVectorLayerUtils.matchAttributesToFields(f1, fields)
+        self.assertEqual(len(f1.attributes()), 5)
+        self.assertEqual(f1.attributes()[0], 'foo')
+        self.assertEqual(f1.attributes()[1], 1)
+        self.assertEqual(f1.attributes()[2], QVariant())
+        self.assertEqual(f1.attributes()[3], 'blah')
+        self.assertEqual(f1.attributes()[4], 'blergh')
+
 
 if __name__ == '__main__':
     unittest.main()