diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/gis_tests/gdal_tests/test_geom.py | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/tests/gis_tests/gdal_tests/test_geom.py b/tests/gis_tests/gdal_tests/test_geom.py index 372a385443..eba90504b2 100644 --- a/tests/gis_tests/gdal_tests/test_geom.py +++ b/tests/gis_tests/gdal_tests/test_geom.py @@ -714,3 +714,16 @@ class OGRGeomTest(SimpleTestCase, TestDataMixin): msg = f"Unsupported geometry type: {type_}" with self.assertRaisesMessage(TypeError, msg): OGRGeometry(f"{geom_type} EMPTY") + + def test_is_3d_and_set_3d(self): + geom = OGRGeometry("POINT (1 2)") + self.assertIs(geom.is_3d, False) + geom.set_3d(True) + self.assertIs(geom.is_3d, True) + self.assertEqual(geom.wkt, "POINT (1 2 0)") + geom.set_3d(False) + self.assertIs(geom.is_3d, False) + self.assertEqual(geom.wkt, "POINT (1 2)") + msg = "Input to 'set_3d' must be a boolean, got 'None'" + with self.assertRaisesMessage(ValueError, msg): + geom.set_3d(None) |