93 lines
2.9 KiB
Python
93 lines
2.9 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import os, mapnik
|
|
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('.'))
|
|
|
|
def test_image_16_8_simple():
|
|
im = mapnik.Image(2,2,mapnik.ImageType.gray16)
|
|
im.set_pixel(0,0, 256)
|
|
im.set_pixel(0,1, 999)
|
|
im.set_pixel(1,0, 5)
|
|
im.set_pixel(1,1, 2)
|
|
im2 = im.cast(mapnik.ImageType.gray8)
|
|
eq_(im2.get_pixel(0,0), 255)
|
|
eq_(im2.get_pixel(0,1), 255)
|
|
eq_(im2.get_pixel(1,0), 5)
|
|
eq_(im2.get_pixel(1,1), 2)
|
|
# Cast back!
|
|
im = im2.cast(mapnik.ImageType.gray16)
|
|
eq_(im.get_pixel(0,0), 255)
|
|
eq_(im.get_pixel(0,1), 255)
|
|
eq_(im.get_pixel(1,0), 5)
|
|
eq_(im.get_pixel(1,1), 2)
|
|
|
|
def test_image_32f_8_simple():
|
|
im = mapnik.Image(2,2,mapnik.ImageType.gray32f)
|
|
im.set_pixel(0,0, 120.1234)
|
|
im.set_pixel(0,1, -23.4)
|
|
im.set_pixel(1,0, 120.6)
|
|
im.set_pixel(1,1, 360.2)
|
|
im2 = im.cast(mapnik.ImageType.gray8)
|
|
eq_(im2.get_pixel(0,0), 120)
|
|
eq_(im2.get_pixel(0,1), 0)
|
|
eq_(im2.get_pixel(1,0), 120) # Notice this is truncated!
|
|
eq_(im2.get_pixel(1,1), 255)
|
|
|
|
def test_image_offset_and_scale():
|
|
im = mapnik.Image(2,2,mapnik.ImageType.gray16)
|
|
eq_(im.offset, 0.0)
|
|
eq_(im.scaling, 1.0)
|
|
im.offset = 1.0
|
|
im.scaling = 2.0
|
|
eq_(im.offset, 1.0)
|
|
eq_(im.scaling, 2.0)
|
|
|
|
def test_image_16_8_scale_and_offset():
|
|
im = mapnik.Image(2,2,mapnik.ImageType.gray16)
|
|
im.set_pixel(0,0, 256)
|
|
im.set_pixel(0,1, 258)
|
|
im.set_pixel(1,0, 99999)
|
|
im.set_pixel(1,1, 615)
|
|
offset = 255
|
|
scaling = 3
|
|
im2 = im.cast(mapnik.ImageType.gray8, offset, scaling)
|
|
eq_(im2.get_pixel(0,0), 0)
|
|
eq_(im2.get_pixel(0,1), 1)
|
|
eq_(im2.get_pixel(1,0), 255)
|
|
eq_(im2.get_pixel(1,1), 120)
|
|
# pixels will be a little off due to offsets in reverting!
|
|
im3 = im2.cast(mapnik.ImageType.gray16)
|
|
eq_(im3.get_pixel(0,0), 255) # Rounding error with ints
|
|
eq_(im3.get_pixel(0,1), 258) # same
|
|
eq_(im3.get_pixel(1,0), 1020) # The other one was way out of range for our scale/offset
|
|
eq_(im3.get_pixel(1,1), 615) # same
|
|
|
|
def test_image_16_32f_scale_and_offset():
|
|
im = mapnik.Image(2,2,mapnik.ImageType.gray16)
|
|
im.set_pixel(0,0, 256)
|
|
im.set_pixel(0,1, 258)
|
|
im.set_pixel(1,0, 0)
|
|
im.set_pixel(1,1, 615)
|
|
offset = 255
|
|
scaling = 3.2
|
|
im2 = im.cast(mapnik.ImageType.gray32f, offset, scaling)
|
|
eq_(im2.get_pixel(0,0), 0.3125)
|
|
eq_(im2.get_pixel(0,1), 0.9375)
|
|
eq_(im2.get_pixel(1,0), -79.6875)
|
|
eq_(im2.get_pixel(1,1), 112.5)
|
|
im3 = im2.cast(mapnik.ImageType.gray16)
|
|
eq_(im3.get_pixel(0,0), 256)
|
|
eq_(im3.get_pixel(0,1), 258)
|
|
eq_(im3.get_pixel(1,0), 0)
|
|
eq_(im3.get_pixel(1,1), 615)
|
|
|
|
if __name__ == "__main__":
|
|
setup()
|
|
exit(run_all(eval(x) for x in dir() if x.startswith("test_")))
|