~ubuntu-branches/ubuntu/wily/python-imaging/wily

« back to all changes in this revision

Viewing changes to Tests/test_image_split.py

  • Committer: Package Import Robot
  • Author(s): Matthias Klose
  • Date: 2013-01-31 20:49:20 UTC
  • mfrom: (27.1.1 raring-proposed)
  • Revision ID: package-import@ubuntu.com-20130131204920-b5zshy6vgfvdionl
Tags: 1.1.7+1.7.8-1ubuntu1
Rewrite build dependencies to allow cross builds.

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
from tester import *
 
2
 
 
3
from PIL import Image
 
4
 
 
5
def test_split():
 
6
    def split(mode):
 
7
        layers = lena(mode).split()
 
8
        return [(i.mode, i.size[0], i.size[1]) for i in layers]
 
9
    assert_equal(split("1"), [('1', 128, 128)])
 
10
    assert_equal(split("L"), [('L', 128, 128)])
 
11
    assert_equal(split("I"), [('I', 128, 128)])
 
12
    assert_equal(split("F"), [('F', 128, 128)])
 
13
    assert_equal(split("P"), [('P', 128, 128)])
 
14
    assert_equal(split("RGB"), [('L', 128, 128), ('L', 128, 128), ('L', 128, 128)])
 
15
    assert_equal(split("RGBA"), [('L', 128, 128), ('L', 128, 128), ('L', 128, 128), ('L', 128, 128)])
 
16
    assert_equal(split("CMYK"), [('L', 128, 128), ('L', 128, 128), ('L', 128, 128), ('L', 128, 128)])
 
17
    assert_equal(split("YCbCr"), [('L', 128, 128), ('L', 128, 128), ('L', 128, 128)])
 
18
 
 
19
def test_split_merge():
 
20
    def split_merge(mode):
 
21
        return Image.merge(mode, lena(mode).split())
 
22
    assert_image_equal(lena("1"), split_merge("1"))
 
23
    assert_image_equal(lena("L"), split_merge("L"))
 
24
    assert_image_equal(lena("I"), split_merge("I"))
 
25
    assert_image_equal(lena("F"), split_merge("F"))
 
26
    assert_image_equal(lena("P"), split_merge("P"))
 
27
    assert_image_equal(lena("RGB"), split_merge("RGB"))
 
28
    assert_image_equal(lena("RGBA"), split_merge("RGBA"))
 
29
    assert_image_equal(lena("CMYK"), split_merge("CMYK"))
 
30
    assert_image_equal(lena("YCbCr"), split_merge("YCbCr"))
 
31
 
 
32
def test_split_open():
 
33
    file = tempfile("temp.png")
 
34
    def split_open(mode):
 
35
        lena(mode).save(file)
 
36
        im = Image.open(file)
 
37
        return len(im.split())
 
38
    assert_equal(split_open("1"), 1)
 
39
    assert_equal(split_open("L"), 1)
 
40
    assert_equal(split_open("P"), 1)
 
41
    assert_equal(split_open("RGB"), 3)
 
42
    assert_equal(split_open("RGBA"), 4)