import os import mapnik import pytest try: import json except ImportError: import simplejson as json from .utilities import execution_path @pytest.fixture(scope="module") def setup(): # All of the paths used are relative, if we run the tests # from another directory we need to chdir() os.chdir(execution_path('.')) yield if 'ogr' in mapnik.DatasourceCache.plugin_names(): # Shapefile initialization def test_shapefile_init(setup): if not os.path.exists('../data/shp/boundaries.shp'): pytest.skip('Missing data') ds = mapnik.Ogr(file='../data/shp/boundaries.shp', layer_by_index=0) e = ds.envelope() assert e.minx == pytest.approx(-11121.6896651, abs=1e-7) assert e.miny == pytest.approx(-724724.216526, abs=1e-6) assert e.maxx == pytest.approx(2463000.67866, abs=1e-5) assert e.maxy == pytest.approx(1649661.267, abs=1e-3) meta = ds.describe() assert meta['geometry_type'] == mapnik.DataGeometryType.Polygon assert '+proj=lcc' in meta['proj4'] # Shapefile properties def test_shapefile_properties(): if not os.path.exists('../data/shp/boundaries.shp'): pytest.skip('Missing data') ds = mapnik.Ogr(file='../data/shp/boundaries.shp', layer_by_index=0) f = list(ds.features_at_point(ds.envelope().center(), 0.001))[0] assert ds.geometry_type() == mapnik.DataGeometryType.Polygon assert f['CGNS_FID'] == u'6f733341ba2011d892e2080020a0f4c9' assert f['COUNTRY'] == u'CAN' assert f['F_CODE'] == u'FA001' assert f['NAME_EN'] == u'Quebec' assert f['Shape_Area'] == 1512185733150.0 assert f['Shape_Leng'] == 19218883.724300001 meta = ds.describe() assert meta['geometry_type'] == mapnik.DataGeometryType.Polygon assert f['NOM_FR'] == u'Qu\xe9bec' assert f['NOM_FR'] == u'Québec' def test_that_nonexistant_query_field_throws(**kwargs): if not os.path.exists('../data/shp/world_merc.shp'): pytest.skip('Missing data') with pytest.raises(RuntimeError): ds = mapnik.Ogr(file='../data/shp/world_merc.shp', layer_by_index=0) assert len(ds.fields()) == 11 assert ds.fields() == ['FIPS', 'ISO2', 'ISO3', 'UN', 'NAME', 'AREA', 'POP2005', 'REGION', 'SUBREGION', 'LON', 'LAT'] assert ds.field_types() == ['str','str','str','int','str','int','int','int','int','float','float'] query = mapnik.Query(ds.envelope()) for fld in ds.fields(): query.add_property_name(fld) # also add an invalid one, triggering throw query.add_property_name('bogus') ds.features(query) # disabled because OGR prints an annoying error: ERROR 1: Invalid Point object. Missing 'coordinates' member. # def test_handling_of_null_features(): # ds = mapnik.Ogr(file='../data/json/null_feature.geojson',layer_by_index=0) # fs = ds.all_features() # assert len(list(fs)) == 1 # OGR plugin extent parameter def test_ogr_extent_parameter(): if not os.path.exists('../data/shp/world_merc.shp'): pytest.skip('Missing data') ds = mapnik.Ogr( file='../data/shp/world_merc.shp', layer_by_index=0, extent='-1,-1,1,1') e = ds.envelope() assert e.minx == -1 assert e.miny == -1 assert e.maxx == 1 assert e.maxy == 1 meta = ds.describe() assert meta['geometry_type'] == mapnik.DataGeometryType.Polygon assert '+proj=merc' in meta['proj4'] def test_ogr_reading_gpx_waypoint(): if not os.path.exists('../data/gpx/empty.gpx'): pytest.skip('Missing data') ds = mapnik.Ogr(file='../data/gpx/empty.gpx', layer='waypoints') e = ds.envelope() assert e.minx == -122 assert e.miny == 48 assert e.maxx == -122 assert e.maxy == 48 meta = ds.describe() assert meta['geometry_type'] == mapnik.DataGeometryType.Point assert '+proj=longlat' in meta['proj4'] def test_ogr_empty_data_should_not_throw(): if not os.path.exists('../data/gpx/empty.gpx'): pytest.skip('Missing data') default_logging_severity = mapnik.logger.get_severity() mapnik.logger.set_severity(getattr(mapnik.severity_type, "None")) # use logger to silence expected warnings for layer in ['routes', 'tracks', 'route_points', 'track_points']: ds = mapnik.Ogr(file='../data/gpx/empty.gpx', layer=layer) e = ds.envelope() assert e.minx == 0 assert e.miny == 0 assert e.maxx == 0 assert e.maxy == 0 mapnik.logger.set_severity(default_logging_severity) meta = ds.describe() assert meta['geometry_type'] == mapnik.DataGeometryType.Point assert '+proj=longlat' in meta['proj4'] # disabled because OGR prints an annoying error: ERROR 1: Invalid Point object. Missing 'coordinates' member. def test_handling_of_null_features(): if not os.path.exists('../data/json/null_feature.geojson'): pytest.skip('Missing data') assert True ds = mapnik.Ogr(file='../data/json/null_feature.geojson',layer_by_index=0) fs = ds.all_features() assert len(list(fs)) == 1 def test_geometry_type(): if not os.path.exists('../data/csv/wkt.csv'): pytest.skip('Missing data') ds = mapnik.Ogr(file='../data/csv/wkt.csv', layer_by_index=0) e = ds.envelope() assert e.minx == pytest.approx(1.0, abs=1e-1) assert e.miny == pytest.approx(1.0, abs=1e-1) assert e.maxx == pytest.approx(45.0, abs=1e-1) assert e.maxy == pytest.approx(45.0, abs=1e-1) meta = ds.describe() assert meta['geometry_type'] == mapnik.DataGeometryType.Point fs = ds.featureset() feat = fs.next() actual = json.loads(feat.to_geojson()) assert actual == {u'geometry': {u'type': u'Point', u'coordinates': [30,10]}, u'type': u'Feature', u'id': 2, u'properties': {u'type': u'point', u'WKT': u' POINT (30 10)'}} feat = fs.next() actual = json.loads(feat.to_geojson()) assert actual == {u'geometry': {u'type': u'LineString', u'coordinates': [[30,10],[10,30],[40,40]]}, u'type': u'Feature', u'id': 3, u'properties': {u'type': u'linestring', u'WKT': u' LINESTRING (30 10, 10 30, 40 40)'}} feat = fs.next() actual = json.loads(feat.to_geojson()) assert actual == {u'geometry': {u'type': u'Polygon', u'coordinates': [[[30,10],[40,40],[20,40],[10,20],[30,10]]]}, u'type': u'Feature', u'id': 4, u'properties': {u'type': u'polygon', u'WKT': u' POLYGON ((30 10, 10 20, 20 40, 40 40, 30 10))'}} feat = fs.next() actual = json.loads(feat.to_geojson()) assert actual == {u'geometry': {u'type': u'Polygon', u'coordinates': [[[35, 10],[45,45],[15,40],[10,20],[35,10]],[[20,30],[35,35],[30,20],[20,30]]]}, u'type': u'Feature', u'id': 5, u'properties': { u'type': u'polygon', u'WKT': u' POLYGON ((35 10, 10 20, 15 40, 45 45, 35 10),(20 30, 35 35, 30 20, 20 30))'}} feat = fs.next() actual = json.loads(feat.to_geojson()) assert actual == {u'geometry': {u'type': u'MultiPoint', u'coordinates': [[10,40],[40,30],[20,20],[30,10]]}, u'type': u'Feature', u'id': 6, u'properties': {u'type': u'multipoint', u'WKT': u' MULTIPOINT ((10 40), (40 30), (20 20), (30 10))'}} feat = fs.next() actual = json.loads(feat.to_geojson()) assert actual == {u'geometry': {u'type': u'MultiLineString', u'coordinates': [[[10,10],[20,20],[10,40]],[[40,40],[30,30],[40,20],[30,10]]]}, u'type': u'Feature', u'id': 7, u'properties': {u'type': u'multilinestring', u'WKT': u' MULTILINESTRING ((10 10, 20 20, 10 40),(40 40, 30 30, 40 20, 30 10))'}} feat = fs.next() actual = json.loads(feat.to_geojson()) assert actual == {u'geometry': {u'type': u'MultiPolygon', u'coordinates': [[[[30,20],[45,40],[10,40],[30,20]]],[[[15,5],[40,10],[10,20],[5,10],[15,5]]]]}, u'type': u'Feature', u'id': 8, u'properties': {u'type': u'multipolygon', u'WKT': u' MULTIPOLYGON (((30 20, 10 40, 45 40, 30 20)),((15 5, 40 10, 10 20, 5 10, 15 5)))'}} feat = fs.next() actual = json.loads(feat.to_geojson()) assert actual == {u'geometry': {u'type': u'MultiPolygon', u'coordinates': [[[[40, 40], [20, 45], [45, 30], [40, 40]]], [[[20, 35], [10, 30], [10, 10], [30, 5], [45, 20], [20, 35]], [[30, 20], [20, 15], [20, 25], [30, 20]]]]}, u'type': u'Feature', u'id': 9, u'properties': {u'type': u'multipolygon', u'WKT': u' MULTIPOLYGON (((40 40, 20 45, 45 30, 40 40)),((20 35, 45 20, 30 5, 10 10, 10 30, 20 35),(30 20, 20 25, 20 15, 30 20)))'}} feat = fs.next() actual = json.loads(feat.to_geojson()) assert actual == {u'geometry': {u'type': u'GeometryCollection', u'geometries': [{u'type': u'Polygon', u'coordinates': [[[1, 1],[2,1],[2,2],[1,2],[1,1]]]}, {u'type': u'Point', u'coordinates': [2,3]}, {u'type': u'LineString', u'coordinates': [[2,3],[3,4]]}]}, u'type': u'Feature', u'id': 10, u'properties': {u'type': u'collection', u'WKT': u' GEOMETRYCOLLECTION(POLYGON((1 1,2 1,2 2,1 2,1 1)),POINT(2 3),LINESTRING(2 3,3 4))'}}