mapnik/tests/visual_tests/compare.py

# -*- coding: utf-8 -*-

import mapnik
import os
from unittest import TestCase

try:
    import json
except ImportError:
    import simplejson as json

# compare two images and return number of different pixels
def compare(actual, expected, alpha=True):
    im1 = mapnik.Image.open(actual)
    im2 = mapnik.Image.open(expected)
    pixels = im1.width() * im1.height()
    delta_pixels = (im2.width() * im2.height()) - pixels
    if delta_pixels != 0:
        return delta_pixels
    return im1.compare(im2, 0, alpha)

def compare_grids(actual, expected, threshold=0, alpha=True):
    global errors
    global passed
    im1 = json.loads(open(actual).read())
    im2 = json.loads(open(expected).read())
    # TODO - real diffing
    if not im1['data'] == im2['data']:
        return 99999999
    if not im1['keys'] == im2['keys']:
        return 99999999
    grid1 = im1['grid']
    grid2 = im2['grid']
    try:
        assertSequenceEqual(grid1, grid2)
        return 0
    except:
        # dimensions must be exact
        width1 = len(grid1[0])
        width2 = len(grid2[0])
        if not width1 == width2:
            return 99999999
        height1 = len(grid1)
        height2 = len(grid2)
        if not height1 == height2:
            return 99999999
        diff = 0;
        for y in range(0,height1-1):
            row1 = grid1[y]
            row2 = grid2[y]
            if row1 == row2:
                continue;
            #width = min(len(row1),len(row2))
            for w in range(0,width1):
                if row1[w] != row2[w]:
                    diff += 1
        return diff