1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69
|
#!/usr/bin/env python
# -*- coding: utf-8 -*-
from __future__ import print_function
import os
from glob import glob
import mapnik
from .utilities import execution_path, run_all
default_logging_severity = mapnik.logger.get_severity()
def setup():
mapnik.logger.set_severity(getattr(mapnik.severity_type, "None"))
# All of the paths used are relative, if we run the tests
# from another directory we need to chdir()
os.chdir(execution_path('.'))
def teardown():
mapnik.logger.set_severity(default_logging_severity)
plugin_mapping = {
'.csv': ['csv'],
'.json': ['geojson', 'ogr'],
'.tif': ['gdal'],
#'.tif' : ['gdal','raster'],
'.kml': ['ogr'],
'.gpx': ['ogr'],
'.vrt': ['gdal']
}
def test_opening_data():
# https://github.com/mapbox/mapnik-test-data
# cd tests/data
# git clone --depth 1 https://github.com/mapbox/mapnik-test-data
if os.path.exists('../data/mapnik-test-data/'):
files = glob('../data/mapnik-test-data/data/*/*.*')
for filepath in files:
ext = os.path.splitext(filepath)[1]
if plugin_mapping.get(ext):
# print 'testing opening %s' % filepath
if 'topo' in filepath:
kwargs = {'type': 'ogr', 'file': filepath}
kwargs['layer_by_index'] = 0
try:
mapnik.Datasource(**kwargs)
except Exception as e:
print('could not open, %s: %s' % (kwargs, e))
else:
for plugin in plugin_mapping[ext]:
kwargs = {'type': plugin, 'file': filepath}
if plugin == 'ogr':
kwargs['layer_by_index'] = 0
try:
mapnik.Datasource(**kwargs)
except Exception as e:
print('could not open, %s: %s' % (kwargs, e))
# else:
# print 'skipping opening %s' % filepath
if __name__ == "__main__":
setup()
exit(run_all(eval(x) for x in dir() if x.startswith("test_")))
|