mapnik/tests/python_tests/webp_encoding_test.py

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

import sys
import os, mapnik
from timeit import Timer, time
from nose.tools import *
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('.'))

if mapnik.has_webp():
    tmp_dir = '/tmp/mapnik-webp/'
    if not os.path.exists(tmp_dir):
       os.makedirs(tmp_dir)

    opts = [
    'webp',
    'webp:quality=64',
    'webp:alpha=false'
    ]


    def gen_filepath(name,format):
        return os.path.join('images/support/encoding-opts',name+'-'+format.replace(":","+")+'.webp')

    def test_quality_threshold():
        im = mapnik.Image(256,256)
        im.tostring('webp:quality=99.99000')
        im.tostring('webp:quality=0')
        im.tostring('webp:quality=0.001')

    @raises(RuntimeError)
    def test_quality_threshold_invalid():
        im = mapnik.Image(256,256)
        im.tostring('webp:quality=101')

    @raises(RuntimeError)
    def test_quality_threshold_invalid2():
        im = mapnik.Image(256,256)
        im.tostring('webp:quality=-1')

    generate = False

    def test_expected_encodings():
        fails = []
        for opt in opts:
            im = mapnik.Image(256,256)
            expected = gen_filepath('blank',opt)
            actual = os.path.join(tmp_dir,os.path.basename(expected))
            if generate or not os.path.exists(expected):
                print 'generating expected image %s' % expected
                im.save(expected,opt)
            im.save(actual,opt)
            if mapnik.Image.open(actual).tostring() != mapnik.Image.open(expected).tostring():
                fails.append('%s (actual) not == to %s (expected)' % (actual,expected))

        for opt in opts:
            im = mapnik.Image(256,256)
            im.background = mapnik.Color('green')
            expected = gen_filepath('solid',opt)
            actual = os.path.join(tmp_dir,os.path.basename(expected))
            if generate or not os.path.exists(expected):
                print 'generating expected image %s' % expected
                im.save(expected,opt)
            im.save(actual,opt)
            if mapnik.Image.open(actual).tostring() != mapnik.Image.open(expected).tostring():
                fails.append('%s (actual) not == to %s (expected)' % (actual,expected))

        for opt in opts:
            im = mapnik.Image.open('images/support/transparency/aerial_rgba.png')
            expected = gen_filepath('aerial_rgba',opt)
            actual = os.path.join(tmp_dir,os.path.basename(expected))
            if generate or not os.path.exists(expected):
                print 'generating expected image %s' % expected
                im.save(expected,opt)
            im.save(actual,opt)
            if mapnik.Image.open(actual).tostring() != mapnik.Image.open(expected).tostring():
                fails.append('%s (actual) not == to %s (expected)' % (actual,expected))
        eq_(fails,[],'\n'+'\n'.join(fails))

    def test_transparency_levels():
        # create partial transparency image
        im = mapnik.Image(256,256)
        im.background = mapnik.Color('rgba(255,255,255,.5)')
        c2 = mapnik.Color('rgba(255,255,0,.2)')
        c3 = mapnik.Color('rgb(0,255,255)')
        for y in range(0,im.height()/2):
            for x in range(0,im.width()/2):
                im.set_pixel(x,y,c2)
        for y in range(im.height()/2,im.height()):
            for x in range(im.width()/2,im.width()):
                im.set_pixel(x,y,c3)

        t0 = tmp_dir + 'white0.webp'

        # octree
        format = 'webp'
        expected = 'images/support/transparency/white0.webp'
        if generate or not os.path.exists(expected):
            im.save('images/support/transparency/white0.webp')
        im.save(t0,format)
        im_in = mapnik.Image.open(t0)
        t0_len = len(im_in.tostring(format))
        eq_(t0_len,len(mapnik.Image.open(expected).tostring(format)))


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