[processing] allow tests for scripts

qgis · Mar 16, 2017 · c603df1 · c603df1
1 parent 3bcf287
commit c603df1
Show file tree

Hide file tree

Showing 4 changed files with 95 additions and 1 deletion.
diff --git a/python/plugins/processing/tests/AlgorithmsTestBase.py b/python/plugins/processing/tests/AlgorithmsTestBase.py
@@ -45,6 +45,9 @@
 from numpy import nan_to_num
 
 import processing
+
+from processing.script.ScriptAlgorithm import ScriptAlgorithm  # NOQA
+
 from processing.modeler.ModelerAlgorithmProvider import ModelerAlgorithmProvider  # NOQA
 from processing.algs.qgis.QGISAlgorithmProvider import QGISAlgorithmProvider  # NOQA
 from processing.algs.grass7.Grass7AlgorithmProvider import Grass7AlgorithmProvider  # NOQA
@@ -90,7 +93,11 @@ def check_algorithm(self, name, defs):
 
         params = self.load_params(defs['params'])
 
-        alg = processing.Processing.getAlgorithm(defs['algorithm']).getCopy()
+        if defs['algorithm'].startswith('scrips:'):
+            filePath = os.path.join(processingTestDataPath(), 'scripts', '{}.py'.format(defs['algorithm'][len('script:'):]))
+            alg = ScriptAlgorithm(filePath)
+        else:
+            alg = processing.Processing.getAlgorithm(defs['algorithm']).getCopy()
 
         if isinstance(params, list):
             for param in zip(alg.parameters, params):

diff --git a/python/plugins/processing/tests/CMakeLists.txt b/python/plugins/processing/tests/CMakeLists.txt
@@ -11,6 +11,7 @@ IF(ENABLE_TESTS)
   ADD_PYTHON_TEST(ProcessingToolsTest ToolsTest.py)
   ADD_PYTHON_TEST(ProcessingQgisAlgorithmsTest QgisAlgorithmsTest.py)
   ADD_PYTHON_TEST(ProcessingGdalAlgorithmsTest GdalAlgorithmsTest.py)
+  ADD_PYTHON_TEST(ProcessingScriptAlgorithmsTest ScriptAlgorithmsTest.py)
   ADD_PYTHON_TEST(ProcessingGrass7AlgorithmsImageryTest Grass7AlgorithmsImageryTest.py)
   ADD_PYTHON_TEST(ProcessingGrass7AlgorithmsRasterTest Grass7AlgorithmsRasterTest.py)
 ENDIF(ENABLE_TESTS)
diff --git a/python/plugins/processing/tests/ScriptAlgorithmsTest.py b/python/plugins/processing/tests/ScriptAlgorithmsTest.py
@@ -0,0 +1,55 @@
+# -*- coding: utf-8 -*-
+
+"""
+***************************************************************************
+    ScriptAlgorithmsTests.py
+    ---------------------
+    Date                 : March 2017
+    Copyright            : (C) 2017 by Alexander Bruy
+    Email                : alexander dot bruy at gmail dot com
+***************************************************************************
+*                                                                         *
+*   This program is free software; you can redistribute it and/or modify  *
+*   it under the terms of the GNU General Public License as published by  *
+*   the Free Software Foundation; either version 2 of the License, or     *
+*   (at your option) any later version.                                   *
+*                                                                         *
+***************************************************************************
+"""
+
+__author__ = 'Alexander Bruy'
+__date__ = 'March 2017'
+__copyright__ = '(C) 2017, Alexander Bruy'
+
+# This will get replaced with a git SHA1 when you do a git archive
+
+__revision__ = ':%H$'
+
+import AlgorithmsTestBase
+
+import nose2
+import shutil
+
+from qgis.testing import start_app, unittest
+
+
+class TestScriptAlgorithms(unittest.TestCase, AlgorithmsTestBase.AlgorithmsTest):
+
+    @classmethod
+    def setUpClass(cls):
+        start_app()
+        from processing.core.Processing import Processing
+        Processing.initialize()
+        cls.cleanup_paths = []
+
+    @classmethod
+    def tearDownClass(cls):
+        for path in cls.cleanup_paths:
+            shutil.rmtree(path)
+
+    def test_definition_file(self):
+        return 'script_algorithm_tests.yaml'
+
+
+if __name__ == '__main__':
+    nose2.main()
diff --git a/python/plugins/processing/tests/testdata/scripts/centroids.py b/python/plugins/processing/tests/testdata/scripts/centroids.py
@@ -0,0 +1,31 @@
+##Centroids=name
+##Geometry=group
+##INPUT_LAYER=vector
+##OUTPUT_LAYER=output vector
+
+from qgis.core import QgsWkbTypes, QgsGeometry
+
+from processing.tools.vector import VectorWriter
+
+layer = processing.getObject(INPUT_LAYER)
+fields = layer.fields()
+
+writer = VectorWriter(OUTPUT_LAYER, 'utf-8', fields, QgsWkbTypes.Point, layer.crs())
+
+features = processing.features(layer)
+count = len(features)
+if count == 0:
+    raise GeoAlgorithmExecutionException('Input layer contains no features.')
+
+total = 100.0 / len(features)
+
+for count, f in enumerate(features):
+    outputFeature = f
+    if f.geometry():
+        outputGeometry = f.geometry().centroid()
+        outputFeature.setGeometry(outputGeometry)
+
+    writer.addFeature(outputFeature)
+    feedback.setProgress(int(count * total))
+
+del writer