mirror of
https://github.com/qgis/QGIS.git
synced 2025-02-27 00:33:48 -05:00
81 lines
3.5 KiB
Python
81 lines
3.5 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 qgis.core import (QgsProcessingModelAlgorithm,
|
|
QgsProcessingParameterString,
|
|
QgsProcessingParameterNumber,
|
|
QgsProcessingParameterField,
|
|
QgsProcessingParameterFile)
|
|
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 = QgsProcessingModelAlgorithm()
|
|
|
|
string_param_1 = QgsProcessingModelAlgorithm.ModelParameter('string')
|
|
m.addModelParameter(QgsProcessingParameterString('string'), string_param_1)
|
|
|
|
string_param_2 = QgsProcessingModelAlgorithm.ModelParameter('string2')
|
|
m.addModelParameter(QgsProcessingParameterString('string2'), string_param_2)
|
|
|
|
num_param = QgsProcessingModelAlgorithm.ModelParameter('number')
|
|
m.addModelParameter(QgsProcessingParameterNumber('number'), num_param)
|
|
|
|
table_field_param = QgsProcessingModelAlgorithm.ModelParameter('field')
|
|
m.addModelParameter(QgsProcessingParameterField('field'), table_field_param)
|
|
|
|
file_param = QgsProcessingModelAlgorithm.ModelParameter('file')
|
|
m.addModelParameter(QgsProcessingParameterFile('file'), file_param)
|
|
|
|
dlg = ModelerParametersDialog(m, m)
|
|
# test single types
|
|
self.assertEqual(set(p.parameterName() for p in dlg.getAvailableValuesOfType(QgsProcessingParameterNumber)),
|
|
set(['number']))
|
|
self.assertEqual(set(p.parameterName() for p in dlg.getAvailableValuesOfType(QgsProcessingParameterField)),
|
|
set(['field']))
|
|
self.assertEqual(set(p.parameterName() for p in dlg.getAvailableValuesOfType(QgsProcessingParameterFile)),
|
|
set(['file']))
|
|
|
|
# test multiple types
|
|
self.assertEqual(set(p.parameterName() for p in dlg.getAvailableValuesOfType([QgsProcessingParameterString, QgsProcessingParameterNumber, QgsProcessingParameterFile])),
|
|
set(['string', 'string2', 'number', 'file']))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|