mirror of
https://github.com/django/django.git
synced 2025-03-08 16:32:32 +00:00
The test is a regression for refs #4740 to show that the original fix of OGRIndexError is no longer needed. This is similar to the removal of GEOSIndexError in 197b1878105504b5ac7e399e1f52a6093c88baa3.
558 lines
23 KiB
Python
558 lines
23 KiB
Python
import json
|
|
import pickle
|
|
import unittest
|
|
from binascii import b2a_hex
|
|
|
|
from django.contrib.gis.gdal import (
|
|
CoordTransform, GDALException, OGRGeometry, OGRGeomType, SpatialReference,
|
|
)
|
|
from django.template import Context
|
|
from django.template.engine import Engine
|
|
|
|
from ..test_data import TestDataMixin
|
|
|
|
|
|
class OGRGeomTest(unittest.TestCase, 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(9)
|
|
|
|
# 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_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(b2a_hex(wkb).upper(), g.hex.encode())
|
|
# 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)
|
|
with self.assertRaises(IndexError):
|
|
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)
|
|
with self.assertRaises(IndexError):
|
|
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))
|
|
|
|
# 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))
|
|
with self.assertRaises(IndexError):
|
|
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)
|
|
|
|
prec = 3
|
|
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)
|
|
|
|
prec = 3
|
|
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.assertEqual(d1, d2)
|
|
self.assertEqual(d1, a - b) # __sub__ is difference operator
|
|
a -= b # testing __isub__
|
|
self.assertEqual(d1, a)
|
|
|
|
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.assertEqual(i1, i2)
|
|
self.assertEqual(i1, a & b) # __and__ is intersection operator
|
|
a &= b # testing __iand__
|
|
self.assertEqual(i1, a)
|
|
|
|
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.assertEqual(d1, d2)
|
|
self.assertEqual(d1, a ^ b) # __xor__ is symmetric difference operator
|
|
a ^= b # testing __ixor__
|
|
self.assertEqual(d1, a)
|
|
|
|
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.assertEqual(u1, u2)
|
|
self.assertEqual(u1, a | b) # __or__ is union operator
|
|
a |= b # testing __ior__
|
|
self.assertEqual(u1, a)
|
|
|
|
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)
|