Merge pull request #2761 from m-kuhn/ptest

Updates to processing tests
qgis · Feb 23, 2016 · 8582f61 · 8582f61
2 parents 50aa805 + 5f0173e
commit 8582f61
Show file tree

Hide file tree

Showing 22 changed files with 210 additions and 2,660 deletions.
diff --git a/ci/travis/linux/before_install.sh b/ci/travis/linux/before_install.sh
@@ -11,6 +11,7 @@ sudo apt-get install --force-yes --no-install-recommends --no-install-suggests \
         cmake-data \
         doxygen \
         flex \
+        gdal-bin \
         git \
         graphviz \
         grass-dev \

diff --git a/python/plugins/processing/gui/TestTools.py b/python/plugins/processing/gui/TestTools.py
@@ -102,7 +102,7 @@ def createTest(text):
     definition['name'] = 'Test ({})'.format(cmdname)
     definition['algorithm'] = cmdname
 
-    params = []
+    params = {}
     results = {}
 
     i = 0
@@ -123,7 +123,7 @@ def createTest(text):
             if not schema:
                 p['location'] = '[The source data is not in the testdata directory. Please use data in the processing/tests/testdata folder.]'
 
-            params.append(p)
+            params[param.name] = p
         elif isinstance(param, ParameterRaster):
             filename = token[1:-1]
             schema, filepath = extractSchemaPath(filename)
@@ -134,7 +134,7 @@ def createTest(text):
             if not schema:
                 p['location'] = '[The source data is not in the testdata directory. Please use data in the processing/tests/testdata folder.]'
 
-            params.append(p)
+            params[param.name] = p
         elif isinstance(param, ParameterMultipleInput):
             multiparams = token[1:-1].split(';')
             newparam = []
@@ -151,9 +151,9 @@ def createTest(text):
             if not schema:
                 p['location'] = '[The source data is not in the testdata directory. Please use data in the processing/tests/testdata folder.]'
 
-            params.append(p)
+            params[param.name] = p
         else:
-            params.append(token)
+            params[param.name] = token
 
     definition['params'] = params
 

diff --git a/...lugins/processing/tests/AlgorithmsTest.py → ...ns/processing/tests/AlgorithmsTestBase.py b/...lugins/processing/tests/AlgorithmsTest.py → ...ns/processing/tests/AlgorithmsTestBase.py
@@ -2,7 +2,7 @@
 
 """
 ***************************************************************************
-    test_algorithms.py
+    AlgorithmsTest.py
     ---------------------
     Date                 : January 2016
     Copyright            : (C) 2016 by Matthias Kuhn
@@ -27,12 +27,12 @@
 
 import qgis
 import os
-import shutil
 import yaml
 import nose2
 import gdal
 import hashlib
 import tempfile
+import re
 
 from osgeo.gdalconst import GA_ReadOnly
 
@@ -46,11 +46,6 @@
     QgsMapLayerRegistry
 )
 
-from qgis.testing import (
-    start_app,
-    unittest
-)
-
 from utilities import (
     unitTestDataPath
 )
@@ -60,25 +55,13 @@ def processingTestDataPath():
     return os.path.join(os.path.dirname(__file__), 'testdata')
 
 
-class TestAlgorithms(unittest.TestCase):
-
-    @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)
+class AlgorithmsTest():
 
     def test_algorithms(self):
         """
         This is the main test function. All others will be executed based on the definitions in testdata/algorithm_tests.yaml
         """
-        with open(os.path.join(processingTestDataPath(), 'algorithm_tests.yaml'), 'r') as stream:
+        with open(os.path.join(processingTestDataPath(), self.test_definition_file()), 'r') as stream:
             algorithm_tests = yaml.load(stream)
 
         for algtest in algorithm_tests['tests']:
@@ -104,8 +87,8 @@ def check_algorithm(self, name, defs):
         for r, p in defs['results'].iteritems():
             alg.setOutputValue(r, self.load_result_param(p))
 
-        self.assertTrue(AlgorithmExecutor.runalg(alg))
         print(alg.getAsCommand())
+        self.assertTrue(AlgorithmExecutor.runalg(alg))
         self.check_results(alg.getOutputValuesAsDictionary(), defs['results'])
 
     def load_params(self, params):
@@ -133,21 +116,27 @@ def load_param(self, param):
             # No type specified, use whatever is there
             return param
 
-        raise KeyError("Unknown type '{}' specified for parameter '{}'".format(param['type'], param['name']))
+        raise KeyError("Unknown type '{}' specified for parameter".format(param['type']))
 
     def load_result_param(self, param):
         """
         Loads a result parameter. Creates a temporary destination where the result should go to and returns this location
         so it can be sent to the algorithm as parameter.
         """
-        if param['type'] in ['vector', 'file']:
+        if param['type'] in ['vector', 'file', 'regex']:
             outdir = tempfile.mkdtemp()
             self.cleanup_paths.append(outdir)
             basename = os.path.basename(param['name'])
             filepath = os.path.join(outdir, basename)
             return filepath
+        elif param['type'] == 'rasterhash':
+            outdir = tempfile.mkdtemp()
+            self.cleanup_paths.append(outdir)
+            basename = 'raster.tif'
+            filepath = os.path.join(outdir, basename)
+            return filepath
 
-        raise KeyError("Unknown type '{}' specified for parameter '{}'".format(param['type'], param['name']))
+        raise KeyError("Unknown type '{}' specified for parameter".format(param['type']))
 
     def load_layer(self, param):
         """
@@ -188,10 +177,7 @@ def check_results(self, results, expected):
 
                 result_lyr = QgsVectorLayer(results[id], id, 'ogr')
 
-                try:
-                    compare = expected_result['compare']
-                except KeyError:
-                    compare = {}
+                compare = expected_result.get('compare', {})
 
                 self.assertLayersEqual(expected_lyr, result_lyr, compare=compare)
 
@@ -205,6 +191,12 @@ def check_results(self, results, expected):
                 result_filepath = results[id]
 
                 self.assertFilesEqual(expected_filepath, result_filepath)
+            elif 'regex' == expected_result['type']:
+                with open(results[id], 'r') as file:
+                    data = file.read()
+
+                for rule in expected_result.get('rules', []):
+                    self.assertRegexpMatches(data, rule)
 
 
 if __name__ == '__main__':

diff --git a/python/plugins/processing/tests/CMakeLists.txt b/python/plugins/processing/tests/CMakeLists.txt
@@ -7,5 +7,6 @@ PLUGIN_INSTALL(processing tests/data ${TEST_DATA_FILES})
 IF(ENABLE_TESTS)
   INCLUDE(UsePythonTest)
   ADD_PYTHON_TEST(ProcessingParametersTest ParametersTest.py)
-  ADD_PYTHON_TEST(ProcessingAlgorithmsTest AlgorithmsTest.py)
+  ADD_PYTHON_TEST(ProcessingQgisAlgorithmsTest QgisAlgorithmsTest.py)
+  ADD_PYTHON_TEST(ProcessingGdalAlgorithmsTest GdalAlgorithmsTest.py)
 ENDIF(ENABLE_TESTS)
diff --git a/python/plugins/processing/tests/GdalAlgorithmsTest.py b/python/plugins/processing/tests/GdalAlgorithmsTest.py
@@ -0,0 +1,58 @@
+# -*- coding: utf-8 -*-
+
+"""
+***************************************************************************
+    GdalAlgorithmTests.py
+    ---------------------
+    Date                 : January 2016
+    Copyright            : (C) 2016 by Matthias Kuhn
+    Email                : matthias@opengis.ch
+***************************************************************************
+*                                                                         *
+*   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__ = 'Matthias Kuhn'
+__date__ = 'January 2016'
+__copyright__ = '(C) 2016, Matthias Kuhn'
+
+# 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 TestGdalAlgorithms(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 'gdal_algorithm_tests.yaml'
+
+
+if __name__ == '__main__':
+    nose2.main()