Pillow/Tests/test_image_access.py

251 lines
7.5 KiB
Python
Raw Normal View History

2014-09-05 13:36:24 +04:00
from helper import unittest, PillowTestCase, hopper
2014-01-05 22:41:25 +04:00
try:
2014-06-10 13:10:47 +04:00
from PIL import PyAccess
2015-05-27 02:15:45 +03:00
except ImportError:
2014-06-10 13:10:47 +04:00
# Skip in setUp()
pass
2014-01-05 22:41:25 +04:00
2014-06-10 13:10:47 +04:00
from PIL import Image
2014-01-05 22:41:25 +04:00
class AccessTest(PillowTestCase):
# initial value
_init_cffi_access = Image.USE_CFFI_ACCESS
_need_cffi_access = False
@classmethod
def setUpClass(cls):
Image.USE_CFFI_ACCESS = cls._need_cffi_access
@classmethod
def tearDownClass(cls):
Image.USE_CFFI_ACCESS = cls._init_cffi_access
class TestImagePutPixel(AccessTest):
def test_sanity(self):
im1 = hopper()
im2 = Image.new(im1.mode, im1.size, 0)
for y in range(im1.size[1]):
for x in range(im1.size[0]):
pos = x, y
im2.putpixel(pos, im1.getpixel(pos))
self.assert_image_equal(im1, im2)
im2 = Image.new(im1.mode, im1.size, 0)
im2.readonly = 1
for y in range(im1.size[1]):
for x in range(im1.size[0]):
pos = x, y
im2.putpixel(pos, im1.getpixel(pos))
self.assertFalse(im2.readonly)
self.assert_image_equal(im1, im2)
im2 = Image.new(im1.mode, im1.size, 0)
pix1 = im1.load()
pix2 = im2.load()
for y in range(im1.size[1]):
for x in range(im1.size[0]):
pix2[x, y] = pix1[x, y]
self.assert_image_equal(im1, im2)
class TestImageGetPixel(AccessTest):
@staticmethod
def color(mode):
bands = Image.getmodebands(mode)
if bands == 1:
return 1
else:
return tuple(range(1, bands + 1))
def check(self, mode, c=None):
if not c:
c = self.color(mode)
# check putpixel
im = Image.new(mode, (1, 1), None)
im.putpixel((0, 0), c)
self.assertEqual(
im.getpixel((0, 0)), c,
"put/getpixel roundtrip failed for mode %s, color %s" % (mode, c))
# Check 0
im = Image.new(mode, (0, 0), None)
with self.assertRaises(IndexError):
im.putpixel((0, 0), c)
with self.assertRaises(IndexError):
im.getpixel((0, 0))
2016-10-02 13:31:53 +03:00
# check initial color
im = Image.new(mode, (1, 1), c)
self.assertEqual(
im.getpixel((0, 0)), c,
"initial color failed for mode %s, color %s " % (mode, c))
# Check 0
im = Image.new(mode, (0, 0), c)
with self.assertRaises(IndexError):
im.getpixel((0, 0))
def test_basic(self):
for mode in ("1", "L", "LA", "I", "I;16", "I;16B", "F",
"P", "PA", "RGB", "RGBA", "RGBX", "CMYK", "YCbCr"):
self.check(mode)
def test_signedness(self):
# see https://github.com/python-pillow/Pillow/issues/452
# pixelaccess is using signed int* instead of uint*
for mode in ("I;16", "I;16B"):
self.check(mode, 2**15-1)
self.check(mode, 2**15)
self.check(mode, 2**15+1)
self.check(mode, 2**16-1)
2014-01-05 22:41:25 +04:00
2014-06-10 13:10:47 +04:00
class TestCffiPutPixel(TestImagePutPixel):
_need_cffi_access = True
2014-06-10 13:10:47 +04:00
def setUp(self):
try:
import cffi
2016-08-04 09:40:12 +03:00
assert cffi # silence warning
2015-05-27 02:15:45 +03:00
except ImportError:
2014-06-10 13:10:47 +04:00
self.skipTest("No cffi")
class TestCffiGetPixel(TestImageGetPixel):
_need_cffi_access = True
2014-06-10 13:10:47 +04:00
def setUp(self):
try:
import cffi
2016-08-04 09:40:12 +03:00
assert cffi # silence warning
2015-05-27 02:15:45 +03:00
except ImportError:
2014-06-10 13:10:47 +04:00
self.skipTest("No cffi")
class TestCffi(AccessTest):
_need_cffi_access = True
2014-06-10 13:10:47 +04:00
def setUp(self):
try:
import cffi
2016-08-04 09:40:12 +03:00
assert cffi # silence warning
2015-05-27 02:15:45 +03:00
except ImportError:
2014-06-10 13:10:47 +04:00
self.skipTest("No cffi")
def _test_get_access(self, im):
2015-04-02 11:57:24 +03:00
"""Do we get the same thing as the old pixel access
2014-06-10 13:10:47 +04:00
2015-04-02 11:57:24 +03:00
Using private interfaces, forcing a capi access and
a pyaccess for the same image"""
2014-06-10 13:10:47 +04:00
caccess = im.im.pixel_access(False)
access = PyAccess.new(im, False)
w, h = im.size
for x in range(0, w, 10):
for y in range(0, h, 10):
self.assertEqual(access[(x, y)], caccess[(x, y)])
2015-07-03 08:03:25 +03:00
# Access an out-of-range pixel
self.assertRaises(ValueError,
lambda: access[(access.xsize+1, access.ysize+1)])
2014-06-10 13:10:47 +04:00
def test_get_vs_c(self):
2014-09-05 13:36:24 +04:00
rgb = hopper('RGB')
2014-06-10 13:10:47 +04:00
rgb.load()
self._test_get_access(rgb)
2014-09-05 13:36:24 +04:00
self._test_get_access(hopper('RGBA'))
self._test_get_access(hopper('L'))
self._test_get_access(hopper('LA'))
self._test_get_access(hopper('1'))
self._test_get_access(hopper('P'))
2015-07-03 09:22:56 +03:00
# self._test_get_access(hopper('PA')) # PA -- how do I make a PA image?
2014-09-05 13:36:24 +04:00
self._test_get_access(hopper('F'))
2014-06-10 13:10:47 +04:00
im = Image.new('I;16', (10, 10), 40000)
self._test_get_access(im)
im = Image.new('I;16L', (10, 10), 40000)
self._test_get_access(im)
im = Image.new('I;16B', (10, 10), 40000)
self._test_get_access(im)
im = Image.new('I', (10, 10), 40000)
self._test_get_access(im)
# These don't actually appear to be modes that I can actually make,
# as unpack sets them directly into the I mode.
# im = Image.new('I;32L', (10, 10), -2**10)
# self._test_get_access(im)
# im = Image.new('I;32B', (10, 10), 2**10)
# self._test_get_access(im)
def _test_set_access(self, im, color):
2015-04-02 11:57:24 +03:00
"""Are we writing the correct bits into the image?
2014-06-10 13:10:47 +04:00
2015-04-02 11:57:24 +03:00
Using private interfaces, forcing a capi access and
a pyaccess for the same image"""
2014-06-10 13:10:47 +04:00
caccess = im.im.pixel_access(False)
access = PyAccess.new(im, False)
w, h = im.size
for x in range(0, w, 10):
for y in range(0, h, 10):
access[(x, y)] = color
self.assertEqual(color, caccess[(x, y)])
2015-07-03 08:03:25 +03:00
# Attempt to set the value on a read-only image
access = PyAccess.new(im, True)
with self.assertRaises(ValueError):
2015-07-03 08:03:25 +03:00
access[(0, 0)] = color
2014-06-10 13:10:47 +04:00
def test_set_vs_c(self):
2014-09-05 13:36:24 +04:00
rgb = hopper('RGB')
2014-06-10 13:10:47 +04:00
rgb.load()
self._test_set_access(rgb, (255, 128, 0))
2014-09-05 13:36:24 +04:00
self._test_set_access(hopper('RGBA'), (255, 192, 128, 0))
self._test_set_access(hopper('L'), 128)
self._test_set_access(hopper('LA'), (128, 128))
self._test_set_access(hopper('1'), 255)
self._test_set_access(hopper('P'), 128)
2014-06-10 13:10:47 +04:00
# self._test_set_access(i, (128, 128)) #PA -- undone how to make
2014-09-05 13:36:24 +04:00
self._test_set_access(hopper('F'), 1024.0)
2014-06-10 13:10:47 +04:00
im = Image.new('I;16', (10, 10), 40000)
self._test_set_access(im, 45000)
im = Image.new('I;16L', (10, 10), 40000)
self._test_set_access(im, 45000)
im = Image.new('I;16B', (10, 10), 40000)
self._test_set_access(im, 45000)
im = Image.new('I', (10, 10), 40000)
self._test_set_access(im, 45000)
# im = Image.new('I;32L', (10, 10), -(2**10))
# self._test_set_access(im, -(2**13)+1)
# im = Image.new('I;32B', (10, 10), 2**10)
# self._test_set_access(im, 2**13-1)
2016-07-04 13:42:45 +03:00
# ref https://github.com/python-pillow/Pillow/pull/2009
def test_reference_counting(self):
size = 10
for _ in range(10):
# Do not save references to the image, only to the access object
px = Image.new('L', (size, 1), 0).load()
for i in range(size):
2016-08-13 05:32:13 +03:00
# pixels can contain garbage if image is released
2016-07-04 13:42:45 +03:00
self.assertEqual(px[i, 0], 0)
2014-06-10 13:10:47 +04:00
if __name__ == '__main__':
unittest.main()