12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073 |
- import json
- import pickle
- from unittest import mock, skipIf
- from django.contrib.gis.gdal import (
- CoordTransform,
- GDALException,
- OGRGeometry,
- OGRGeomType,
- SpatialReference,
- )
- from django.contrib.gis.gdal.geometries import CircularString, CurvePolygon
- from django.contrib.gis.geos import GEOSException
- from django.contrib.gis.geos.libgeos import geos_version_tuple
- from django.template import Context
- from django.template.engine import Engine
- from django.test import SimpleTestCase
- from ..test_data import TestDataMixin
- class OGRGeomTest(SimpleTestCase, TestDataMixin):
- "This tests the OGR Geometry."
- def test_geomtype(self):
- "Testing OGRGeomType object."
- # OGRGeomType should initialize on all these inputs.
- OGRGeomType(1)
- OGRGeomType(7)
- OGRGeomType("point")
- OGRGeomType("GeometrycollectioN")
- OGRGeomType("LINearrING")
- OGRGeomType("Unknown")
- # Should throw TypeError on this input
- with self.assertRaises(GDALException):
- OGRGeomType(23)
- with self.assertRaises(GDALException):
- OGRGeomType("fooD")
- with self.assertRaises(GDALException):
- OGRGeomType(4001)
- # Equivalence can take strings, ints, and other OGRGeomTypes
- self.assertEqual(OGRGeomType(1), OGRGeomType(1))
- self.assertEqual(OGRGeomType(7), "GeometryCollection")
- self.assertEqual(OGRGeomType("point"), "POINT")
- self.assertNotEqual(OGRGeomType("point"), 2)
- self.assertEqual(OGRGeomType("unknown"), 0)
- self.assertEqual(OGRGeomType(6), "MULtiPolyGON")
- self.assertEqual(OGRGeomType(1), OGRGeomType("point"))
- self.assertNotEqual(OGRGeomType("POINT"), OGRGeomType(6))
- # Testing the Django field name equivalent property.
- self.assertEqual("PointField", OGRGeomType("Point").django)
- self.assertEqual("GeometryField", OGRGeomType("Geometry").django)
- self.assertEqual("GeometryField", OGRGeomType("Unknown").django)
- self.assertIsNone(OGRGeomType("none").django)
- # 'Geometry' initialization implies an unknown geometry type.
- gt = OGRGeomType("Geometry")
- self.assertEqual(0, gt.num)
- self.assertEqual("Unknown", gt.name)
- def test_geom_type_repr(self):
- self.assertEqual(repr(OGRGeomType("point")), "<OGRGeomType: Point>")
- def test_geomtype_25d(self):
- "Testing OGRGeomType object with 25D types."
- wkb25bit = OGRGeomType.wkb25bit
- self.assertEqual(OGRGeomType(wkb25bit + 1), "Point25D")
- self.assertEqual(OGRGeomType("MultiLineString25D"), (5 + wkb25bit))
- self.assertEqual(
- "GeometryCollectionField", OGRGeomType("GeometryCollection25D").django
- )
- def test_wkt(self):
- "Testing WKT output."
- for g in self.geometries.wkt_out:
- geom = OGRGeometry(g.wkt)
- self.assertEqual(g.wkt, geom.wkt)
- def test_ewkt(self):
- "Testing EWKT input/output."
- for ewkt_val in ("POINT (1 2 3)", "LINEARRING (0 0,1 1,2 1,0 0)"):
- # First with ewkt output when no SRID in EWKT
- self.assertEqual(ewkt_val, OGRGeometry(ewkt_val).ewkt)
- # No test consumption with an SRID specified.
- ewkt_val = "SRID=4326;%s" % ewkt_val
- geom = OGRGeometry(ewkt_val)
- self.assertEqual(ewkt_val, geom.ewkt)
- self.assertEqual(4326, geom.srs.srid)
- def test_gml(self):
- "Testing GML output."
- for g in self.geometries.wkt_out:
- geom = OGRGeometry(g.wkt)
- exp_gml = g.gml
- self.assertEqual(exp_gml, geom.gml)
- def test_hex(self):
- "Testing HEX input/output."
- for g in self.geometries.hex_wkt:
- geom1 = OGRGeometry(g.wkt)
- self.assertEqual(g.hex.encode(), geom1.hex)
- # Constructing w/HEX
- geom2 = OGRGeometry(g.hex)
- self.assertEqual(geom1, geom2)
- def test_wkb(self):
- "Testing WKB input/output."
- for g in self.geometries.hex_wkt:
- geom1 = OGRGeometry(g.wkt)
- wkb = geom1.wkb
- self.assertEqual(wkb.hex().upper(), g.hex)
- # Constructing w/WKB.
- geom2 = OGRGeometry(wkb)
- self.assertEqual(geom1, geom2)
- def test_json(self):
- "Testing GeoJSON input/output."
- for g in self.geometries.json_geoms:
- geom = OGRGeometry(g.wkt)
- if not hasattr(g, "not_equal"):
- # Loading jsons to prevent decimal differences
- self.assertEqual(json.loads(g.json), json.loads(geom.json))
- self.assertEqual(json.loads(g.json), json.loads(geom.geojson))
- self.assertEqual(OGRGeometry(g.wkt), OGRGeometry(geom.json))
- # Test input with some garbage content (but valid json) (#15529)
- geom = OGRGeometry(
- '{"type": "Point", "coordinates": [ 100.0, 0.0 ], "other": "<test>"}'
- )
- self.assertIsInstance(geom, OGRGeometry)
- def test_points(self):
- "Testing Point objects."
- OGRGeometry("POINT(0 0)")
- for p in self.geometries.points:
- if not hasattr(p, "z"): # No 3D
- pnt = OGRGeometry(p.wkt)
- self.assertEqual(1, pnt.geom_type)
- self.assertEqual("POINT", pnt.geom_name)
- self.assertEqual(p.x, pnt.x)
- self.assertEqual(p.y, pnt.y)
- self.assertEqual((p.x, p.y), pnt.tuple)
- def test_multipoints(self):
- "Testing MultiPoint objects."
- for mp in self.geometries.multipoints:
- mgeom1 = OGRGeometry(mp.wkt) # First one from WKT
- self.assertEqual(4, mgeom1.geom_type)
- self.assertEqual("MULTIPOINT", mgeom1.geom_name)
- mgeom2 = OGRGeometry("MULTIPOINT") # Creating empty multipoint
- mgeom3 = OGRGeometry("MULTIPOINT")
- for g in mgeom1:
- mgeom2.add(g) # adding each point from the multipoints
- mgeom3.add(g.wkt) # should take WKT as well
- self.assertEqual(mgeom1, mgeom2) # they should equal
- self.assertEqual(mgeom1, mgeom3)
- self.assertEqual(mp.coords, mgeom2.coords)
- self.assertEqual(mp.n_p, mgeom2.point_count)
- def test_linestring(self):
- "Testing LineString objects."
- prev = OGRGeometry("POINT(0 0)")
- for ls in self.geometries.linestrings:
- linestr = OGRGeometry(ls.wkt)
- self.assertEqual(2, linestr.geom_type)
- self.assertEqual("LINESTRING", linestr.geom_name)
- self.assertEqual(ls.n_p, linestr.point_count)
- self.assertEqual(ls.coords, linestr.tuple)
- self.assertEqual(linestr, OGRGeometry(ls.wkt))
- self.assertNotEqual(linestr, prev)
- msg = "Index out of range when accessing points of a line string: %s."
- with self.assertRaisesMessage(IndexError, msg % len(linestr)):
- linestr.__getitem__(len(linestr))
- prev = linestr
- # Testing the x, y properties.
- x = [tmpx for tmpx, tmpy in ls.coords]
- y = [tmpy for tmpx, tmpy in ls.coords]
- self.assertEqual(x, linestr.x)
- self.assertEqual(y, linestr.y)
- def test_multilinestring(self):
- "Testing MultiLineString objects."
- prev = OGRGeometry("POINT(0 0)")
- for mls in self.geometries.multilinestrings:
- mlinestr = OGRGeometry(mls.wkt)
- self.assertEqual(5, mlinestr.geom_type)
- self.assertEqual("MULTILINESTRING", mlinestr.geom_name)
- self.assertEqual(mls.n_p, mlinestr.point_count)
- self.assertEqual(mls.coords, mlinestr.tuple)
- self.assertEqual(mlinestr, OGRGeometry(mls.wkt))
- self.assertNotEqual(mlinestr, prev)
- prev = mlinestr
- for ls in mlinestr:
- self.assertEqual(2, ls.geom_type)
- self.assertEqual("LINESTRING", ls.geom_name)
- msg = "Index out of range when accessing geometry in a collection: %s."
- with self.assertRaisesMessage(IndexError, msg % len(mlinestr)):
- mlinestr.__getitem__(len(mlinestr))
- def test_linearring(self):
- "Testing LinearRing objects."
- prev = OGRGeometry("POINT(0 0)")
- for rr in self.geometries.linearrings:
- lr = OGRGeometry(rr.wkt)
- # self.assertEqual(101, lr.geom_type.num)
- self.assertEqual("LINEARRING", lr.geom_name)
- self.assertEqual(rr.n_p, len(lr))
- self.assertEqual(lr, OGRGeometry(rr.wkt))
- self.assertNotEqual(lr, prev)
- prev = lr
- def test_polygons(self):
- "Testing Polygon objects."
- # Testing `from_bbox` class method
- bbox = (-180, -90, 180, 90)
- p = OGRGeometry.from_bbox(bbox)
- self.assertEqual(bbox, p.extent)
- prev = OGRGeometry("POINT(0 0)")
- for p in self.geometries.polygons:
- poly = OGRGeometry(p.wkt)
- self.assertEqual(3, poly.geom_type)
- self.assertEqual("POLYGON", poly.geom_name)
- self.assertEqual(p.n_p, poly.point_count)
- self.assertEqual(p.n_i + 1, len(poly))
- msg = "Index out of range when accessing rings of a polygon: %s."
- with self.assertRaisesMessage(IndexError, msg % len(poly)):
- poly.__getitem__(len(poly))
- # Testing area & centroid.
- self.assertAlmostEqual(p.area, poly.area, 9)
- x, y = poly.centroid.tuple
- self.assertAlmostEqual(p.centroid[0], x, 9)
- self.assertAlmostEqual(p.centroid[1], y, 9)
- # Testing equivalence
- self.assertEqual(poly, OGRGeometry(p.wkt))
- self.assertNotEqual(poly, prev)
- if p.ext_ring_cs:
- ring = poly[0]
- self.assertEqual(p.ext_ring_cs, ring.tuple)
- self.assertEqual(p.ext_ring_cs, poly[0].tuple)
- self.assertEqual(len(p.ext_ring_cs), ring.point_count)
- for r in poly:
- self.assertEqual("LINEARRING", r.geom_name)
- def test_polygons_templates(self):
- # Accessing Polygon attributes in templates should work.
- engine = Engine()
- template = engine.from_string("{{ polygons.0.wkt }}")
- polygons = [OGRGeometry(p.wkt) for p in self.geometries.multipolygons[:2]]
- content = template.render(Context({"polygons": polygons}))
- self.assertIn("MULTIPOLYGON (((100", content)
- def test_closepolygons(self):
- "Testing closing Polygon objects."
- # Both rings in this geometry are not closed.
- poly = OGRGeometry("POLYGON((0 0, 5 0, 5 5, 0 5), (1 1, 2 1, 2 2, 2 1))")
- self.assertEqual(8, poly.point_count)
- with self.assertRaises(GDALException):
- poly.centroid
- poly.close_rings()
- self.assertEqual(
- 10, poly.point_count
- ) # Two closing points should've been added
- self.assertEqual(OGRGeometry("POINT(2.5 2.5)"), poly.centroid)
- def test_multipolygons(self):
- "Testing MultiPolygon objects."
- OGRGeometry("POINT(0 0)")
- for mp in self.geometries.multipolygons:
- mpoly = OGRGeometry(mp.wkt)
- self.assertEqual(6, mpoly.geom_type)
- self.assertEqual("MULTIPOLYGON", mpoly.geom_name)
- if mp.valid:
- self.assertEqual(mp.n_p, mpoly.point_count)
- self.assertEqual(mp.num_geom, len(mpoly))
- msg = "Index out of range when accessing geometry in a collection: %s."
- with self.assertRaisesMessage(IndexError, msg % len(mpoly)):
- mpoly.__getitem__(len(mpoly))
- for p in mpoly:
- self.assertEqual("POLYGON", p.geom_name)
- self.assertEqual(3, p.geom_type)
- self.assertEqual(mpoly.wkt, OGRGeometry(mp.wkt).wkt)
- def test_srs(self):
- "Testing OGR Geometries with Spatial Reference objects."
- for mp in self.geometries.multipolygons:
- # Creating a geometry w/spatial reference
- sr = SpatialReference("WGS84")
- mpoly = OGRGeometry(mp.wkt, sr)
- self.assertEqual(sr.wkt, mpoly.srs.wkt)
- # Ensuring that SRS is propagated to clones.
- klone = mpoly.clone()
- self.assertEqual(sr.wkt, klone.srs.wkt)
- # Ensuring all children geometries (polygons and their rings) all
- # return the assigned spatial reference as well.
- for poly in mpoly:
- self.assertEqual(sr.wkt, poly.srs.wkt)
- for ring in poly:
- self.assertEqual(sr.wkt, ring.srs.wkt)
- # Ensuring SRS propagate in topological ops.
- a = OGRGeometry(self.geometries.topology_geoms[0].wkt_a, sr)
- b = OGRGeometry(self.geometries.topology_geoms[0].wkt_b, sr)
- diff = a.difference(b)
- union = a.union(b)
- self.assertEqual(sr.wkt, diff.srs.wkt)
- self.assertEqual(sr.srid, union.srs.srid)
- # Instantiating w/an integer SRID
- mpoly = OGRGeometry(mp.wkt, 4326)
- self.assertEqual(4326, mpoly.srid)
- mpoly.srs = SpatialReference(4269)
- self.assertEqual(4269, mpoly.srid)
- self.assertEqual("NAD83", mpoly.srs.name)
- # Incrementing through the multipolygon after the spatial reference
- # has been re-assigned.
- for poly in mpoly:
- self.assertEqual(mpoly.srs.wkt, poly.srs.wkt)
- poly.srs = 32140
- for ring in poly:
- # Changing each ring in the polygon
- self.assertEqual(32140, ring.srs.srid)
- self.assertEqual("NAD83 / Texas South Central", ring.srs.name)
- ring.srs = str(SpatialReference(4326)) # back to WGS84
- self.assertEqual(4326, ring.srs.srid)
- # Using the `srid` property.
- ring.srid = 4322
- self.assertEqual("WGS 72", ring.srs.name)
- self.assertEqual(4322, ring.srid)
- # srs/srid may be assigned their own values, even when srs is None.
- mpoly = OGRGeometry(mp.wkt, srs=None)
- mpoly.srs = mpoly.srs
- mpoly.srid = mpoly.srid
- def test_srs_transform(self):
- "Testing transform()."
- orig = OGRGeometry("POINT (-104.609 38.255)", 4326)
- trans = OGRGeometry("POINT (992385.4472045 481455.4944650)", 2774)
- # Using an srid, a SpatialReference object, and a CoordTransform object
- # or transformations.
- t1, t2, t3 = orig.clone(), orig.clone(), orig.clone()
- t1.transform(trans.srid)
- t2.transform(SpatialReference("EPSG:2774"))
- ct = CoordTransform(SpatialReference("WGS84"), SpatialReference(2774))
- t3.transform(ct)
- # Testing use of the `clone` keyword.
- k1 = orig.clone()
- k2 = k1.transform(trans.srid, clone=True)
- self.assertEqual(k1, orig)
- self.assertNotEqual(k1, k2)
- # Different PROJ versions use different transformations, all are
- # correct as having a 1 meter accuracy.
- prec = -1
- for p in (t1, t2, t3, k2):
- self.assertAlmostEqual(trans.x, p.x, prec)
- self.assertAlmostEqual(trans.y, p.y, prec)
- def test_transform_dim(self):
- "Testing coordinate dimension is the same on transformed geometries."
- ls_orig = OGRGeometry("LINESTRING(-104.609 38.255)", 4326)
- ls_trans = OGRGeometry("LINESTRING(992385.4472045 481455.4944650)", 2774)
- # Different PROJ versions use different transformations, all are
- # correct as having a 1 meter accuracy.
- prec = -1
- ls_orig.transform(ls_trans.srs)
- # Making sure the coordinate dimension is still 2D.
- self.assertEqual(2, ls_orig.coord_dim)
- self.assertAlmostEqual(ls_trans.x[0], ls_orig.x[0], prec)
- self.assertAlmostEqual(ls_trans.y[0], ls_orig.y[0], prec)
- def test_difference(self):
- "Testing difference()."
- for i in range(len(self.geometries.topology_geoms)):
- a = OGRGeometry(self.geometries.topology_geoms[i].wkt_a)
- b = OGRGeometry(self.geometries.topology_geoms[i].wkt_b)
- d1 = OGRGeometry(self.geometries.diff_geoms[i].wkt)
- d2 = a.difference(b)
- self.assertTrue(d1.geos.equals(d2.geos))
- self.assertTrue(
- d1.geos.equals((a - b).geos)
- ) # __sub__ is difference operator
- a -= b # testing __isub__
- self.assertTrue(d1.geos.equals(a.geos))
- def test_intersection(self):
- "Testing intersects() and intersection()."
- for i in range(len(self.geometries.topology_geoms)):
- a = OGRGeometry(self.geometries.topology_geoms[i].wkt_a)
- b = OGRGeometry(self.geometries.topology_geoms[i].wkt_b)
- i1 = OGRGeometry(self.geometries.intersect_geoms[i].wkt)
- self.assertTrue(a.intersects(b))
- i2 = a.intersection(b)
- self.assertTrue(i1.geos.equals(i2.geos))
- self.assertTrue(
- i1.geos.equals((a & b).geos)
- ) # __and__ is intersection operator
- a &= b # testing __iand__
- self.assertTrue(i1.geos.equals(a.geos))
- def test_symdifference(self):
- "Testing sym_difference()."
- for i in range(len(self.geometries.topology_geoms)):
- a = OGRGeometry(self.geometries.topology_geoms[i].wkt_a)
- b = OGRGeometry(self.geometries.topology_geoms[i].wkt_b)
- d1 = OGRGeometry(self.geometries.sdiff_geoms[i].wkt)
- d2 = a.sym_difference(b)
- self.assertTrue(d1.geos.equals(d2.geos))
- self.assertTrue(
- d1.geos.equals((a ^ b).geos)
- ) # __xor__ is symmetric difference operator
- a ^= b # testing __ixor__
- self.assertTrue(d1.geos.equals(a.geos))
- def test_union(self):
- "Testing union()."
- for i in range(len(self.geometries.topology_geoms)):
- a = OGRGeometry(self.geometries.topology_geoms[i].wkt_a)
- b = OGRGeometry(self.geometries.topology_geoms[i].wkt_b)
- u1 = OGRGeometry(self.geometries.union_geoms[i].wkt)
- u2 = a.union(b)
- self.assertTrue(u1.geos.equals(u2.geos))
- self.assertTrue(u1.geos.equals((a | b).geos)) # __or__ is union operator
- a |= b # testing __ior__
- self.assertTrue(u1.geos.equals(a.geos))
- def test_add(self):
- "Testing GeometryCollection.add()."
- # Can't insert a Point into a MultiPolygon.
- mp = OGRGeometry("MultiPolygon")
- pnt = OGRGeometry("POINT(5 23)")
- with self.assertRaises(GDALException):
- mp.add(pnt)
- # GeometryCollection.add may take an OGRGeometry (if another collection
- # of the same type all child geoms will be added individually) or WKT.
- for mp in self.geometries.multipolygons:
- mpoly = OGRGeometry(mp.wkt)
- mp1 = OGRGeometry("MultiPolygon")
- mp2 = OGRGeometry("MultiPolygon")
- mp3 = OGRGeometry("MultiPolygon")
- for poly in mpoly:
- mp1.add(poly) # Adding a geometry at a time
- mp2.add(poly.wkt) # Adding WKT
- mp3.add(mpoly) # Adding a MultiPolygon's entire contents at once.
- for tmp in (mp1, mp2, mp3):
- self.assertEqual(mpoly, tmp)
- def test_extent(self):
- "Testing `extent` property."
- # The xmin, ymin, xmax, ymax of the MultiPoint should be returned.
- mp = OGRGeometry("MULTIPOINT(5 23, 0 0, 10 50)")
- self.assertEqual((0.0, 0.0, 10.0, 50.0), mp.extent)
- # Testing on the 'real world' Polygon.
- poly = OGRGeometry(self.geometries.polygons[3].wkt)
- ring = poly.shell
- x, y = ring.x, ring.y
- xmin, ymin = min(x), min(y)
- xmax, ymax = max(x), max(y)
- self.assertEqual((xmin, ymin, xmax, ymax), poly.extent)
- def test_25D(self):
- "Testing 2.5D geometries."
- pnt_25d = OGRGeometry("POINT(1 2 3)")
- self.assertEqual("Point25D", pnt_25d.geom_type.name)
- self.assertEqual(3.0, pnt_25d.z)
- self.assertEqual(3, pnt_25d.coord_dim)
- ls_25d = OGRGeometry("LINESTRING(1 1 1,2 2 2,3 3 3)")
- self.assertEqual("LineString25D", ls_25d.geom_type.name)
- self.assertEqual([1.0, 2.0, 3.0], ls_25d.z)
- self.assertEqual(3, ls_25d.coord_dim)
- def test_pickle(self):
- "Testing pickle support."
- g1 = OGRGeometry("LINESTRING(1 1 1,2 2 2,3 3 3)", "WGS84")
- g2 = pickle.loads(pickle.dumps(g1))
- self.assertEqual(g1, g2)
- self.assertEqual(4326, g2.srs.srid)
- self.assertEqual(g1.srs.wkt, g2.srs.wkt)
- def test_ogrgeometry_transform_workaround(self):
- "Testing coordinate dimensions on geometries after transformation."
- # A bug in GDAL versions prior to 1.7 changes the coordinate
- # dimension of a geometry after it has been transformed.
- # This test ensures that the bug workarounds employed within
- # `OGRGeometry.transform` indeed work.
- wkt_2d = "MULTILINESTRING ((0 0,1 1,2 2))"
- wkt_3d = "MULTILINESTRING ((0 0 0,1 1 1,2 2 2))"
- srid = 4326
- # For both the 2D and 3D MultiLineString, ensure _both_ the dimension
- # of the collection and the component LineString have the expected
- # coordinate dimension after transform.
- geom = OGRGeometry(wkt_2d, srid)
- geom.transform(srid)
- self.assertEqual(2, geom.coord_dim)
- self.assertEqual(2, geom[0].coord_dim)
- self.assertEqual(wkt_2d, geom.wkt)
- geom = OGRGeometry(wkt_3d, srid)
- geom.transform(srid)
- self.assertEqual(3, geom.coord_dim)
- self.assertEqual(3, geom[0].coord_dim)
- self.assertEqual(wkt_3d, geom.wkt)
- # Testing binary predicates, `assertIs` is used to check that bool is returned.
- def test_equivalence_regression(self):
- "Testing equivalence methods with non-OGRGeometry instances."
- self.assertIsNotNone(OGRGeometry("POINT(0 0)"))
- self.assertNotEqual(OGRGeometry("LINESTRING(0 0, 1 1)"), 3)
- def test_contains(self):
- self.assertIs(
- OGRGeometry("POINT(0 0)").contains(OGRGeometry("POINT(0 0)")), True
- )
- self.assertIs(
- OGRGeometry("POINT(0 0)").contains(OGRGeometry("POINT(0 1)")), False
- )
- def test_crosses(self):
- self.assertIs(
- OGRGeometry("LINESTRING(0 0, 1 1)").crosses(
- OGRGeometry("LINESTRING(0 1, 1 0)")
- ),
- True,
- )
- self.assertIs(
- OGRGeometry("LINESTRING(0 0, 0 1)").crosses(
- OGRGeometry("LINESTRING(1 0, 1 1)")
- ),
- False,
- )
- def test_disjoint(self):
- self.assertIs(
- OGRGeometry("LINESTRING(0 0, 1 1)").disjoint(
- OGRGeometry("LINESTRING(0 1, 1 0)")
- ),
- False,
- )
- self.assertIs(
- OGRGeometry("LINESTRING(0 0, 0 1)").disjoint(
- OGRGeometry("LINESTRING(1 0, 1 1)")
- ),
- True,
- )
- def test_equals(self):
- self.assertIs(
- OGRGeometry("POINT(0 0)").contains(OGRGeometry("POINT(0 0)")), True
- )
- self.assertIs(
- OGRGeometry("POINT(0 0)").contains(OGRGeometry("POINT(0 1)")), False
- )
- def test_intersects(self):
- self.assertIs(
- OGRGeometry("LINESTRING(0 0, 1 1)").intersects(
- OGRGeometry("LINESTRING(0 1, 1 0)")
- ),
- True,
- )
- self.assertIs(
- OGRGeometry("LINESTRING(0 0, 0 1)").intersects(
- OGRGeometry("LINESTRING(1 0, 1 1)")
- ),
- False,
- )
- def test_overlaps(self):
- self.assertIs(
- OGRGeometry("POLYGON ((0 0, 0 2, 2 2, 2 0, 0 0))").overlaps(
- OGRGeometry("POLYGON ((1 1, 1 5, 5 5, 5 1, 1 1))")
- ),
- True,
- )
- self.assertIs(
- OGRGeometry("POINT(0 0)").overlaps(OGRGeometry("POINT(0 1)")), False
- )
- def test_touches(self):
- self.assertIs(
- OGRGeometry("POLYGON ((0 0, 0 1, 1 1, 1 0, 0 0))").touches(
- OGRGeometry("LINESTRING(0 2, 2 0)")
- ),
- True,
- )
- self.assertIs(
- OGRGeometry("POINT(0 0)").touches(OGRGeometry("POINT(0 1)")), False
- )
- def test_within(self):
- self.assertIs(
- OGRGeometry("POINT(0.5 0.5)").within(
- OGRGeometry("POLYGON ((0 0, 0 1, 1 1, 1 0, 0 0))")
- ),
- True,
- )
- self.assertIs(
- OGRGeometry("POINT(0 0)").within(OGRGeometry("POINT(0 1)")), False
- )
- def test_from_gml(self):
- self.assertEqual(
- OGRGeometry("POINT(0 0)"),
- OGRGeometry.from_gml(
- '<gml:Point gml:id="p21" '
- 'srsName="http://www.opengis.net/def/crs/EPSG/0/4326">'
- ' <gml:pos srsDimension="2">0 0</gml:pos>'
- "</gml:Point>"
- ),
- )
- def test_empty(self):
- self.assertIs(OGRGeometry("POINT (0 0)").empty, False)
- self.assertIs(OGRGeometry("POINT EMPTY").empty, True)
- def test_empty_point_to_geos(self):
- p = OGRGeometry("POINT EMPTY", srs=4326)
- self.assertEqual(p.geos.ewkt, p.ewkt)
- def test_geometry_types(self):
- tests = [
- ("Point", 1, True),
- ("LineString", 2, True),
- ("Polygon", 3, True),
- ("MultiPoint", 4, True),
- ("Multilinestring", 5, True),
- ("MultiPolygon", 6, True),
- ("GeometryCollection", 7, True),
- ("CircularString", 8, True),
- ("CompoundCurve", 9, True),
- ("CurvePolygon", 10, True),
- ("MultiCurve", 11, True),
- ("MultiSurface", 12, True),
- # 13 (Curve) and 14 (Surface) are abstract types.
- ("PolyhedralSurface", 15, False),
- ("TIN", 16, False),
- ("Triangle", 17, False),
- ("Linearring", 2, True),
- # Types 1 - 7 with Z dimension have 2.5D enums.
- ("Point Z", -2147483647, True), # 1001
- ("LineString Z", -2147483646, True), # 1002
- ("Polygon Z", -2147483645, True), # 1003
- ("MultiPoint Z", -2147483644, True), # 1004
- ("Multilinestring Z", -2147483643, True), # 1005
- ("MultiPolygon Z", -2147483642, True), # 1006
- ("GeometryCollection Z", -2147483641, True), # 1007
- ("CircularString Z", 1008, True),
- ("CompoundCurve Z", 1009, True),
- ("CurvePolygon Z", 1010, True),
- ("MultiCurve Z", 1011, True),
- ("MultiSurface Z", 1012, True),
- ("PolyhedralSurface Z", 1015, False),
- ("TIN Z", 1016, False),
- ("Triangle Z", 1017, False),
- ("Point M", 2001, True),
- ("LineString M", 2002, True),
- ("Polygon M", 2003, True),
- ("MultiPoint M", 2004, True),
- ("MultiLineString M", 2005, True),
- ("MultiPolygon M", 2006, True),
- ("GeometryCollection M", 2007, True),
- ("CircularString M", 2008, True),
- ("CompoundCurve M", 2009, True),
- ("CurvePolygon M", 2010, True),
- ("MultiCurve M", 2011, True),
- ("MultiSurface M", 2012, True),
- ("PolyhedralSurface M", 2015, False),
- ("TIN M", 2016, False),
- ("Triangle M", 2017, False),
- ("Point ZM", 3001, True),
- ("LineString ZM", 3002, True),
- ("Polygon ZM", 3003, True),
- ("MultiPoint ZM", 3004, True),
- ("MultiLineString ZM", 3005, True),
- ("MultiPolygon ZM", 3006, True),
- ("GeometryCollection ZM", 3007, True),
- ("CircularString ZM", 3008, True),
- ("CompoundCurve ZM", 3009, True),
- ("CurvePolygon ZM", 3010, True),
- ("MultiCurve ZM", 3011, True),
- ("MultiSurface ZM", 3012, True),
- ("PolyhedralSurface ZM", 3015, False),
- ("TIN ZM", 3016, False),
- ("Triangle ZM", 3017, False),
- ]
- for test in tests:
- geom_type, num, supported = test
- with self.subTest(geom_type=geom_type, num=num, supported=supported):
- if supported:
- g = OGRGeometry(f"{geom_type} EMPTY")
- self.assertEqual(g.geom_type.num, num)
- else:
- type_ = geom_type.replace(" ", "")
- 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)
- def test_wkt_and_wkb_output(self):
- tests = [
- # 2D
- ("POINT (1 2)", "0101000000000000000000f03f0000000000000040"),
- (
- "LINESTRING (30 10,10 30)",
- "0102000000020000000000000000003e400000000000002"
- "44000000000000024400000000000003e40",
- ),
- (
- "POLYGON ((30 10,40 40,20 40,30 10))",
- "010300000001000000040000000000000000003e400000000000002440000000000000"
- "44400000000000004440000000000000344000000000000044400000000000003e4000"
- "00000000002440",
- ),
- (
- "MULTIPOINT (10 40,40 30)",
- "0104000000020000000101000000000000000000244000000000000044400101000000"
- "00000000000044400000000000003e40",
- ),
- (
- "MULTILINESTRING ((10 10,20 20),(40 40,30 30,40 20))",
- "0105000000020000000102000000020000000000000000002440000000000000244000"
- "0000000000344000000000000034400102000000030000000000000000004440000000"
- "00000044400000000000003e400000000000003e400000000000004440000000000000"
- "3440",
- ),
- (
- "MULTIPOLYGON (((30 20,45 40,10 40,30 20)),((15 5,40 10,10 20,15 5)))",
- "010600000002000000010300000001000000040000000000000000003e400000000000"
- "0034400000000000804640000000000000444000000000000024400000000000004440"
- "0000000000003e40000000000000344001030000000100000004000000000000000000"
- "2e40000000000000144000000000000044400000000000002440000000000000244000"
- "000000000034400000000000002e400000000000001440",
- ),
- (
- "GEOMETRYCOLLECTION (POINT (40 10))",
- "010700000001000000010100000000000000000044400000000000002440",
- ),
- # 3D
- (
- "POINT (1 2 3)",
- "0101000080000000000000f03f00000000000000400000000000000840",
- ),
- (
- "LINESTRING (30 10 3,10 30 3)",
- "0102000080020000000000000000003e40000000000000244000000000000008400000"
- "0000000024400000000000003e400000000000000840",
- ),
- (
- "POLYGON ((30 10 3,40 40 3,30 10 3))",
- "010300008001000000030000000000000000003e400000000000002440000000000000"
- "08400000000000004440000000000000444000000000000008400000000000003e4000"
- "000000000024400000000000000840",
- ),
- (
- "MULTIPOINT (10 40 3,40 30 3)",
- "0104000080020000000101000080000000000000244000000000000044400000000000"
- "000840010100008000000000000044400000000000003e400000000000000840",
- ),
- (
- "MULTILINESTRING ((10 10 3,20 20 3))",
- "0105000080010000000102000080020000000000000000002440000000000000244000"
- "00000000000840000000000000344000000000000034400000000000000840",
- ),
- (
- "MULTIPOLYGON (((30 20 3,45 40 3,30 20 3)))",
- "010600008001000000010300008001000000030000000000000000003e400000000000"
- "0034400000000000000840000000000080464000000000000044400000000000000840"
- "0000000000003e4000000000000034400000000000000840",
- ),
- (
- "GEOMETRYCOLLECTION (POINT (40 10 3))",
- "0107000080010000000101000080000000000000444000000000000024400000000000"
- "000840",
- ),
- ]
- for geom, wkb in tests:
- with self.subTest(geom=geom):
- g = OGRGeometry(geom)
- self.assertEqual(g.wkt, geom)
- self.assertEqual(g.wkb.hex(), wkb)
- def test_measure_is_measure_and_set_measure(self):
- geom = OGRGeometry("POINT (1 2 3)")
- self.assertIs(geom.is_measured, False)
- geom.set_measured(True)
- self.assertIs(geom.is_measured, True)
- self.assertEqual(geom.wkt, "POINT ZM (1 2 3 0)")
- geom.set_measured(False)
- self.assertIs(geom.is_measured, False)
- self.assertEqual(geom.wkt, "POINT (1 2 3)")
- msg = "Input to 'set_measured' must be a boolean, got 'None'"
- with self.assertRaisesMessage(ValueError, msg):
- geom.set_measured(None)
- def test_point_m_coordinate(self):
- geom = OGRGeometry("POINT ZM (1 2 3 4)")
- self.assertEqual(geom.m, 4)
- geom = OGRGeometry("POINT (1 2 3 4)")
- self.assertEqual(geom.m, 4)
- geom = OGRGeometry("POINT M (1 2 3)")
- self.assertEqual(geom.m, 3)
- geom = OGRGeometry("POINT Z (1 2 3)")
- self.assertEqual(geom.m, None)
- def test_point_m_tuple(self):
- geom = OGRGeometry("POINT ZM (1 2 3 4)")
- self.assertEqual(geom.tuple, (geom.x, geom.y, geom.z, geom.m))
- geom = OGRGeometry("POINT M (1 2 3)")
- self.assertEqual(geom.tuple, (geom.x, geom.y, geom.m))
- geom = OGRGeometry("POINT Z (1 2 3)")
- self.assertEqual(geom.tuple, (geom.x, geom.y, geom.z))
- geom = OGRGeometry("POINT (1 2 3)")
- self.assertEqual(geom.tuple, (geom.x, geom.y, geom.z))
- def test_point_m_wkt_wkb(self):
- wkt = "POINT ZM (1 2 3 4)"
- geom = OGRGeometry(wkt)
- self.assertEqual(geom.wkt, wkt)
- self.assertEqual(
- geom.wkb.hex(),
- "01b90b0000000000000000f03f00000000000000"
- "4000000000000008400000000000001040",
- )
- wkt = "POINT M (1 2 3)"
- geom = OGRGeometry(wkt)
- self.assertEqual(geom.wkt, wkt)
- self.assertEqual(
- geom.wkb.hex(),
- "01d1070000000000000000f03f00000000000000400000000000000840",
- )
- def test_point_m_dimension_types(self):
- geom = OGRGeometry("POINT ZM (1 2 3 4)")
- self.assertEqual(geom.geom_type.name, "PointZM")
- self.assertEqual(geom.geom_type.num, 3001)
- geom = OGRGeometry("POINT M (1 2 3)")
- self.assertEqual(geom.geom_type.name, "PointM")
- self.assertEqual(geom.geom_type.num, 2001)
- @skipIf(geos_version_tuple() < (3, 12), "GEOS >= 3.12.0 is required")
- def test_point_m_dimension_geos(self):
- geo_zm = OGRGeometry("POINT ZM (1 2 3 4)")
- self.assertEqual(geo_zm.geos.wkt, "POINT ZM (1 2 3 4)")
- geo_m = OGRGeometry("POINT M (1 2 3)")
- self.assertEqual(geo_m.geos.wkt, "POINT M (1 2 3)")
- @mock.patch("django.contrib.gis.geos.libgeos.geos_version", lambda: b"3.11.0")
- def test_point_m_dimension_geos_version(self):
- geo_zm = OGRGeometry("POINT ZM (1 2 3 4)")
- self.assertEqual(geo_zm.geos.wkt, "POINT Z (1 2 3)")
- geo_m = OGRGeometry("POINT M (1 2 3)")
- self.assertEqual(geo_m.geos.wkt, "POINT (1 2)")
- def test_centroid(self):
- point = OGRGeometry("POINT (1 2 3)")
- self.assertEqual(point.centroid.wkt, "POINT (1 2)")
- linestring = OGRGeometry("LINESTRING (0 0 0, 1 1 1, 2 2 2)")
- self.assertEqual(linestring.centroid.wkt, "POINT (1 1)")
- polygon = OGRGeometry("POLYGON((0 0, 10 0, 10 10, 0 10, 0 0))")
- self.assertEqual(polygon.centroid.wkt, "POINT (5 5)")
- multipoint = OGRGeometry("MULTIPOINT (0 0,10 10)")
- self.assertEqual(multipoint.centroid.wkt, "POINT (5 5)")
- multilinestring = OGRGeometry(
- "MULTILINESTRING ((0 0,0 10,0 20),(10 0,10 10,10 20))"
- )
- self.assertEqual(multilinestring.centroid.wkt, "POINT (5 10)")
- multipolygon = OGRGeometry(
- "MULTIPOLYGON(((0 0, 10 0, 10 10, 0 10, 0 0)),"
- "((20 20, 20 30, 30 30, 30 20, 20 20)))"
- )
- self.assertEqual(multipolygon.centroid.wkt, "POINT (15 15)")
- geometrycollection = OGRGeometry(
- "GEOMETRYCOLLECTION (POINT (110 260),LINESTRING (110 0,110 60))"
- )
- self.assertEqual(geometrycollection.centroid.wkt, "POINT (110 30)")
- def test_linestring_m_dimension(self):
- geom = OGRGeometry("LINESTRING(0 1 2 10, 1 2 3 11, 2 3 4 12)")
- self.assertIs(geom.is_measured, True)
- self.assertEqual(geom.m, [10.0, 11.0, 12.0])
- self.assertEqual(geom[0], (0.0, 1.0, 2.0, 10.0))
- geom = OGRGeometry("LINESTRING M (0 1 10, 1 2 11)")
- self.assertIs(geom.is_measured, True)
- self.assertEqual(geom.m, [10.0, 11.0])
- self.assertEqual(geom[0], (0.0, 1.0, 10.0))
- geom.set_measured(False)
- self.assertIs(geom.is_measured, False)
- self.assertIs(geom.m, None)
- def test_polygon_m_dimension(self):
- geom = OGRGeometry("POLYGON Z ((0 0 0, 10 0 0, 10 10 0, 0 10 0, 0 0 0))")
- self.assertIs(geom.is_measured, False)
- self.assertEqual(
- geom.shell.wkt, "LINEARRING (0 0 0,10 0 0,10 10 0,0 10 0,0 0 0)"
- )
- geom = OGRGeometry("POLYGON M ((0 0 0, 10 0 0, 10 10 0, 0 10 0, 0 0 0))")
- self.assertIs(geom.is_measured, True)
- self.assertEqual(
- geom.shell.wkt, "LINEARRING M (0 0 0,10 0 0,10 10 0,0 10 0,0 0 0)"
- )
- geom = OGRGeometry(
- "POLYGON ZM ((0 0 0 1, 10 0 0 1, 10 10 0 1, 0 10 0 1, 0 0 0 1))"
- )
- self.assertIs(geom.is_measured, True)
- self.assertEqual(
- geom.shell.wkt,
- "LINEARRING ZM (0 0 0 1,10 0 0 1,10 10 0 1,0 10 0 1,0 0 0 1)",
- )
- geom.set_measured(False)
- self.assertEqual(geom.wkt, "POLYGON ((0 0 0,10 0 0,10 10 0,0 10 0,0 0 0))")
- self.assertEqual(
- geom.shell.wkt, "LINEARRING (0 0 0,10 0 0,10 10 0,0 10 0,0 0 0)"
- )
- def test_multi_geometries_m_dimension(self):
- tests = [
- "MULTIPOINT M ((10 40 10), (40 30 10), (20 20 10))",
- "MULTIPOINT ZM ((10 40 0 10), (40 30 1 10), (20 20 1 10))",
- "MULTILINESTRING M ((10 10 1, 20 20 2),(40 40 1, 30 30 2))",
- "MULTILINESTRING ZM ((10 10 0 1, 20 20 0 2),(40 40 1, 30 30 0 2))",
- (
- "MULTIPOLYGON ZM (((30 20 1 0, 45 40 1 0, 30 20 1 0)),"
- "((15 5 0 0, 40 10 0 0, 15 5 0 0)))"
- ),
- (
- "GEOMETRYCOLLECTION M (POINT M (40 10 0),"
- "LINESTRING M (10 10 0, 20 20 0, 10 40 0))"
- ),
- (
- "GEOMETRYCOLLECTION ZM (POINT ZM (40 10 0 1),"
- "LINESTRING ZM (10 10 1 0, 20 20 1 0, 10 40 1 0))"
- ),
- ]
- for geom_input in tests:
- with self.subTest(geom_input=geom_input):
- geom = OGRGeometry(geom_input)
- self.assertIs(geom.is_measured, True)
- def test_has_curve(self):
- for geom in self.geometries.curved_geoms:
- with self.subTest(wkt=geom.wkt):
- geom = OGRGeometry(geom.wkt)
- self.assertIs(geom.has_curve, True)
- msg = f"GEOS does not support {geom.__class__.__qualname__}."
- with self.assertRaisesMessage(GEOSException, msg):
- geom.geos
- geom = OGRGeometry("POINT (0 1)")
- self.assertIs(geom.has_curve, False)
- def test_get_linear_geometry(self):
- geom = OGRGeometry("CIRCULARSTRING (-0.797 0.466,-0.481 0.62,-0.419 0.473)")
- linear = geom.get_linear_geometry()
- self.assertEqual(linear.geom_name, "LINESTRING")
- self.assertIs(linear.has_curve, False)
- def test_get_linear_geometry_no_conversion_possible(self):
- wkt = "POINT (0 0)"
- geom = OGRGeometry(wkt)
- geom2 = geom.get_linear_geometry()
- self.assertEqual(geom2.wkt, wkt)
- def test_get_curve_geometry(self):
- linear_string = OGRGeometry(
- "LINESTRING (-0.797 0.466,-0.797500910583869 0.479079607685707,"
- "-0.797096828208069 0.49216256476959,-0.795789684575482 0.505186328593822,"
- "-0.793585728444384 0.518088639471983,-0.79049549575663 0.530807818319715,"
- "-0.786533759270668 0.543283061509385,-0.781719457941079 0.555454731539925,"
- "-0.776075606381369 0.567264642132187,-0.769629184843353 0.578656336386302,"
- "-0.76241101023902 0.589575356672327,-0.754455588821145 0.599969504963013,"
- "-0.745800951227352 0.609789092364991,-0.736488470675795 0.618987176654798,"
- "-0.726562665181888 0.627519786684672,-0.716070984741265 0.635346132585369,"
- "-0.705063584496685 0.642428800760598,-0.693593084972889 0.648733932741749,"
- "-0.681714320525941 0.654231387047048,-0.669484077209319 0.658894883272069,"
- "-0.656960821309923 0.662702127722269,-0.644204419852031 0.665634919987354,"
- "-0.631275854404748 0.667679239947688,-0.618236929561618 0.668825314797118,"
- "-0.60514997748578 0.669067665761503,-0.592077559933017 0.66840513428977,"
- "-0.579082169177269 0.666840887592428,-0.566225929268313 0.664382403500809,"
- "-0.553570299049824 0.661041434719465,-0.541175778357228 0.656833952642756,"
- "-0.529101618800212 0.651780071004197,-0.5174055405123 0.645903949723276,"
- "-0.506143456221622 0.639233679409784,-0.495369203961872 0.631801147077652,"
- "-0.485134289701335 0.623641883709865,-0.475487641120239 0.614794894404014,"
- "-0.46647537371355 0.605302471909454,-0.458140570337321 0.595209994448282,"
- "-0.450523075252448 0.58456570878613,-0.443659303650563 0.573420499590156,"
- "-0.437582067572208 0.561827646176397,-0.432320419050072 0.549842567809747,"
- "-0.427899511226613 0.537522558773986,-0.424340478110267 0.524926514478182,"
- "-0.421660333544978 0.512114649909193,-0.419871889876113 0.499148211775737,"
- "-0.418983696701434 0.486089185720561,-0.419 0.473)"
- )
- curve = linear_string.get_curve_geometry()
- self.assertEqual(curve.geom_name, "CIRCULARSTRING")
- self.assertEqual(
- curve.wkt,
- "CIRCULARSTRING (-0.797 0.466,-0.618236929561618 "
- "0.668825314797118,-0.419 0.473)",
- )
- def test_get_curve_geometry_no_conversion_possible(self):
- geom = OGRGeometry("LINESTRING (0 0, 1 0, 2 0)")
- geom2 = geom.get_curve_geometry()
- self.assertEqual(geom2.wkt, geom.wkt)
- def test_curved_geometries(self):
- for geom in self.geometries.curved_geoms:
- with self.subTest(wkt=geom.wkt, geom_name=geom.name):
- g = OGRGeometry(geom.wkt)
- self.assertEqual(geom.name, g.geom_type.name)
- self.assertEqual(geom.num, g.geom_type.num)
- msg = f"GEOS does not support {g.__class__.__qualname__}."
- with self.assertRaisesMessage(GEOSException, msg):
- g.geos
- def test_circularstring_has_linestring_features(self):
- geom = OGRGeometry("CIRCULARSTRING ZM (1 5 0 1, 6 2 0 2, 7 3 0 3)")
- self.assertIsInstance(geom, CircularString)
- self.assertEqual(geom.x, [1, 6, 7])
- self.assertEqual(geom.y, [5, 2, 3])
- self.assertEqual(geom.z, [0, 0, 0])
- self.assertEqual(geom.m, [1, 2, 3])
- self.assertEqual(
- geom.tuple,
- ((1.0, 5.0, 0.0, 1.0), (6.0, 2.0, 0.0, 2.0), (7.0, 3.0, 0.0, 3.0)),
- )
- self.assertEqual(geom[0], (1, 5, 0, 1))
- self.assertEqual(len(geom), 3)
- def test_curvepolygon_has_polygon_features(self):
- geom = OGRGeometry(
- "CURVEPOLYGON ZM (CIRCULARSTRING ZM (0 0 0 0, 4 0 0 0, 4 4 0 0, 0 4 0 0, "
- "0 0 0 0), (1 1 0 0, 3 3 0 0, 3 1 0 0, 1 1 0 0))"
- )
- self.assertIsInstance(geom, CurvePolygon)
- self.assertIsInstance(geom.shell, CircularString)
|