-
-
Notifications
You must be signed in to change notification settings - Fork 3k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[processing] Allow modification of feature request when using
vector.features Allows for optimising the request through subsets of attributes or no geometry fetching
- Loading branch information
1 parent
f34c79c
commit 9e1ddcb
Showing
3 changed files
with
101 additions
and
10 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,89 @@ | ||
# -*- coding: utf-8 -*- | ||
|
||
""" | ||
*************************************************************************** | ||
ToolsTest | ||
--------------------- | ||
Date : July 2017 | ||
This comment has been minimized.
Sorry, something went wrong.
This comment has been minimized.
Sorry, something went wrong.
sebastic
Contributor
|
||
Copyright : (C) 2017 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__ = 'July 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.tests.TestData import points2 | ||
from processing.tools import vector | ||
from qgis.core import (QgsVectorLayer, QgsFeatureRequest) | ||
from processing.core.ProcessingConfig import ProcessingConfig | ||
|
||
start_app() | ||
|
||
|
||
class VectorTest(unittest.TestCase): | ||
|
||
def testFeatures(self): | ||
ProcessingConfig.initialize() | ||
|
||
test_data = points2() | ||
test_layer = QgsVectorLayer(test_data, 'test', 'ogr') | ||
|
||
# test with all features | ||
features = vector.features(test_layer) | ||
self.assertEqual(len(features), 8) | ||
self.assertEqual(set([f.id() for f in features]), set([0, 1, 2, 3, 4, 5, 6, 7])) | ||
|
||
# test with selected features | ||
previous_value = ProcessingConfig.getSetting(ProcessingConfig.USE_SELECTED) | ||
ProcessingConfig.setSettingValue(ProcessingConfig.USE_SELECTED, True) | ||
test_layer.selectByIds([2, 4, 6]) | ||
features = vector.features(test_layer) | ||
self.assertEqual(len(features), 3) | ||
self.assertEqual(set([f.id() for f in features]), set([2, 4, 6])) | ||
|
||
# selection, but not using selected features | ||
ProcessingConfig.setSettingValue(ProcessingConfig.USE_SELECTED, False) | ||
test_layer.selectByIds([2, 4, 6]) | ||
features = vector.features(test_layer) | ||
self.assertEqual(len(features), 8) | ||
self.assertEqual(set([f.id() for f in features]), set([0, 1, 2, 3, 4, 5, 6, 7])) | ||
|
||
# using selected features, but no selection | ||
ProcessingConfig.setSettingValue(ProcessingConfig.USE_SELECTED, True) | ||
test_layer.removeSelection() | ||
features = vector.features(test_layer) | ||
self.assertEqual(len(features), 8) | ||
self.assertEqual(set([f.id() for f in features]), set([0, 1, 2, 3, 4, 5, 6, 7])) | ||
|
||
# test that feature request is honored | ||
ProcessingConfig.setSettingValue(ProcessingConfig.USE_SELECTED, False) | ||
features = vector.features(test_layer, QgsFeatureRequest().setFilterFids([1, 3, 5])) | ||
self.assertEqual(set([f.id() for f in features]), set([1, 3, 5])) | ||
|
||
# test that feature request is honored when using selections | ||
ProcessingConfig.setSettingValue(ProcessingConfig.USE_SELECTED, True) | ||
test_layer.selectByIds([2, 4, 6]) | ||
features = vector.features(test_layer, QgsFeatureRequest().setFlags(QgsFeatureRequest.NoGeometry)) | ||
self.assertTrue(all([f.geometry() == None for f in features])) | ||
features = vector.features(test_layer, QgsFeatureRequest().setFlags(QgsFeatureRequest.NoGeometry)) | ||
self.assertEqual(set([f.id() for f in features]), set([2, 4, 6])) | ||
|
||
ProcessingConfig.setSettingValue(ProcessingConfig.USE_SELECTED, previous_value) | ||
|
||
|
||
if __name__ == '__main__': | ||
unittest.main() |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Timewarp!