tests for rasterize_over

qgis · Nov 4, 2019 · 17daa48 · 17daa48
1 parent 20e542c
commit 17daa48
Showing 1 changed file with 36 additions and 0 deletions.
diff --git a/python/plugins/processing/tests/GdalAlgorithmsRasterTest.py b/python/plugins/processing/tests/GdalAlgorithmsRasterTest.py
@@ -67,6 +67,7 @@
 from processing.algs.gdal.merge import merge
 from processing.algs.gdal.nearblack import nearblack
 from processing.algs.gdal.slope import slope
+from processing.algs.gdal.rasterize_over import rasterize_over
 
 testDataPath = os.path.join(os.path.dirname(__file__), 'testdata')
 
@@ -1503,6 +1504,41 @@ def testRasterize(self):
                  source + ' ' +
                  outdir + '/check.jpg'])
 
+    def testRasterizeOver(self):
+        context = QgsProcessingContext()
+        feedback = QgsProcessingFeedback()
+        raster = os.path.join(testDataPath, 'dem.tif')
+        vector = os.path.join(testDataPath, 'polys.gml')
+        alg = rasterize_over()
+        alg.initAlgorithm()
+
+        with tempfile.TemporaryDirectory() as outdir:
+            self.assertEqual(
+                alg.getConsoleCommands({'INPUT': vector,
+                                        'FIELD': 'id',
+                                        'INPUT_RASTER': raster}, context, feedback),
+                ['gdal_rasterize',
+                 '-l polys2 -a id ' +
+                 vector + ' ' + raster])
+
+            self.assertEqual(
+                alg.getConsoleCommands({'INPUT': vector,
+                                        'FIELD': 'id',
+                                        'ADD': True,
+                                        'INPUT_RASTER': raster}, context, feedback),
+                ['gdal_rasterize',
+                 '-l polys2 -a id -add ' +
+                 vector + ' ' + raster])
+
+            self.assertEqual(
+                alg.getConsoleCommands({'INPUT': vector,
+                                        'FIELD': 'id',
+                                        'EXTRA': '-i',
+                                        'INPUT_RASTER': raster}, context, feedback),
+                ['gdal_rasterize',
+                 '-l polys2 -a id -i ' +
+                 vector + ' ' + raster])
+
     def testRetile(self):
         context = QgsProcessingContext()
         feedback = QgsProcessingFeedback()