mapnik/tests/python_tests/datasource_test.py

#!/usr/bin/env python

from nose.tools import *
from utilities import execution_path

import os, mapnik2

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('.'))
    
def test_field_listing():
    lyr = mapnik2.Layer('test')
    lyr.datasource = mapnik2.Shapefile(file='../data/shp/poly.shp')
    fields = lyr.datasource.fields()
    eq_(fields, ['AREA', 'EAS_ID', 'PRFEDEA'])

def test_total_feature_count_shp():
    lyr = mapnik2.Layer('test')
    lyr.datasource = mapnik2.Shapefile(file='../data/shp/poly.shp')
    features = lyr.datasource.all_features()
    num_feats = len(features)
    eq_(num_feats, 10)

def test_total_feature_count_json():
    lyr = mapnik2.Layer('test')
    lyr.datasource = mapnik2.Ogr(file='../data/json/points.json',layer_by_index=0)
    features = lyr.datasource.all_features()
    num_feats = len(features)
    eq_(num_feats, 5)

def test_reading_json_from_string():
    json = open('../data/json/points.json','r').read()
    lyr = mapnik2.Layer('test')
    lyr.datasource = mapnik2.Ogr(file=json,layer_by_index=0)
    features = lyr.datasource.all_features()
    num_feats = len(features)
    eq_(num_feats, 5)
    
def test_feature_envelope():
    lyr = mapnik2.Layer('test')
    lyr.datasource = mapnik2.Shapefile(file='../data/shp/poly.shp')
    features = lyr.datasource.all_features()
    for feat in features:
        env = feat.envelope()
        contains = lyr.envelope().contains(env)
        eq_(contains, True)
        intersects = lyr.envelope().contains(env)
        eq_(intersects, True)

def test_feature_attributes():
    lyr = mapnik2.Layer('test')
    lyr.datasource = mapnik2.Shapefile(file='../data/shp/poly.shp')
    features = lyr.datasource.all_features()
    feat = features[0]
    attrs = {'PRFEDEA': u'35043411', 'EAS_ID': 168, 'AREA': 215229.266}
    eq_(feat.attributes, attrs)
    eq_(lyr.datasource.fields(),['AREA', 'EAS_ID', 'PRFEDEA'])
    eq_(lyr.datasource.field_types(),[float,int,str])