Pillow/Tests/test_image_getdata.py

30 lines
937 B
Python
Raw Normal View History

from .helper import PillowTestCase, hopper
2014-06-10 13:10:47 +04:00
class TestImageGetData(PillowTestCase):
2014-06-10 13:10:47 +04:00
def test_sanity(self):
data = hopper().getdata()
2014-06-10 13:10:47 +04:00
len(data)
list(data)
self.assertEqual(data[0], (20, 20, 70))
2014-06-10 13:10:47 +04:00
def test_roundtrip(self):
2014-06-10 13:10:47 +04:00
def getdata(mode):
im = hopper(mode).resize((32, 30))
2014-06-10 13:10:47 +04:00
data = im.getdata()
return data[0], len(data), len(list(data))
self.assertEqual(getdata("1"), (0, 960, 960))
2016-07-11 01:47:25 +03:00
self.assertEqual(getdata("L"), (16, 960, 960))
2019-05-08 14:58:33 +03:00
self.assertEqual(getdata("I"), (4313, 960, 960))
2016-07-11 01:47:25 +03:00
self.assertEqual(getdata("F"), (16.0, 960, 960))
2018-10-02 11:55:28 +03:00
self.assertEqual(getdata("RGB"), ((11, 13, 52), 960, 960))
2016-07-11 01:47:25 +03:00
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))