python-pillow/Pillow
1
1
Fork 0
mirror of https://github.com/python-pillow/Pillow.git synced 2025-08-16 18:24:45 +03:00
Code Issues Packages Projects Releases Wiki Activity

Restored __array__ for libraries that do not support __array_interface__

Browse Source
This commit is contained in:
Andrew Murray 2022-08-06 19:52:02 +10:00
parent 92b0f2c919
commit d871ef51df
2 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
Tests/test_image_array.py
View File

@ -94,3 +94,10 @@ def test_fromarray_palette():
# Assert that the Python and C palettes match # Assert that the Python and C palettes match
assert len(out.palette.colors) == len(out.im.getpalette()) / 3 assert len(out.palette.colors) == len(out.im.getpalette()) / 3
def test_array():
i = im.convert("L")
a = numpy.array(i)
assert (a == i.__array__()).all()

5
src/PIL/Image.py
View File

@ -671,6 +671,11 @@ class Image:
raise ValueError("Could not save to PNG for display") from e raise ValueError("Could not save to PNG for display") from e
return b.getvalue() return b.getvalue()
def __array__(self, dtype=None):
import numpy as np
return np.array(self, dtype)
@property @property
def __array_interface__(self): def __array_interface__(self):
# numpy array interface support # numpy array interface support