mapnik/tests/python_tests/osm_test.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

from nose.tools import *
from utilities import execution_path, Todo

import os, mapnik

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('.'))

if 'osm' in mapnik.DatasourceCache.instance().plugin_names():

    # Shapefile initialization
    def test_osm_init():
        ds = mapnik.Osm(file='../data/osm/nodes.osm')

        e = ds.envelope()

        # these are hardcoded in the plugin… ugh
        eq_(e.minx >= -180.0,True)
        eq_(e.miny >= -90.0,True)
        eq_(e.maxx <= 180.0,True)
        eq_(e.maxy <= 90,True)

    @raises(RuntimeError)
    def test_that_nonexistant_query_field_throws(**kwargs):
        raise Todo("fixme")
        ds = mapnik.Osm(file='../data/osm/nodes.osm')
        # ugh, more odd stuff hardcoded...
        eq_(len(ds.fields()),5)
        eq_(ds.fields(),['bicycle', 'foot', 'horse', 'name', 'width'])
        eq_(ds.field_types(),['str', 'str', 'str', 'str', 'str'])
        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')
        fs = ds.features(query)


if __name__ == "__main__":
    setup()
    [eval(run)() for run in dir() if 'test_' in run]