Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
[processing] added test for SAGA output to non-ascii files
  • Loading branch information
volaya committed Jan 25, 2019
1 parent d627b56 commit 8bbd559
Showing 1 changed file with 56 additions and 1 deletion.
57 changes: 56 additions & 1 deletion python/plugins/processing/tests/SagaAlgorithmsTest.py
Expand Up @@ -29,7 +29,16 @@
import shutil

from qgis.core import (QgsProcessingParameterNumber,
QgsProcessingParameterDefinition)
QgsProcessingParameterDefinition,
QgsVectorLayer,
QgsApplication,
QgsFeature,
QgsGeometry,
QgsPointXY,
QgsProcessingContext,
QgsProject,
QgsProcessingFeedback,
QgsProcessingFeatureSourceDefinition)
from qgis.testing import start_app, unittest

from processing.algs.saga.SagaParameters import Parameters, SagaImageOutputParam
Expand All @@ -45,6 +54,9 @@ def setUpClass(cls):
Processing.initialize()
cls.cleanup_paths = []

cls.temp_dir = tempfile.mkdtemp()
cls.cleanup_paths.append(cls.temp_dir)

@classmethod
def tearDownClass(cls):
from processing.core.Processing import Processing
Expand Down Expand Up @@ -82,6 +94,49 @@ def test_param_line(self):
self.assertEqual(param.defaultFileExtension(), 'tif')
self.assertEqual(param.supportedOutputRasterLayerExtensions(), ['tif'])

def test_non_ascii_output(self):
# create a memory layer and add to project and context
layer = QgsVectorLayer("Point?crs=epsg:3857&field=fldtxt:string&field=fldint:integer",
"testmem", "memory")
self.assertTrue(layer.isValid())
pr = layer.dataProvider()
f = QgsFeature()
f.setAttributes(["test", 123])
f.setGeometry(QgsGeometry.fromPointXY(QgsPointXY(100, 200)))
f2 = QgsFeature()
f2.setAttributes(["test2", 457])
f2.setGeometry(QgsGeometry.fromPointXY(QgsPointXY(110, 200)))
self.assertTrue(pr.addFeatures([f, f2]))
self.assertEqual(layer.featureCount(), 2)
QgsProject.instance().addMapLayer(layer)
context = QgsProcessingContext()
context.setProject(QgsProject.instance())

alg = QgsApplication.processingRegistry().createAlgorithmById('saga:fixeddistancebuffer')
self.assertIsNotNone(alg)

temp_file = os.path.join(self.temp_dir, 'non_ascii_ñññ.gpkg')
parameters = {'SHAPES':'testmem',
'DIST_FIELD_DEFAULT':5,
'NZONES':1,
'DARC':5,
'DISSOLVE':True,
'POLY_INNER':False,
'BUFFER':temp_file}
feedback = QgsProcessingFeedback()

results, ok = alg.run(parameters, context, feedback)
self.assertTrue(ok)
self.assertTrue(os.path.exists(temp_file))

# make sure that layer has correct features
res = QgsVectorLayer(temp_file, 'res')
self.assertTrue(res.isValid())
self.assertEqual(res.featureCount(), 2)

QgsProject.instance().removeMapLayer(layer)



if __name__ == '__main__':
nose2.main()

0 comments on commit 8bbd559

Please sign in to comment.