Ticket #12312: 12312_ogrgeometry_transform_wkt_3d_v2.diff

django/contrib/gis/tests/geo3d/tests.py

@@ -2 +2 @@
 from django.contrib.gis.db.models import Union, Extent3D
 from django.contrib.gis.geos import GEOSGeometry, Point, Polygon
 from django.contrib.gis.utils import LayerMapping, LayerMapError
+from django.contrib.gis.gdal import OGRGeometry
 
 from models import City3D, Interstate2D, Interstate3D, \
     InterstateProj2D, InterstateProj3D, \
@@ -228 +229 @@
             for city in City3D.objects.translate(0, 0, ztrans):
                 self.assertEqual(city_dict[city.name][2] + ztrans, city.translate.z)
 
+    def test08_ogrgeometry_transform_wkt(self):
+        "Testing dimensions on WKT output of geometries transformed with OGRGeometry.transform"
+        wkt_2d = "MULTILINESTRING ((0 0,1 1,2 2))"
+        wkt_3d = "MULTILINESTRING ((0 0 0,1 1 1,2 2 2))"
+        srid = 4326
+
+        geom = OGRGeometry(wkt_2d, srid)
+        geom.transform(srid)
+        self.assertEqual(geom.coord_dim, 2)
+        self.assertEqual(geom.wkt, wkt_2d)
+
+        geom = OGRGeometry(wkt_3d, srid)
+        geom.transform(srid)
+        self.assertEaual(geom.coord_dim, 3)
+        self.assertEaual(geom.wkt, wkt_3d)
+
+
+
 def suite():
     s = unittest.TestSuite()
     s.addTest(unittest.makeSuite(Geo3DTest))