sharpetronics/QGIS
1
0
Fork 0
mirror of https://github.com/qgis/QGIS.git synced 2025-02-25 00:58:06 -05:00
Code Issues Packages Projects Releases Wiki Activity

Add test for qgsfunction.handlesnull

Browse Source
This commit is contained in:
Matthias Kuhn 2018-07-01 06:17:25 -05:00
parent 60730abba7
commit 4d5eb49f1c
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/src/python/test_qgsexpression.py
View File

@ -67,6 +67,11 @@ class TestQgsExpressionCustomFunctions(unittest.TestCase):
def referenced_columns_set(values, feature, parent):
return 2
@qgsfunction(args=-1, group='testing', register=False, handlesnull=True)
def null_mean(values, feature, parent):
vals = [val for val in values if val != NULL]
return sum(vals) / len(vals)
def tearDown(self):
QgsExpression.unregisterFunction('testfun')
@ -161,6 +166,11 @@ class TestQgsExpressionCustomFunctions(unittest.TestCase):
exp = QgsExpression('referenced_columns_set()')
self.assertEqual(set(exp.referencedColumns()), set(['a', 'b']))
def testHandlesNull(self):
QgsExpression.registerFunction(self.handles_null)
exp = QgsExpression('null_mean(1, 2, NULL, 3)')
self.assertEqual(set(exp.evaluate()), 2)
def testCantOverrideBuiltinsWithUnregister(self):
success = QgsExpression.unregisterFunction("sqrt")
self.assertFalse(success)