mirror of
https://github.com/qgis/QGIS.git
synced 2025-03-10 00:13:55 -04:00
97 lines
3.6 KiB
Python
97 lines
3.6 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
"""
|
||
|
|
***************************************************************************
|
||
|
|
QgisAlgorithmTests.py
|
||
|
|
---------------------
|
||
|
|
Date : January 2019
|
||
|
|
Copyright : (C) 2019 by Nyall Dawson
|
||
|
|
Email : nyall dot dawson 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__ = 'Nyall Dawson'
|
||
|
|
__date__ = 'January 2019'
|
||
|
|
__copyright__ = '(C) 2019, Nyall Dawson'
|
||
|
|
|
||
|
|
# This will get replaced with a git SHA1 when you do a git archive
|
||
|
|
|
||
|
|
__revision__ = ':%H$'
|
||
|
|
|
||
|
|
import nose2
|
||
|
|
import shutil
|
||
|
|
import gc
|
||
|
|
|
||
|
|
from qgis.core import (QgsApplication,
|
||
|
|
QgsProcessing,
|
||
|
|
QgsProcessingContext,
|
||
|
|
QgsVectorLayer)
|
||
|
|
from qgis.PyQt import sip
|
||
|
|
from qgis.analysis import (QgsNativeAlgorithms)
|
||
|
|
from qgis.testing import start_app, unittest
|
||
|
|
import processing
|
||
|
|
from processing.tests.TestData import points
|
||
|
|
|
||
|
|
|
||
|
|
class TestProcessingGeneral(unittest.TestCase):
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def setUpClass(cls):
|
||
|
|
start_app()
|
||
|
|
from processing.core.Processing import Processing
|
||
|
|
Processing.initialize()
|
||
|
|
QgsApplication.processingRegistry().addProvider(QgsNativeAlgorithms())
|
||
|
|
cls.cleanup_paths = []
|
||
|
|
cls.in_place_layers = {}
|
||
|
|
cls.vector_layer_params = {}
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def tearDownClass(cls):
|
||
|
|
from processing.core.Processing import Processing
|
||
|
|
Processing.deinitialize()
|
||
|
|
for path in cls.cleanup_paths:
|
||
|
|
shutil.rmtree(path)
|
||
|
|
|
||
|
|
def testRun(self):
|
||
|
|
context = QgsProcessingContext()
|
||
|
|
|
||
|
|
# try running an alg using processing.run - ownership of result layer should be transferred back to the caller
|
||
|
|
res = processing.run('qgis:buffer',
|
||
|
|
{'DISTANCE': 1, 'INPUT': points(), 'OUTPUT': QgsProcessing.TEMPORARY_OUTPUT},
|
||
|
|
context=context)
|
||
|
|
self.assertIn('OUTPUT', res)
|
||
|
|
# output should be the layer instance itself
|
||
|
|
self.assertIsInstance(res['OUTPUT'], QgsVectorLayer)
|
||
|
|
# Python should have ownership
|
||
|
|
self.assertTrue(sip.ispyowned(res['OUTPUT']))
|
||
|
|
del context
|
||
|
|
gc.collect()
|
||
|
|
self.assertFalse(sip.isdeleted(res['OUTPUT']))
|
||
|
|
|
||
|
|
# now try using processing.run with is_child_algorithm = True. Ownership should remain with the context
|
||
|
|
context = QgsProcessingContext()
|
||
|
|
res = processing.run('qgis:buffer',
|
||
|
|
{'DISTANCE': 1, 'INPUT': points(), 'OUTPUT': QgsProcessing.TEMPORARY_OUTPUT},
|
||
|
|
context=context, is_child_algorithm=True)
|
||
|
|
self.assertIn('OUTPUT', res)
|
||
|
|
# output should be a layer string reference, NOT the layer itself
|
||
|
|
self.assertIsInstance(res['OUTPUT'], str)
|
||
|
|
layer = context.temporaryLayerStore().mapLayer(res['OUTPUT'])
|
||
|
|
self.assertIsInstance(layer, QgsVectorLayer)
|
||
|
|
# context should have ownership
|
||
|
|
self.assertFalse(sip.ispyowned(layer))
|
||
|
|
del context
|
||
|
|
gc.collect()
|
||
|
|
self.assertTrue(sip.isdeleted(layer))
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
nose2.main()
|