def test_long_attribute_names(self):
line = Line(
[(1.0, 5.0), (5.0, 5.0), (5.0, 1.0), (3.0, 3.0), (1.0, 1.0)],
properties={
"geom_id": 27,
"name": "test line",
"description": "line for testing",
"description_en": "Line for testing."
},
crs=LonLatWGS84)
with warnings.catch_warnings():
warnings.simplefilter("ignore")
line.to_shapefile(os.path.join(TESTDIR,
"data/line_truncated_attr"))
for fnm in ("line_truncated_attr.shx", "line_truncated_attr.shx",
"line_truncated_attr.dbf", "line_truncated_attr.prj"):
self.assertTrue(os.path.isfile(os.path.join(TESTDIR, "data", fnm)))
line2 = read_shapefile(
os.path.join(TESTDIR, "data", "line_truncated_attr"))[0]
self.assertTrue("DESCRIPTIO" in line2.properties)
self.assertTrue("DESCRIPTI2" in line2.properties)
self.assertTrue("GEOM_ID" in line2.properties)
self.assertTrue("NAME" in line2.properties)
def test_long_attribute_names(self):
line = Line([(1.0,5.0),(5.0,5.0),(5.0,1.0),(3.0,3.0),(1.0,1.0)],
properties={
"geom_id": 27,
"name": "test line",
"description": "line for testing",
"description_en": "Line for testing."
},
crs=LonLatWGS84)
with warnings.catch_warnings():
warnings.simplefilter("ignore")
line.to_shapefile(os.path.join(TESTDIR, "data/line_truncated_attr"))
for fnm in ("line_truncated_attr.shx",
"line_truncated_attr.shx",
"line_truncated_attr.dbf",
"line_truncated_attr.prj"):
self.assertTrue(os.path.isfile(os.path.join(TESTDIR, "data", fnm)))
line2 = read_shapefile(os.path.join(TESTDIR, "data", "line_truncated_attr"))[0]
self.assertTrue("DESCRIPTIO" in line2.properties)
self.assertTrue("DESCRIPTI2" in line2.properties)
self.assertTrue("GEOM_ID" in line2.properties)
self.assertTrue("NAME" in line2.properties)
class TestShapefile(unittest.TestCase):
def setUp(self):
self.points = [Point((1, 1), data={"species": "T. officianale"}, crs=LonLatWGS84),
Point((3, 1), data={"species": "C. tectorum"}, crs=LonLatWGS84),
Point((4, 3), data={"species": "M. alba"}, crs=LonLatWGS84),
Point((2, 2), data={"species": "V. cracca"}, crs=LonLatWGS84)]
self.multipoint = Multipoint([(1,1), (3,1), (4,3), (2,2)],
data={"species": ["T. officianale", "C. tectorum",
"M. alba", "V. cracca"]},
crs=LonLatWGS84)
self.line = Line([(1.0,5.0),(5.0,5.0),(5.0,1.0),(3.0,3.0),(1.0,1.0)],
properties={"geom_id": 27, "name": "test line"},
crs=LonLatWGS84)
self.polygon = Polygon([(1.0,5.0),(5.0,5.0),(5.0,1.0),(3.0,3.0),(1.0,1.0)],
crs=LonLatWGS84)
self.points3 = [Point((1, 1, 0), crs=LonLatWGS84),
Point((3, 1, 3), crs=LonLatWGS84),
Point((4, 3, 2), crs=LonLatWGS84),
Point((2, 2, -1), crs=LonLatWGS84)]
self.line3 = Line([(1,5,2),(5,5,-1),(5,1,3),(3,3,1),(1,1,0)], crs=LonLatWGS84)
self.polygon3 = Polygon([(1,5,2),(5,5,-1),(5,1,3),(3,3,1),(1,1,0)], crs=LonLatWGS84)
testfiles = ["points.shp", "line.shp", "polygon.shp"]
if any(not exists(join(TMPDATA, "shapefiles/", fnm)) for fnm in testfiles):
self.saveTestData()
return
def saveTestData(self):
testfiles = [(self.multipoint, "points"),
(self.line, "line"),
(self.polygon, "polygon")]
os.makedirs(os.path.join(TMPDATA, "shapefiles"))
for (geom, fnm) in testfiles:
geom.to_shapefile(os.path.join(TMPDATA, "shapefiles", fnm))
return
def assertGeomEqual(self, this, that):
self.assertTrue(np.all(this.get_vertices() == that.get_vertices()))
try:
self.assertEqual(this.crs.get_proj4(), that.crs.get_proj4())
except AttributeError:
print("warning: crs equality not established")
return
def test_writepoint(self):
point = self.points[0]
point.to_shapefile(os.path.join(TESTDIR, "data/point"))
for fnm in ("point.shx", "point.shx", "point.dbf", "point.prj"):
self.assertTrue(os.path.isfile(os.path.join(TESTDIR, "data", fnm)))
return
def test_writepoints(self):
points = self.points
shp.write_shapefile(os.path.join(TESTDIR, "data/points.shp"), *points)
for fnm in ("points.shx", "points.shx", "points.dbf", "points.prj"):
self.assertTrue(os.path.isfile(os.path.join(TESTDIR, "data", fnm)))
return
def test_writemultipoint(self):
mp = Multipoint(self.points)
mp.to_shapefile(os.path.join(TESTDIR, "data/multipoint"))
for fnm in ("multipoint.shx", "multipoint.shx", "multipoint.dbf", "multipoint.prj"):
self.assertTrue(os.path.isfile(os.path.join(TESTDIR, "data", fnm)))
return
def test_writeline(self):
self.line.to_shapefile(os.path.join(TESTDIR, "data/line"))
for fnm in ("line.shx", "line.shx", "line.dbf", "line.prj"):
self.assertTrue(os.path.isfile(os.path.join(TESTDIR, "data", fnm)))
return
def test_writepoly(self):
self.polygon.to_shapefile(os.path.join(TESTDIR, "data/polygon"))
for fnm in ("polygon.shx", "polygon.shx", "polygon.dbf", "polygon.prj"):
self.assertTrue(os.path.isfile(os.path.join(TESTDIR, "data", fnm)))
return
def test_writepoints3(self):
mp = Multipoint(self.points3)
mp.to_shapefile(os.path.join(TESTDIR, "data/multipointz"))
for fnm in ("multipointz.shx", "multipointz.shx", "multipointz.dbf", "multipointz.prj"):
self.assertTrue(os.path.isfile(os.path.join(TESTDIR, "data", fnm)))
return
def test_writeline3(self):
self.line3.to_shapefile(os.path.join(TESTDIR, "data/linez"))
for fnm in ("linez.shx", "linez.shx", "linez.dbf", "linez.prj"):
self.assertTrue(os.path.isfile(os.path.join(TESTDIR, "data", fnm)))
return
def test_writepoly3(self):
self.polygon3.to_shapefile(os.path.join(TESTDIR, "data/polygonz"))
for fnm in ("polygonz.shx", "polygonz.shx", "polygonz.dbf", "polygonz.prj"):
self.assertTrue(os.path.isfile(os.path.join(TESTDIR, "data", fnm)))
return
def test_write_collection_multipoint(self):
mp = Multipoint([p.vertex for p in self.points])
mp0 = copy(mp)
mp1 = copy(mp.shift((4, 2)))
mp2 = copy(mp.shift((-2, 3)))
shp.write_shapefile(os.path.join(TESTDIR, "data/mp_collection.shp"),
mp0, mp1, mp2)
for fnm in ("mp_collection.shx", "mp_collection.shx", "mp_collection.dbf", "mp_collection.prj"):
self.assertTrue(os.path.isfile(os.path.join(TESTDIR, "data", fnm)))
return
def test_write_collection_lines(self):
line0 = copy(self.line)
line1 = copy(self.line.shift((4, 2)))
line2 = copy(self.line.shift((-2, 3)))
shp.write_shapefile(os.path.join(TESTDIR, "data/line_collection.shp"),
line0, line1, line2)
for fnm in ("line_collection.shx", "line_collection.shx", "line_collection.dbf", "line_collection.prj"):
self.assertTrue(os.path.isfile(os.path.join(TESTDIR, "data", fnm)))
return
def test_read_points(self):
points = read_shapefile(os.path.join(TESTDATA, "shp_input", "points"))
self.assertEqual(len(points), 4)
pt = points[0]
self.assertTrue("+proj=lonlat" in pt.crs.get_proj4())
self.assertTrue("+a=6378137.0" in pt.crs.get_proj4())
self.assertTrue("+f=0.00335281" in pt.crs.get_proj4())
mp = Multipoint(points)
self.assertEqual(mp.d["species"], ['T. officianale', 'C. tectorum', 'M. alba', 'V. cracca'])
self.assertEqual(mp.d["ID"], ['0', '1', '2', '3'])
self.assertEqual(mp.coordinates, ((1.0, 3.0, 4.0, 2.0), (1.0, 1.0, 3.0, 2.0)))
def test_read_line(self):
line = read_shapefile(os.path.join(TESTDATA, "shp_input", "line"))[0]
self.assertTrue("+proj=lonlat" in line.crs.get_proj4())
self.assertTrue("+a=6378137.0" in line.crs.get_proj4())
self.assertTrue("+f=0.00335281" in line.crs.get_proj4())
self.assertEqual(line.coordinates, ((1.0, 5.0, 5.0, 3.0, 1.0), (5.0, 5.0, 1.0, 3.0, 1.0)))
return
def test_read_polygon(self):
polygon = read_shapefile(os.path.join(TESTDATA, "shp_input", "polygon"))[0]
self.assertTrue("+proj=lonlat" in polygon.crs.get_proj4())
self.assertTrue("+a=6378137.0" in polygon.crs.get_proj4())
self.assertTrue("+f=0.00335281" in polygon.crs.get_proj4())
self.assertEqual(polygon.coordinates, ((1.0, 5.0, 5.0, 3.0, 1.0), (5.0, 5.0, 1.0, 3.0, 1.0)))
return
def test_read_points_newp(self):
# Read a multipoint with a projected cooridnate system
newp = read_shapefile(os.path.join(TESTDATA, "shp_input",
"newp_nsidc_north"))
proj4 = ('+proj=stere +lat_0=90 +lat_ts=70 +lon_0=-45 +k=1 +x_0=0 '
'+y_0=0 +a=6378273 +b=6356889.449 +units=m +no_defs')
for part in proj4.split():
self.assertTrue(part[:8] in newp[0].crs.get_proj4())
coords = list(zip(*[pt.vertex[:2] for pt in newp]))
self.assertEqual(coords, [(521236.8297444395, 521236.8297444395,
521236.8297444395, 547490.4452879033,
547490.4452879033, 547490.4452879033,
587584.1578033275, 587584.1578033275,
587584.1578033275, 571828.4918982167,
571828.4918982167),
(-888853.1384770898, -888853.1384770898,
-888853.1384770898, -902049.3617542256,
-902049.3617542256, -902049.3617542256,
-871214.0673764511, -871214.0673764511,
-871214.0673764511, -850080.914674058,
-850080.914674058)])
meterno = [pt.properties["meterno"] for pt in newp]
self.assertEqual(meterno, ['IMS1/1', 'IMS2/1', '5952/2', 'IMS4/1',
'5953/2', '1963/13', 'IMS5/1', '5213/A',
'2121/13', 'IMS3/1', '3613/2'])
depth = [pt.properties["depth_m"] for pt in newp]
self.assertEqual(depth, ['73', '143', '247', '86', '147', '250', '74',
'142', '235', '150', '248'])
return
class TestShapefile(unittest.TestCase):
def setUp(self):
self.points = [Point((1, 1), data={"species": "T. officianale"}, crs=LonLatWGS84),
Point((3, 1), data={"species": "C. tectorum"}, crs=LonLatWGS84),
Point((4, 3), data={"species": "M. alba"}, crs=LonLatWGS84),
Point((2, 2), data={"species": "V. cracca"}, crs=LonLatWGS84)]
self.multipoint = Multipoint([(1,1), (3,1), (4,3), (2,2)],
data={"species": ["T. officianale", "C. tectorum",
"M. alba", "V. cracca"]},
crs=LonLatWGS84)
self.line = Line([(1.0,5.0),(5.0,5.0),(5.0,1.0),(3.0,3.0),(1.0,1.0)],
crs=LonLatWGS84)
self.polygon = Polygon([(1.0,5.0),(5.0,5.0),(5.0,1.0),(3.0,3.0),(1.0,1.0)],
crs=LonLatWGS84)
self.points3 = [Point((1, 1, 0), crs=LonLatWGS84),
Point((3, 1, 3), crs=LonLatWGS84),
Point((4, 3, 2), crs=LonLatWGS84),
Point((2, 2, -1), crs=LonLatWGS84)]
self.line3 = Line([(1,5,2),(5,5,-1),(5,1,3),(3,3,1),(1,1,0)], crs=LonLatWGS84)
self.polygon3 = Polygon([(1,5,2),(5,5,-1),(5,1,3),(3,3,1),(1,1,0)], crs=LonLatWGS84)
testfiles = ["points.shp", "line.shp", "polygon.shp"]
if any(not exists(join(TESTDATA, "shapefiles/", fnm)) for fnm in testfiles):
self.saveTestData()
return
def saveTestData(self):
testfiles = [(self.multipoint, "points"),
(self.line, "line"),
(self.polygon, "polygon")]
for (geom, fnm) in testfiles:
geom.to_shapefile(os.path.join(TESTDATA, "shapefiles", fnm))
return
def assertGeomEqual(self, this, that):
self.assertTrue(np.all(this.get_vertices() == that.get_vertices()))
self.assertEqual(this._crs, that._crs)
return
def test_writepoints(self):
mp = Multipoint([p.vertex for p in self.points])
mp.to_shapefile("data/points_shp")
return
def test_writeline(self):
self.line.to_shapefile("data/line_shp")
return
def test_writepoly(self):
self.polygon.to_shapefile("data/polygon_shp")
return
def test_writepoints3(self):
mp = Multipoint([p.vertex for p in self.points3])
mp.to_shapefile("data/pointsz_shp")
return
def test_writeline3(self):
self.line3.to_shapefile("data/linez_shp")
return
def test_writepoly3(self):
self.polygon3.to_shapefile("data/polygonz_shp")
return
def test_write_collection_points(self):
mp = Multipoint([p.vertex for p in self.points])
mp0 = copy(mp)
mp1 = copy(mp.shift((4, 2)))
mp2 = copy(mp.shift((-2, 3)))
shp.write_shapefile([mp0, mp1, mp2], "data/points_collection")
return
def test_write_collection_lines(self):
line0 = copy(self.line)
line1 = copy(self.line.shift((4, 2)))
line2 = copy(self.line.shift((-2, 3)))
shp.write_shapefile([line0, line1, line2], "data/line_collection")
return
def test_dbase_type(self):
self.assertEqual(shp.property_field_type(1.0), "N")
self.assertEqual(shp.property_field_type(1), "N")
self.assertEqual(shp.property_field_type(np.float32(1.0)), "N")
self.assertEqual(shp.property_field_type(np.int16(1)), "N")
#self.assertEqual(shp.property_field_type(1.0), "O")
#self.assertEqual(shp.property_field_type(1), "I")
#self.assertEqual(shp.property_field_type(np.float32(1.0)), "O")
#self.assertEqual(shp.property_field_type(np.int16(1)), "I")
#self.assertEqual(shp.property_field_type(True), "L")
#self.assertEqual(shp.property_field_type(False), "L")
self.assertEqual(shp.property_field_type("pale ale"), "C")
self.assertEqual(shp.property_field_type(datetime.date(1986, 8, 17)), "D")
self.assertEqual(shp.property_field_type(datetime.datetime(2013, 5, 4, 20, 40, 21)), "@")
return
def test_read_points(self):
shps = read_shapefile(os.path.join(TESTDATA, "newp"))
mp = shps[0]
self.assertEqual(mp.vertices,
[(-14.612, 80.50906666666667), (-14.612, 80.50906666666667),
(-14.612, 80.50906666666667), (-13.744733333333333, 80.28181666666667),
(-13.744733333333333, 80.28181666666667), (-13.744733333333333, 80.28181666666667),
(-11.002583333333334, 80.32173333333333), (-11.002583333333334, 80.32173333333333),
(-11.002583333333334, 80.32173333333333), (-11.07225, 80.56316666666666),
(-11.07225, 80.56316666666666)])
self.assertEqual(mp.data.getfield("meterno"),
['IMS1/1', 'IMS2/1', '5952/2', 'IMS4/1', '5953/2', '1963/13',
'IMS5/1', '5213/A', '2121/13', 'IMS3/1', '3613/2'])
self.assertEqual(mp.data.getfield("depth_m"),
[73, 143, 247, 86, 147, 250, 74, 142, 235, 150, 248])
return
def test_read_multipoint_attributes(self):
mp = read_shapefile(os.path.join(TESTDATA, "shapefiles", "points"))
self.assertEqual(mp[0].data.getfield("species"), self.multipoint.data.getfield("species"))
return
def test_read_line(self):
line = read_shapefile(os.path.join(TESTDATA, "shapefiles", "line"))[0]
self.assertGeomEqual(line, self.line)
return
def test_read_polygon(self):
polygon = read_shapefile(os.path.join(TESTDATA, "shapefiles", "polygon"))[0]
self.assertGeomEqual(polygon, self.polygon)
return
class TestShapefile(unittest.TestCase):
def setUp(self):
self.points = [
Point((1, 1),
properties={"species": "T. officianale"},
crs=LonLatWGS84),
Point((3, 1),
properties={"species": "C. tectorum"},
crs=LonLatWGS84),
Point((4, 3), properties={"species": "M. alba"}, crs=LonLatWGS84),
Point((2, 2), properties={"species": "V. cracca"}, crs=LonLatWGS84)
]
self.multipoint = Multipoint(
[(1, 1), (3, 1), (4, 3), (2, 2)],
data={
"species":
["T. officianale", "C. tectorum", "M. alba", "V. cracca"]
},
crs=LonLatWGS84)
self.line = Line([(1.0, 5.0), (5.0, 5.0), (5.0, 1.0), (3.0, 3.0),
(1.0, 1.0)],
properties={
"geom_id": 27,
"name": "test line"
},
crs=LonLatWGS84)
self.polygon = Polygon([(1.0, 5.0), (5.0, 5.0), (5.0, 1.0), (3.0, 3.0),
(1.0, 1.0)],
crs=LonLatWGS84)
self.points3 = [
Point((1, 1, 0), crs=LonLatWGS84),
Point((3, 1, 3), crs=LonLatWGS84),
Point((4, 3, 2), crs=LonLatWGS84),
Point((2, 2, -1), crs=LonLatWGS84)
]
self.line3 = Line([(1, 5, 2), (5, 5, -1), (5, 1, 3), (3, 3, 1),
(1, 1, 0)],
crs=LonLatWGS84)
self.polygon3 = Polygon([(1, 5, 2), (5, 5, -1), (5, 1, 3), (3, 3, 1),
(1, 1, 0)],
crs=LonLatWGS84)
testfiles = ["points.shp", "line.shp", "polygon.shp"]
if any(not exists(join(TMPDATA, "shapefiles/", fnm))
for fnm in testfiles):
self.saveTestData()
def saveTestData(self):
testfiles = [(self.multipoint, "points"), (self.line, "line"),
(self.polygon, "polygon")]
if not os.path.isdir(os.path.join(TMPDATA, "shapefiles")):
os.makedirs(os.path.join(TMPDATA, "shapefiles"))
for (geom, fnm) in testfiles:
geom.to_shapefile(os.path.join(TMPDATA, "shapefiles", fnm))
def assertGeomEqual(self, this, that):
self.assertTrue(np.all(this.get_vertices() == that.get_vertices()))
try:
self.assertEqual(this.crs.get_proj4(), that.crs.get_proj4())
except AttributeError:
print("warning: crs equality not established")
def test_write_point(self):
point = self.points[0]
point.to_shapefile(os.path.join(TESTDIR, "data/point"))
for fnm in ("point.shx", "point.shx", "point.dbf", "point.prj"):
self.assertTrue(os.path.isfile(os.path.join(TESTDIR, "data", fnm)))
def test_write_points(self):
points = self.points
shp.write_shapefile(os.path.join(TESTDIR, "data/points.shp"), *points)
for fnm in ("points.shx", "points.shx", "points.dbf", "points.prj"):
self.assertTrue(os.path.isfile(os.path.join(TESTDIR, "data", fnm)))
def test_write_line(self):
self.line.to_shapefile(os.path.join(TESTDIR, "data/line"))
for fnm in ("line.shx", "line.shx", "line.dbf", "line.prj"):
self.assertTrue(os.path.isfile(os.path.join(TESTDIR, "data", fnm)))
def test_write_poly(self):
self.polygon.to_shapefile(os.path.join(TESTDIR, "data/polygon"))
for fnm in ("polygon.shx", "polygon.shx", "polygon.dbf",
"polygon.prj"):
self.assertTrue(os.path.isfile(os.path.join(TESTDIR, "data", fnm)))
def test_write_points3(self):
mp = Multipoint(self.points3)
mp.to_shapefile(os.path.join(TESTDIR, "data/multipointz"))
for fnm in ("multipointz.shx", "multipointz.shx", "multipointz.dbf",
"multipointz.prj"):
self.assertTrue(os.path.isfile(os.path.join(TESTDIR, "data", fnm)))
def test_write_line3(self):
self.line3.to_shapefile(os.path.join(TESTDIR, "data/linez"))
for fnm in ("linez.shx", "linez.shx", "linez.dbf", "linez.prj"):
self.assertTrue(os.path.isfile(os.path.join(TESTDIR, "data", fnm)))
def test_write_poly3(self):
self.polygon3.to_shapefile(os.path.join(TESTDIR, "data/polygonz"))
for fnm in ("polygonz.shx", "polygonz.shx", "polygonz.dbf",
"polygonz.prj"):
self.assertTrue(os.path.isfile(os.path.join(TESTDIR, "data", fnm)))
def test_write_multipoint(self):
mp = Multipoint(self.points)
mp.to_shapefile(os.path.join(TESTDIR, "data/multipoint"))
for fnm in ("multipoint.shx", "multipoint.shx", "multipoint.dbf",
"multipoint.prj"):
self.assertTrue(os.path.isfile(os.path.join(TESTDIR, "data", fnm)))
def test_write_multiline(self):
g = Multiline([[(0, 0), (1, 1),
(2, 2)], [(1, 0), (2, 1),
(3, 2)], [(2, 0), (1, 1), (0, 2)]],
crs=LonLatWGS84)
g.to_shapefile(os.path.join(TESTDIR, "data/multiline"))
for fnm in ("multiline.shx", "multiline.shx", "multiline.dbf",
"multiline.prj"):
self.assertTrue(os.path.isfile(os.path.join(TESTDIR, "data", fnm)))
def test_write_multipolygon(self):
g = Multipolygon([[[(0, 0), (2, 2),
(1, 3)]], [[(2, 0), (4, 2),
(3, 3)]], [[(2, -2), (1, 0),
(-1, -1)]]],
crs=LonLatWGS84)
g.to_shapefile(os.path.join(TESTDIR, "data/multipoly"))
for fnm in ("multipoly.shx", "multipoly.shx", "multipoly.dbf",
"multipoly.prj"):
self.assertTrue(os.path.isfile(os.path.join(TESTDIR, "data", fnm)))
def test_write_collection_multipoint(self):
mp = Multipoint([p.vertex for p in self.points])
mp0 = copy(mp)
mp1 = copy(mp.shift((4, 2)))
mp2 = copy(mp.shift((-2, 3)))
shp.write_shapefile(os.path.join(TESTDIR, "data/mp_collection.shp"),
mp0, mp1, mp2)
for fnm in ("mp_collection.shx", "mp_collection.shx",
"mp_collection.dbf", "mp_collection.prj"):
self.assertTrue(os.path.isfile(os.path.join(TESTDIR, "data", fnm)))
def test_write_collection_lines(self):
line0 = copy(self.line)
line1 = copy(self.line.shift((4, 2)))
line2 = copy(self.line.shift((-2, 3)))
shp.write_shapefile(os.path.join(TESTDIR, "data/line_collection.shp"),
line0, line1, line2)
for fnm in ("line_collection.shx", "line_collection.shx",
"line_collection.dbf", "line_collection.prj"):
self.assertTrue(os.path.isfile(os.path.join(TESTDIR, "data", fnm)))
def test_read_points(self):
points = read_shapefile(os.path.join(TESTDATA, "shapefile", "points"))
self.assertEqual(len(points), 4)
pt = points[0]
self.assertTrue("+proj=lonlat" in pt.crs.get_proj4())
self.assertTrue("+a=6378137.0" in pt.crs.get_proj4())
self.assertTrue("+f=0.00335281" in pt.crs.get_proj4())
mp = Multipoint(points)
self.assertEqual(
mp.d["species"],
['T. officianale', 'C. tectorum', 'M. alba', 'V. cracca'])
self.assertEqual(mp.d["ID"], ['0', '1', '2', '3'])
x, y = mp.coordinates
self.assertTrue(np.all(x == np.array((1.0, 3.0, 4.0, 2.0))))
self.assertTrue(np.all(y == np.array((1.0, 1.0, 3.0, 2.0))))
def test_read_line(self):
line = read_shapefile(os.path.join(TESTDATA, "shapefile", "line"))[0]
self.assertTrue("+proj=lonlat" in line.crs.get_proj4())
self.assertTrue("+a=6378137.0" in line.crs.get_proj4())
self.assertTrue("+f=0.00335281" in line.crs.get_proj4())
x, y = line.coordinates
self.assertTrue(np.all(x == np.array([1.0, 5.0, 5.0, 3.0, 1.0])))
self.assertTrue(np.all(y == np.array([5.0, 5.0, 1.0, 3.0, 1.0])))
def test_read_polygon(self):
polygon = read_shapefile(os.path.join(TESTDATA, "shapefile",
"polygon"))[0]
self.assertTrue("+proj=lonlat" in polygon.crs.get_proj4())
self.assertTrue("+a=6378137.0" in polygon.crs.get_proj4())
self.assertTrue("+f=0.00335281" in polygon.crs.get_proj4())
x, y = polygon.coordinates
self.assertTrue(np.all(x == np.array([1.0, 5.0, 5.0, 3.0, 1.0])))
self.assertTrue(np.all(y == np.array([5.0, 5.0, 1.0, 3.0, 1.0])))
def test_read_points_newp(self):
# Read a multipoint with a projected cooridnate system
newp = read_shapefile(
os.path.join(TESTDATA, "shapefile", "newp_nsidc_north"))
proj4 = ('+proj=stere +lat_0=90 +lat_ts=70 +lon_0=-45 +k=1 +x_0=0 '
'+y_0=0 +a=6378273 +b=6356889.449 +units=m +no_defs')
for part in proj4.split():
self.assertTrue(part[:8] in newp[0].crs.get_proj4())
coords = list(zip(*[pt.vertex[:2] for pt in newp]))
self.assertEqual(
coords,
[(521236.8297444395, 521236.8297444395, 521236.8297444395,
547490.4452879033, 547490.4452879033, 547490.4452879033,
587584.1578033275, 587584.1578033275, 587584.1578033275,
571828.4918982167, 571828.4918982167),
(-888853.1384770898, -888853.1384770898, -888853.1384770898,
-902049.3617542256, -902049.3617542256, -902049.3617542256,
-871214.0673764511, -871214.0673764511, -871214.0673764511,
-850080.914674058, -850080.914674058)])
meterno = [pt.properties["meterno"] for pt in newp]
self.assertEqual(meterno, [
'IMS1/1', 'IMS2/1', '5952/2', 'IMS4/1', '5953/2', '1963/13',
'IMS5/1', '5213/A', '2121/13', 'IMS3/1', '3613/2'
])
depth = [pt.properties["depth_m"] for pt in newp]
self.assertEqual(depth, [
'73', '143', '247', '86', '147', '250', '74', '142', '235', '150',
'248'
])
