mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-09-21 11:28:58 +03:00
30 lines
971 B
Python
30 lines
971 B
Python
from PIL import Image
|
|
|
|
from .helper import PillowTestCase, hopper
|
|
|
|
|
|
class TestImageGetData(PillowTestCase):
|
|
def test_sanity(self):
|
|
|
|
data = hopper().getdata()
|
|
|
|
len(data)
|
|
list(data)
|
|
|
|
self.assertEqual(data[0], (20, 20, 70))
|
|
|
|
def test_roundtrip(self):
|
|
def getdata(mode):
|
|
im = hopper(mode).resize((32, 30), Image.NEAREST)
|
|
data = im.getdata()
|
|
return data[0], len(data), len(list(data))
|
|
|
|
self.assertEqual(getdata("1"), (0, 960, 960))
|
|
self.assertEqual(getdata("L"), (17, 960, 960))
|
|
self.assertEqual(getdata("I"), (17, 960, 960))
|
|
self.assertEqual(getdata("F"), (17.0, 960, 960))
|
|
self.assertEqual(getdata("RGB"), ((11, 13, 52), 960, 960))
|
|
self.assertEqual(getdata("RGBA"), ((11, 13, 52, 255), 960, 960))
|
|
self.assertEqual(getdata("CMYK"), ((244, 242, 203, 0), 960, 960))
|
|
self.assertEqual(getdata("YCbCr"), ((16, 147, 123), 960, 960))
|