|
62 | 62 | from processing.algs.gdal.gdaladdo import gdaladdo
|
63 | 63 | from processing.algs.gdal.sieve import sieve
|
64 | 64 | from processing.algs.gdal.gdal2xyz import gdal2xyz
|
| 65 | +from processing.algs.gdal.polygonize import polygonize |
65 | 66 |
|
66 | 67 | from processing.tools.system import isWindows
|
67 | 68 |
|
@@ -2547,6 +2548,56 @@ def testGdal2Xyz(self):
|
2547 | 2548 | source + ' ' +
|
2548 | 2549 | outsource])
|
2549 | 2550 |
|
| 2551 | + def testGdalPolygonize(self): |
| 2552 | + context = QgsProcessingContext() |
| 2553 | + feedback = QgsProcessingFeedback() |
| 2554 | + source = os.path.join(testDataPath, 'dem.tif') |
| 2555 | + |
| 2556 | + with tempfile.TemporaryDirectory() as outdir: |
| 2557 | + outsource = outdir + '/check.shp' |
| 2558 | + alg = polygonize() |
| 2559 | + alg.initAlgorithm() |
| 2560 | + |
| 2561 | + # defaults |
| 2562 | + self.assertEqual( |
| 2563 | + alg.getConsoleCommands({'INPUT': source, |
| 2564 | + 'BAND': 1, |
| 2565 | + 'FIELD': 'DN', |
| 2566 | + 'EIGHT_CONNECTEDNESS': False, |
| 2567 | + 'OUTPUT': outsource}, context, feedback), |
| 2568 | + ['gdal_polygonize.py', |
| 2569 | + source + ' ' + |
| 2570 | + outsource + ' ' + |
| 2571 | + '-b 1 -f "ESRI Shapefile" DN' |
| 2572 | + ]) |
| 2573 | + |
| 2574 | + # 8 connectedness |
| 2575 | + self.assertEqual( |
| 2576 | + alg.getConsoleCommands({'INPUT': source, |
| 2577 | + 'BAND': 1, |
| 2578 | + 'FIELD': 'DN', |
| 2579 | + 'EIGHT_CONNECTEDNESS': True, |
| 2580 | + 'OUTPUT': outsource}, context, feedback), |
| 2581 | + ['gdal_polygonize.py', |
| 2582 | + source + ' ' + |
| 2583 | + outsource + ' ' + |
| 2584 | + '-8 -b 1 -f "ESRI Shapefile" DN' |
| 2585 | + ]) |
| 2586 | + |
| 2587 | + # custom output format |
| 2588 | + outsource = outdir + '/check.gpkg' |
| 2589 | + self.assertEqual( |
| 2590 | + alg.getConsoleCommands({'INPUT': source, |
| 2591 | + 'BAND': 1, |
| 2592 | + 'FIELD': 'DN', |
| 2593 | + 'EIGHT_CONNECTEDNESS': False, |
| 2594 | + 'OUTPUT': outsource}, context, feedback), |
| 2595 | + ['gdal_polygonize.py', |
| 2596 | + source + ' ' + |
| 2597 | + outsource + ' ' + |
| 2598 | + '-b 1 -f "GPKG" DN' |
| 2599 | + ]) |
| 2600 | + |
2550 | 2601 |
|
2551 | 2602 | class TestGdalOgrToPostGis(unittest.TestCase):
|
2552 | 2603 |
|
|
0 commit comments