mirror of
https://github.com/qgis/QGIS.git
synced 2025-02-27 00:33:48 -05:00
97 lines
4.0 KiB
Python
97 lines
4.0 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
***************************************************************************
|
|
ModelerTest
|
|
---------------------
|
|
Date : November 2016
|
|
Copyright : (C) 2016 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. *
|
|
* *
|
|
***************************************************************************8
|
|
"""
|
|
|
|
__author__ = 'Nyall Dawson'
|
|
__date__ = 'November 2016'
|
|
__copyright__ = '(C) 2016, Nyall Dawson'
|
|
|
|
# This will get replaced with a git SHA1 when you do a git archive
|
|
|
|
__revision__ = '$Format:%H$'
|
|
|
|
from qgis.testing import start_app, unittest
|
|
|
|
from processing.modeler.ModelerAlgorithm import (Algorithm,
|
|
ModelerAlgorithm,
|
|
ModelerParameter,
|
|
ModelerOutput,
|
|
ValueFromOutput)
|
|
from processing.modeler.ModelerParametersDialog import (ModelerParametersDialog)
|
|
from processing.core.parameters import (ParameterFile,
|
|
ParameterNumber,
|
|
ParameterString,
|
|
ParameterTableField)
|
|
start_app()
|
|
|
|
|
|
class ModelerTest(unittest.TestCase):
|
|
|
|
def testModelerParametersDialogAvailableValuesOfType(self):
|
|
# test getAvailableValuesOfType from ModelerParametersDialog
|
|
|
|
m = ModelerAlgorithm()
|
|
string_param_1 = ModelerParameter(ParameterString('string', 'string desc'))
|
|
m.addParameter(string_param_1)
|
|
string_param_2 = ModelerParameter(ParameterString('string2', 'string desc'))
|
|
m.addParameter(string_param_2)
|
|
num_param = ModelerParameter(ParameterNumber('number', 'number desc'))
|
|
m.addParameter(num_param)
|
|
table_field_param = ModelerParameter(ParameterTableField('field', 'field desc'))
|
|
m.addParameter(table_field_param)
|
|
file_param = ModelerParameter(ParameterFile('file', 'file desc'))
|
|
m.addParameter(file_param)
|
|
|
|
dlg = ModelerParametersDialog(m, m)
|
|
# test single types
|
|
self.assertEqual(set(p.name for p in dlg.getAvailableValuesOfType(ParameterNumber)),
|
|
set(['number']))
|
|
self.assertEqual(set(p.name for p in dlg.getAvailableValuesOfType(ParameterTableField)),
|
|
set(['field']))
|
|
self.assertEqual(set(p.name for p in dlg.getAvailableValuesOfType(ParameterFile)),
|
|
set(['file']))
|
|
|
|
# test multiple types
|
|
self.assertEqual(set(p.name for p in dlg.getAvailableValuesOfType([ParameterString, ParameterNumber, ParameterFile])),
|
|
set(['string', 'string2', 'number', 'file']))
|
|
|
|
def testModelerAlgorithmHasDependencies(self):
|
|
# test hasDependencies from ModelerAlgorithm
|
|
|
|
m = ModelerAlgorithm()
|
|
|
|
a = Algorithm("qgis:clip")
|
|
m.addAlgorithm(a)
|
|
a2 = Algorithm("qgis:clip")
|
|
m.addAlgorithm(a2)
|
|
|
|
# test parent algorithm dependency
|
|
self.assertEqual(m.hasDependencies('QGISCLIP_1'), False)
|
|
a2.dependencies = ['QGISCLIP_1']
|
|
self.assertEqual(m.hasDependencies('QGISCLIP_1'), True)
|
|
|
|
# test output algorithm dependency
|
|
a2.dependencies = []
|
|
a.outputs['OUTPUT'] = ModelerOutput('out')
|
|
a2.params['INPUT'] = ValueFromOutput('QGISCLIP_1', 'OUTPUT')
|
|
self.assertEqual(m.hasDependencies('QGISCLIP_1'), True)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|