Tests QgsProcessingParameterNumber

qgis · May 4, 2021 · 44a5c97 · 44a5c97
1 parent 7b97bd9
commit 44a5c97
Showing 1 changed file with 39 additions and 0 deletions.
diff --git a/python/plugins/processing/tests/ParametersTest.py b/python/plugins/processing/tests/ParametersTest.py
@@ -83,6 +83,45 @@ def testParameterStringDesc(self):
         self.assertFalse(param.multiLine())
         self.assertTrue(param.flags() & QgsProcessingParameterDefinition.FlagOptional)
 
+    def testParameterNumberDesc(self):
+        desc = 'QgsProcessingParameterNumber|in_number|Input number'
+        param = getParameterFromString(desc)
+        self.assertIsNotNone(param)
+        self.assertEqual(param.type(), 'number')
+        self.assertEqual(param.name(), 'in_number')
+        self.assertEqual(param.description(), 'Input Number')
+        self.assertEqual(param.dataType(), QgsProcessingParameterNumber.Integer)
+        self.assertFalse(param.flags() & QgsProcessingParameterDefinition.FlagOptional)
+
+        desc = 'QgsProcessingParameterNumber|in_number|Input number|QgsProcessingParameterNumber.Double'
+        param = getParameterFromString(desc)
+        self.assertIsNotNone(param)
+        self.assertEqual(param.type(), 'number')
+        self.assertEqual(param.name(), 'in_number')
+        self.assertEqual(param.description(), 'Input Number')
+        self.assertEqual(param.dataType(), QgsProcessingParameterNumber.Double)
+        self.assertFalse(param.flags() & QgsProcessingParameterDefinition.FlagOptional)
+
+        desc = 'QgsProcessingParameterNumber|in_number|Input number|QgsProcessingParameterNumber.Integer|10'
+        param = getParameterFromString(desc)
+        self.assertIsNotNone(param)
+        self.assertEqual(param.type(), 'number')
+        self.assertEqual(param.name(), 'in_number')
+        self.assertEqual(param.description(), 'Input Number')
+        self.assertEqual(param.dataType(), QgsProcessingParameterNumber.Integer)
+        self.assertEqual(param.defaultValue(), 10)
+        self.assertFalse(param.flags() & QgsProcessingParameterDefinition.FlagOptional)
+
+        desc = 'QgsProcessingParameterNumber|in_number|Input number|QgsProcessingParameterNumber.Integer|None|True'
+        param = getParameterFromString(desc)
+        self.assertIsNotNone(param)
+        self.assertEqual(param.type(), 'number')
+        self.assertEqual(param.name(), 'in_number')
+        self.assertEqual(param.description(), 'Input Number')
+        self.assertEqual(param.dataType(), QgsProcessingParameterNumber.Integer)
+        self.assertIsNone(param.defaultValue())
+        self.assertTrue(param.flags() & QgsProcessingParameterDefinition.FlagOptional)
+
 
 if __name__ == '__main__':
     unittest.main()