sharpetronics/QGIS
1
0
Fork 0
mirror of https://github.com/qgis/QGIS.git synced 2025-04-13 00:03:09 -04:00
Code Issues Packages Projects Releases Wiki Activity

Allow multiple primary key fields to be specified for processing tests

Browse Source 
`
This commit is contained in:
Nyall Dawson 2017-09-08 16:33:55 +10:00
parent c750cb2154
commit e9fd409552
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
python/testing/__init__.py
View File

@ -105,7 +105,11 @@ class TestCase(_TestCase):
def sort_by_pk_or_fid(f):
if 'pk' in kwargs and kwargs['pk'] is not None:
return f[kwargs['pk']]
key = kwargs['pk']
if isinstance(key, list) or isinstance(key, tuple):
return [f[k] for k in key]
else:
return f[kwargs['pk']]
else:
return f.id()