File: pdf_printing_test.py

package info (click to toggle)
python-mapnik 1%3A0.0~20240222-5ab32f020-4
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 11,364 kB
  • sloc: python: 11,685; cpp: 5,776; sh: 242; makefile: 10
file content (53 lines) | stat: -rw-r--r-- 1,927 bytes parent folder | download
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
import mapnik
import os
import pytest
from .utilities import execution_path

@pytest.fixture(scope="module")
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('.'))
    yield

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(setup):
                if not os.path.exists('../data/good_maps/marker-text-line.xml'):
                    pytest.skip('Missing data')

                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)
                assert diff < 1500, msg

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