mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-12-26 09:56:17 +03:00
62 lines
2.0 KiB
Python
62 lines
2.0 KiB
Python
import pytest
|
|
|
|
from PIL import Image
|
|
|
|
from .helper import hopper
|
|
|
|
im = hopper().resize((128, 100))
|
|
|
|
|
|
def test_toarray():
|
|
def test(mode):
|
|
ai = im.convert(mode).__array_interface__
|
|
return ai["version"], ai["shape"], ai["typestr"], len(ai["data"])
|
|
|
|
# assert test("1") == (3, (100, 128), '|b1', 1600))
|
|
assert test("L") == (3, (100, 128), "|u1", 12800)
|
|
|
|
# FIXME: wrong?
|
|
assert test("I") == (3, (100, 128), Image._ENDIAN + "i4", 51200)
|
|
# FIXME: wrong?
|
|
assert test("F") == (3, (100, 128), Image._ENDIAN + "f4", 51200)
|
|
|
|
assert test("LA") == (3, (100, 128, 2), "|u1", 25600)
|
|
assert test("RGB") == (3, (100, 128, 3), "|u1", 38400)
|
|
assert test("RGBA") == (3, (100, 128, 4), "|u1", 51200)
|
|
assert test("RGBX") == (3, (100, 128, 4), "|u1", 51200)
|
|
|
|
|
|
def test_fromarray():
|
|
class Wrapper:
|
|
""" Class with API matching Image.fromarray """
|
|
|
|
def __init__(self, img, arr_params):
|
|
self.img = img
|
|
self.__array_interface__ = arr_params
|
|
|
|
def tobytes(self):
|
|
return self.img.tobytes()
|
|
|
|
def test(mode):
|
|
i = im.convert(mode)
|
|
a = i.__array_interface__
|
|
a["strides"] = 1 # pretend it's non-contiguous
|
|
# Make wrapper instance for image, new array interface
|
|
wrapped = Wrapper(i, a)
|
|
out = Image.fromarray(wrapped)
|
|
return out.mode, out.size, list(i.getdata()) == list(out.getdata())
|
|
|
|
# assert test("1") == ("1", (128, 100), True)
|
|
assert test("L") == ("L", (128, 100), True)
|
|
assert test("I") == ("I", (128, 100), True)
|
|
assert test("F") == ("F", (128, 100), True)
|
|
assert test("LA") == ("LA", (128, 100), True)
|
|
assert test("RGB") == ("RGB", (128, 100), True)
|
|
assert test("RGBA") == ("RGBA", (128, 100), True)
|
|
assert test("RGBX") == ("RGBA", (128, 100), True)
|
|
|
|
# Test mode is None with no "typestr" in the array interface
|
|
with pytest.raises(TypeError):
|
|
wrapped = Wrapper(test("L"), {"shape": (100, 128)})
|
|
Image.fromarray(wrapped)
|