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

Followup fb7c683, 4e4743

Browse Source
This commit is contained in:
Nyall Dawson 2016-02-26 19:05:46 +11:00
parent fb7c683969
commit ae347bfd89
1 changed files with 29 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
tests/src/python/test_qgsvectorfilewriter.py
View File

@ -25,6 +25,7 @@ from qgis.core import (QgsVectorLayer,
) )
from PyQt4.QtCore import QDate, QTime, QDateTime, QVariant, QDir from PyQt4.QtCore import QDate, QTime, QDateTime, QVariant, QDir
import os import os
import platform
from qgis.testing import ( from qgis.testing import (
start_app, start_app,
unittest unittest
@ -171,10 +172,14 @@ class TestQgsVectorLayer(unittest.TestCase):
assert isinstance(f.attributes()[datetime_idx], QDateTime) assert isinstance(f.attributes()[datetime_idx], QDateTime)
self.assertEqual(f.attributes()[datetime_idx], QDateTime(QDate(2014, 3, 5), QTime(13, 45, 22))) self.assertEqual(f.attributes()[datetime_idx], QDateTime(QDate(2014, 3, 5), QTime(13, 45, 22)))
# Fails on Travis Linux build for some reason
@unittest.expectedFailure
def testWriteShapefileWithZ(self): def testWriteShapefileWithZ(self):
"""Check writing geometries with Z dimension to an ESRI shapefile.""" """Check writing geometries with Z dimension to an ESRI shapefile."""
if os.environ.get('TRAVIS') and 'linux' in platform.system().lower():
# This test fails on Travis Linux build for unknown reason (probably GDAL version related)
return
#start by saving a memory layer and forcing z
ml = QgsVectorLayer( ml = QgsVectorLayer(
('Point?crs=epsg:4326&field=id:int'), ('Point?crs=epsg:4326&field=id:int'),
'test', 'test',
@ -215,6 +220,28 @@ class TestQgsVectorLayer(unittest.TestCase):
expWkt = 'PointZ (1 2 3)' expWkt = 'PointZ (1 2 3)'
assert compareWkt(expWkt, wkt), "saving geometry with Z failed: mismatch Expected:\n%s\nGot:\n%s\n" % (expWkt, wkt) assert compareWkt(expWkt, wkt), "saving geometry with Z failed: mismatch Expected:\n%s\nGot:\n%s\n" % (expWkt, wkt)
#also try saving out the shapefile version again, as an extra test
#this tests that saving a layer with z WITHOUT explicitly telling the writer to keep z values,
#will stay retain the z values
dest_file_name = os.path.join(str(QDir.tempPath()), 'point_{}_copy.shp'.format(QgsWKBTypes.displayString(t)))
print(dest_file_name)
crs = QgsCoordinateReferenceSystem()
crs.createFromId(4326, QgsCoordinateReferenceSystem.EpsgCrsId)
write_result = QgsVectorFileWriter.writeAsVectorFormat(
created_layer,
dest_file_name,
'utf-8',
crs,
'ESRI Shapefile')
self.assertEqual(write_result, QgsVectorFileWriter.NoError)
# Open result and check
created_layer_from_shp = QgsVectorLayer(u'{}|layerid=0'.format(dest_file_name), u'test', u'ogr')
f = created_layer_from_shp.getFeatures(QgsFeatureRequest()).next()
g = f.geometry()
wkt = g.exportToWkt()
assert compareWkt(expWkt, wkt), "saving geometry with Z failed: mismatch Expected:\n%s\nGot:\n%s\n" % (expWkt, wkt)
def testWriteShapefileWithMultiConversion(self): def testWriteShapefileWithMultiConversion(self):
"""Check writing geometries to an ESRI shapefile with conversion to multi.""" """Check writing geometries to an ESRI shapefile with conversion to multi."""
ml = QgsVectorLayer( ml = QgsVectorLayer(