File: pdf_printing_test.py

package info (click to toggle)
python-mapnik 1%3A0.0~20200224-7da019cf9-3
  • links: PTS, VCS
  • area: main
  • in suites: bullseye
  • size: 10,784 kB
  • sloc: python: 12,085; cpp: 5,717; sh: 101; makefile: 18
file content (61 lines) | stat: -rw-r--r-- 1,848 bytes parent folder | download | duplicates (4)
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
#!/usr/bin/env python

import os

from nose.tools import eq_
from nose.plugins.skip import SkipTest

import mapnik
from .utilities import execution_path, run_all

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 make_map_from_xml(source_xml):
	m = mapnik.Map(100, 100)
	mapnik.load_map(m, source_xml, True)
	m.zoom_all()

	return m

def make_pdf(m, output_pdf, esri_wkt):
	# renders a PDF with a grid and a legend
	page = mapnik.printing.PDFPrinter(use_ocg_layers=True)

	page.render_map(m, output_pdf)
	page.render_grid_on_map(m)
	page.render_legend(m)

	page.finish()
	page.add_geospatial_pdf_header(m, output_pdf, wkt=esri_wkt)

if mapnik.has_pycairo():
	import mapnik.printing

	def test_pdf_printing():
                if not os.path.exists('../data/good_maps/marker-text-line.xml'):
                    raise SkipTest

		source_xml = '../data/good_maps/marker-text-line.xml'.encode('utf-8')
		m = make_map_from_xml(source_xml)

		actual_pdf = "/tmp/pdf-printing-actual.pdf"
		esri_wkt = 'GEOGCS["GCS_WGS_1984",DATUM["D_WGS_1984",SPHEROID["WGS_1984",6378137,298.257223563]],PRIMEM["Greenwich",0],UNIT["Degree",0.017453292519943295]]'
		make_pdf(m, actual_pdf, esri_wkt)

		expected_pdf = 'images/pycairo/pdf-printing-expected.pdf'

		diff = abs(os.stat(expected_pdf).st_size - os.stat(actual_pdf).st_size)
		msg = 'diff in size (%s) between actual (%s) and expected(%s)' % (diff, actual_pdf, 'tests/python_tests/' + expected_pdf)
		eq_(diff < 1500, True, msg)

# TODO: ideas for further testing on printing module
# - test with and without pangocairo
# - test legend with attribution
# - test graticule (bug at the moment)

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