mirror of
https://github.com/qgis/QGIS.git
synced 2025-04-15 00:04:00 -04:00
83115cd16a
Conflicts: python/core/qgsconditionalstyle.sip python/core/qgsfielduiproperties.sip src/core/qgsfielduiproperties.cpp src/core/qgsfielduiproperties.h src/gui/attributetable/qgsattributetablemodel.cpp src/ui/qgsfieldconditionalformatwidget.ui tests/src/python/test_qgsconditionalstyle.py
75 lines
2.7 KiB
Python
75 lines
2.7 KiB
Python
|
|
# -*- coding: utf-8 -*-
|
|
"""QGIS Unit tests for the memory layer provider.
|
|
|
|
.. note:: 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__ = 'Nathan.Woodrow'
|
|
__date__ = '2015-08-11'
|
|
__copyright__ = 'Copyright 2015, The QGIS Project'
|
|
# This will get replaced with a git SHA1 when you do a git archive
|
|
__revision__ = '$Format:%H$'
|
|
|
|
from qgis.core import QgsConditionalStyle, QgsFeature, QgsFields, QgsField, QgsExpressionContextUtils
|
|
from utilities import (unitTestDataPath,
|
|
getQgisTestApp,
|
|
unittest,
|
|
TestCase,
|
|
compareWkt)
|
|
from PyQt4.QtCore import QVariant
|
|
#
|
|
QGISAPP, CANVAS, IFACE, PARENT = getQgisTestApp()
|
|
TEST_DATA_DIR = unitTestDataPath()
|
|
|
|
|
|
class TestPyQgsConditionalStyle(TestCase):
|
|
def new_feature(self):
|
|
feature = QgsFeature()
|
|
fields = QgsFields()
|
|
fields.append(QgsField("testfield", QVariant.Int))
|
|
feature.setFields(fields, True)
|
|
feature["testfield"] = 20
|
|
return feature
|
|
|
|
def test_MatchesReturnsTrueForSimpleMatch(self):
|
|
style = QgsConditionalStyle("@value > 10")
|
|
context = QgsExpressionContextUtils.createFeatureBasedContext(QgsFeature(), QgsFields())
|
|
assert style.matches(20, context)
|
|
|
|
def test_MatchesReturnsTrueForComplexMatch(self):
|
|
style = QgsConditionalStyle("@value > 10 and @value = 20")
|
|
context = QgsExpressionContextUtils.createFeatureBasedContext(QgsFeature(), QgsFields())
|
|
assert style.matches(20, context)
|
|
|
|
def test_MatchesTrueForFields(self):
|
|
feature = QgsFeature()
|
|
fields = QgsFields()
|
|
fields.append(QgsField("testfield", QVariant.Int))
|
|
feature.setFields(fields, True)
|
|
feature["testfield"] = 20
|
|
style = QgsConditionalStyle('"testfield" = @value')
|
|
context = QgsExpressionContextUtils.createFeatureBasedContext(feature, fields)
|
|
assert style.matches(20, context)
|
|
|
|
def test_MatchingStylesReturnsListOfCorrectStyles(self):
|
|
styles = []
|
|
style = QgsConditionalStyle("@value > 10")
|
|
style.setName("1")
|
|
styles.append(style)
|
|
style = QgsConditionalStyle("@value > 10")
|
|
style.setName("2")
|
|
styles.append(style)
|
|
style = QgsConditionalStyle("@value < 5")
|
|
style.setName("3")
|
|
styles.append(style)
|
|
out = QgsConditionalStyle.matchingConditionalStyles(styles, 20, self.new_feature())
|
|
assert len(out) == 2
|
|
out[0].name() == "1"
|
|
out[1].name() == "2"
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|