mapnik/tests/python_tests/python_plugin_test.py

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

# import os
# import math
# import mapnik
# import sys
# from utilities import execution_path, run_all
# from nose.tools import *

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

# class PointDatasource(mapnik.PythonDatasource):
#     def __init__(self):
#         super(PointDatasource, self).__init__(
#                 geometry_type = mapnik.DataGeometryType.Point,
#                 envelope = mapnik.Box2d(0,-10,100,110),
#                 data_type = mapnik.DataType.Vector
#         )

#     def features(self, query):
#         return mapnik.PythonDatasource.wkt_features(
#             keys = ('label',),
#             features = (
#                 ( 'POINT (5 6)', { 'label': 'foo-bar'} ),
#                 ( 'POINT (60 50)', { 'label': 'buzz-quux'} ),
#             )
#         )

# class ConcentricCircles(object):
#     def __init__(self, centre, bounds, step=1):
#         self.centre = centre
#         self.bounds = bounds
#         self.step = step

#     class Iterator(object):
#         def __init__(self, container):
#             self.container = container

#             centre = self.container.centre
#             bounds = self.container.bounds
#             step = self.container.step

#             self.radius = step

#         def next(self):
#             points = []
#             for alpha in xrange(0, 361, 5):
#                 x = math.sin(math.radians(alpha)) * self.radius + self.container.centre[0]
#                 y = math.cos(math.radians(alpha)) * self.radius + self.container.centre[1]
#                 points.append('%s %s' % (x,y))
#             circle = 'POLYGON ((' + ','.join(points) + '))'

#             # has the circle grown so large that the boundary is entirely within it?
#             tl = (self.container.bounds.maxx, self.container.bounds.maxy)
#             tr = (self.container.bounds.maxx, self.container.bounds.maxy)
#             bl = (self.container.bounds.minx, self.container.bounds.miny)
#             br = (self.container.bounds.minx, self.container.bounds.miny)
#             def within_circle(p):
#                 delta_x = p[0] - self.container.centre[0]
#                 delta_y = p[0] - self.container.centre[0]
#                 return delta_x*delta_x + delta_y*delta_y < self.radius*self.radius

#             if all(within_circle(p) for p in (tl,tr,bl,br)):
#                 raise StopIteration()

#             self.radius += self.container.step
#             return ( circle, { } )

#     def __iter__(self):
#         return ConcentricCircles.Iterator(self)

# class CirclesDatasource(mapnik.PythonDatasource):
#     def __init__(self, centre_x=-20, centre_y=0, step=10):
#         super(CirclesDatasource, self).__init__(
#                 geometry_type = mapnik.DataGeometryType.Polygon,
#                 envelope = mapnik.Box2d(-180, -90, 180, 90),
#                 data_type = mapnik.DataType.Vector
#         )

#         # note that the plugin loader will set all arguments to strings and will not try to parse them
#         centre_x = int(centre_x)
#         centre_y = int(centre_y)
#         step = int(step)

#         self.centre_x = centre_x
#         self.centre_y = centre_y
#         self.step = step

#     def features(self, query):
#         centre = (self.centre_x, self.centre_y)

#         return mapnik.PythonDatasource.wkt_features(
#             keys = (),
#             features = ConcentricCircles(centre, query.bbox, self.step)
#         )

# if 'python' in mapnik.DatasourceCache.plugin_names():
#     # make sure we can load from ourself as a module
#     sys.path.append(execution_path('.'))

#     def test_python_point_init():
#         ds = mapnik.Python(factory='python_plugin_test:PointDatasource')
#         e = ds.envelope()

#         assert_almost_equal(e.minx, 0, places=7)
#         assert_almost_equal(e.miny, -10, places=7)
#         assert_almost_equal(e.maxx, 100, places=7)
#         assert_almost_equal(e.maxy, 110, places=7)

#     def test_python_circle_init():
#         ds = mapnik.Python(factory='python_plugin_test:CirclesDatasource')
#         e = ds.envelope()

#         assert_almost_equal(e.minx, -180, places=7)
#         assert_almost_equal(e.miny, -90, places=7)
#         assert_almost_equal(e.maxx, 180, places=7)
#         assert_almost_equal(e.maxy, 90, places=7)

#     def test_python_circle_init_with_args():
#         ds = mapnik.Python(factory='python_plugin_test:CirclesDatasource', centre_x=40, centre_y=7)
#         e = ds.envelope()

#         assert_almost_equal(e.minx, -180, places=7)
#         assert_almost_equal(e.miny, -90, places=7)
#         assert_almost_equal(e.maxx, 180, places=7)
#         assert_almost_equal(e.maxy, 90, places=7)

#     def test_python_point_rendering():
#         m = mapnik.Map(512,512)
#         mapnik.load_map(m,'../data/python_plugin/python_point_datasource.xml')
#         m.zoom_all()
#         im = mapnik.Image(512,512)
#         mapnik.render(m,im)
#         actual = '/tmp/mapnik-python-point-render1.png'
#         expected = 'images/support/mapnik-python-point-render1.png'
#         im.save(actual)
#         expected_im = mapnik.Image.open(expected)
#         eq_(im.tostring('png32'),expected_im.tostring('png32'),
#                 'failed comparing actual (%s) and expected (%s)' % (actual,'tests/python_tests/'+ expected))

#     def test_python_circle_rendering():
#         m = mapnik.Map(512,512)
#         mapnik.load_map(m,'../data/python_plugin/python_circle_datasource.xml')
#         m.zoom_all()
#         im = mapnik.Image(512,512)
#         mapnik.render(m,im)
#         actual = '/tmp/mapnik-python-circle-render1.png'
#         expected = 'images/support/mapnik-python-circle-render1.png'
#         im.save(actual)
#         expected_im = mapnik.Image.open(expected)
#         eq_(im.tostring('png32'),expected_im.tostring('png32'),
#                 'failed comparing actual (%s) and expected (%s)' % (actual,'tests/python_tests/'+ expected))

# if __name__ == "__main__":
#     setup()
#     run_all(eval(x) for x in dir() if x.startswith("test_"))