mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-09-21 03:18:57 +03:00
d50445ff30
Similar to the recent adoption of Black. isort is a Python utility to sort imports alphabetically and automatically separate into sections. By using isort, contributors can quickly and automatically conform to the projects style without thinking. Just let the tool do it. Uses the configuration recommended by the Black to avoid conflicts of style. Rewrite TestImageQt.test_deprecated to no rely on import order.
94 lines
3.0 KiB
Python
94 lines
3.0 KiB
Python
import array
|
|
import struct
|
|
|
|
from PIL import Image, ImagePath
|
|
from PIL._util import py3
|
|
|
|
from .helper import PillowTestCase
|
|
|
|
|
|
class TestImagePath(PillowTestCase):
|
|
def test_path(self):
|
|
|
|
p = ImagePath.Path(list(range(10)))
|
|
|
|
# sequence interface
|
|
self.assertEqual(len(p), 5)
|
|
self.assertEqual(p[0], (0.0, 1.0))
|
|
self.assertEqual(p[-1], (8.0, 9.0))
|
|
self.assertEqual(list(p[:1]), [(0.0, 1.0)])
|
|
with self.assertRaises(TypeError) as cm:
|
|
p["foo"]
|
|
self.assertEqual(str(cm.exception), "Path indices must be integers, not str")
|
|
self.assertEqual(
|
|
list(p), [(0.0, 1.0), (2.0, 3.0), (4.0, 5.0), (6.0, 7.0), (8.0, 9.0)]
|
|
)
|
|
|
|
# method sanity check
|
|
self.assertEqual(
|
|
p.tolist(), [(0.0, 1.0), (2.0, 3.0), (4.0, 5.0), (6.0, 7.0), (8.0, 9.0)]
|
|
)
|
|
self.assertEqual(
|
|
p.tolist(1), [0.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0]
|
|
)
|
|
|
|
self.assertEqual(p.getbbox(), (0.0, 1.0, 8.0, 9.0))
|
|
|
|
self.assertEqual(p.compact(5), 2)
|
|
self.assertEqual(list(p), [(0.0, 1.0), (4.0, 5.0), (8.0, 9.0)])
|
|
|
|
p.transform((1, 0, 1, 0, 1, 1))
|
|
self.assertEqual(list(p), [(1.0, 2.0), (5.0, 6.0), (9.0, 10.0)])
|
|
|
|
# alternative constructors
|
|
p = ImagePath.Path([0, 1])
|
|
self.assertEqual(list(p), [(0.0, 1.0)])
|
|
p = ImagePath.Path([0.0, 1.0])
|
|
self.assertEqual(list(p), [(0.0, 1.0)])
|
|
p = ImagePath.Path([0, 1])
|
|
self.assertEqual(list(p), [(0.0, 1.0)])
|
|
p = ImagePath.Path([(0, 1)])
|
|
self.assertEqual(list(p), [(0.0, 1.0)])
|
|
p = ImagePath.Path(p)
|
|
self.assertEqual(list(p), [(0.0, 1.0)])
|
|
p = ImagePath.Path(p.tolist(0))
|
|
self.assertEqual(list(p), [(0.0, 1.0)])
|
|
p = ImagePath.Path(p.tolist(1))
|
|
self.assertEqual(list(p), [(0.0, 1.0)])
|
|
p = ImagePath.Path(array.array("f", [0, 1]))
|
|
self.assertEqual(list(p), [(0.0, 1.0)])
|
|
|
|
arr = array.array("f", [0, 1])
|
|
if hasattr(arr, "tobytes"):
|
|
p = ImagePath.Path(arr.tobytes())
|
|
else:
|
|
p = ImagePath.Path(arr.tostring())
|
|
self.assertEqual(list(p), [(0.0, 1.0)])
|
|
|
|
def test_overflow_segfault(self):
|
|
# Some Pythons fail getting the argument as an integer, and it falls
|
|
# through to the sequence. Seeing this on 32-bit Windows.
|
|
with self.assertRaises((TypeError, MemoryError)):
|
|
# post patch, this fails with a memory error
|
|
x = evil()
|
|
|
|
# This fails due to the invalid malloc above,
|
|
# and segfaults
|
|
for i in range(200000):
|
|
if py3:
|
|
x[i] = b"0" * 16
|
|
else:
|
|
x[i] = "0" * 16
|
|
|
|
|
|
class evil:
|
|
def __init__(self):
|
|
self.corrupt = Image.core.path(0x4000000000000000)
|
|
|
|
def __getitem__(self, i):
|
|
x = self.corrupt[i]
|
|
return struct.pack("dd", x[0], x[1])
|
|
|
|
def __setitem__(self, i, x):
|
|
self.corrupt[i] = struct.unpack("dd", x)
|