mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-11-10 19:56:47 +03:00
Merge pull request #5102 from hugovk/add-path-tests
Add tests for ImagePath.Path
This commit is contained in:
Andrew Murray
GitHub
commit
c52c3ae3e2
|
|
@ -1,4 +1,5 @@
|
|||
import array
|
||||
import math
|
||||
import struct
|
||||
|
||||
import pytest
|
||||
|
|
@ -6,75 +7,175 @@ import pytest
|
|||
from PIL import Image, ImagePath
|
||||
|
||||
|
||||
class TestImagePath:
|
||||
def test_path(self):
|
||||
def test_path():
|
||||
|
||||
p = ImagePath.Path(list(range(10)))
|
||||
p = ImagePath.Path(list(range(10)))
|
||||
|
||||
# sequence interface
|
||||
assert len(p) == 5
|
||||
assert p[0] == (0.0, 1.0)
|
||||
assert p[-1] == (8.0, 9.0)
|
||||
assert list(p[:1]) == [(0.0, 1.0)]
|
||||
with pytest.raises(TypeError) as cm:
|
||||
p["foo"]
|
||||
assert str(cm.value) == "Path indices must be integers, not str"
|
||||
assert list(p) == [(0.0, 1.0), (2.0, 3.0), (4.0, 5.0), (6.0, 7.0), (8.0, 9.0)]
|
||||
# sequence interface
|
||||
assert len(p) == 5
|
||||
assert p[0] == (0.0, 1.0)
|
||||
assert p[-1] == (8.0, 9.0)
|
||||
assert list(p[:1]) == [(0.0, 1.0)]
|
||||
with pytest.raises(TypeError) as cm:
|
||||
p["foo"]
|
||||
assert str(cm.value) == "Path indices must be integers, not str"
|
||||
assert 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
|
||||
assert p.tolist() == [
|
||||
(0.0, 1.0),
|
||||
(2.0, 3.0),
|
||||
(4.0, 5.0),
|
||||
(6.0, 7.0),
|
||||
(8.0, 9.0),
|
||||
]
|
||||
assert p.tolist(1) == [0.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0]
|
||||
# method sanity check
|
||||
assert p.tolist() == [
|
||||
(0.0, 1.0),
|
||||
(2.0, 3.0),
|
||||
(4.0, 5.0),
|
||||
(6.0, 7.0),
|
||||
(8.0, 9.0),
|
||||
]
|
||||
assert p.tolist(1) == [0.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0]
|
||||
|
||||
assert p.getbbox() == (0.0, 1.0, 8.0, 9.0)
|
||||
assert p.getbbox() == (0.0, 1.0, 8.0, 9.0)
|
||||
|
||||
assert p.compact(5) == 2
|
||||
assert list(p) == [(0.0, 1.0), (4.0, 5.0), (8.0, 9.0)]
|
||||
assert p.compact(5) == 2
|
||||
assert list(p) == [(0.0, 1.0), (4.0, 5.0), (8.0, 9.0)]
|
||||
|
||||
p.transform((1, 0, 1, 0, 1, 1))
|
||||
assert list(p) == [(1.0, 2.0), (5.0, 6.0), (9.0, 10.0)]
|
||||
p.transform((1, 0, 1, 0, 1, 1))
|
||||
assert list(p) == [(1.0, 2.0), (5.0, 6.0), (9.0, 10.0)]
|
||||
|
||||
# alternative constructors
|
||||
p = ImagePath.Path([0, 1])
|
||||
assert list(p) == [(0.0, 1.0)]
|
||||
p = ImagePath.Path([0.0, 1.0])
|
||||
assert list(p) == [(0.0, 1.0)]
|
||||
p = ImagePath.Path([0, 1])
|
||||
assert list(p) == [(0.0, 1.0)]
|
||||
p = ImagePath.Path([(0, 1)])
|
||||
assert list(p) == [(0.0, 1.0)]
|
||||
p = ImagePath.Path(p)
|
||||
assert list(p) == [(0.0, 1.0)]
|
||||
p = ImagePath.Path(p.tolist(0))
|
||||
assert list(p) == [(0.0, 1.0)]
|
||||
p = ImagePath.Path(p.tolist(1))
|
||||
assert list(p) == [(0.0, 1.0)]
|
||||
p = ImagePath.Path(array.array("f", [0, 1]))
|
||||
assert list(p) == [(0.0, 1.0)]
|
||||
# alternative constructors
|
||||
p = ImagePath.Path([0, 1])
|
||||
assert list(p) == [(0.0, 1.0)]
|
||||
p = ImagePath.Path([0.0, 1.0])
|
||||
assert list(p) == [(0.0, 1.0)]
|
||||
p = ImagePath.Path([0, 1])
|
||||
assert list(p) == [(0.0, 1.0)]
|
||||
p = ImagePath.Path([(0, 1)])
|
||||
assert list(p) == [(0.0, 1.0)]
|
||||
p = ImagePath.Path(p)
|
||||
assert list(p) == [(0.0, 1.0)]
|
||||
p = ImagePath.Path(p.tolist(0))
|
||||
assert list(p) == [(0.0, 1.0)]
|
||||
p = ImagePath.Path(p.tolist(1))
|
||||
assert list(p) == [(0.0, 1.0)]
|
||||
p = ImagePath.Path(array.array("f", [0, 1]))
|
||||
assert 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())
|
||||
assert 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())
|
||||
assert 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 pytest.raises((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):
|
||||
x[i] = b"0" * 16
|
||||
def test_invalid_coords():
|
||||
# Arrange
|
||||
coords = ["a", "b"]
|
||||
|
||||
# Act / Assert
|
||||
with pytest.raises(SystemError):
|
||||
ImagePath.Path(coords)
|
||||
|
||||
|
||||
def test_path_odd_number_of_coordinates():
|
||||
# Arrange
|
||||
coords = [0]
|
||||
|
||||
# Act / Assert
|
||||
with pytest.raises(ValueError) as e:
|
||||
ImagePath.Path(coords)
|
||||
|
||||
assert str(e.value) == "wrong number of coordinates"
|
||||
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
"coords, expected",
|
||||
[
|
||||
([0, 1, 2, 3], (0.0, 1.0, 2.0, 3.0)),
|
||||
([3, 2, 1, 0], (1.0, 0.0, 3.0, 2.0)),
|
||||
],
|
||||
)
|
||||
def test_getbbox(coords, expected):
|
||||
# Arrange
|
||||
p = ImagePath.Path(coords)
|
||||
|
||||
# Act / Assert
|
||||
assert p.getbbox() == expected
|
||||
|
||||
|
||||
def test_getbbox_no_args():
|
||||
# Arrange
|
||||
p = ImagePath.Path([0, 1, 2, 3])
|
||||
|
||||
# Act / Assert
|
||||
with pytest.raises(TypeError):
|
||||
p.getbbox(1)
|
||||
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
"coords, expected",
|
||||
[
|
||||
(0, []),
|
||||
(list(range(6)), [(0.0, 3.0), (4.0, 9.0), (8.0, 15.0)]),
|
||||
],
|
||||
)
|
||||
def test_map(coords, expected):
|
||||
# Arrange
|
||||
p = ImagePath.Path(coords)
|
||||
|
||||
# Act
|
||||
# Modifies the path in-place
|
||||
p.map(lambda x, y: (x * 2, y * 3))
|
||||
|
||||
# Assert
|
||||
assert list(p) == expected
|
||||
|
||||
|
||||
def test_transform():
|
||||
# Arrange
|
||||
p = ImagePath.Path([0, 1, 2, 3])
|
||||
theta = math.pi / 15
|
||||
|
||||
# Act
|
||||
# Affine transform, in-place
|
||||
p.transform(
|
||||
(math.cos(theta), math.sin(theta), 20, -math.sin(theta), math.cos(theta), 20),
|
||||
)
|
||||
|
||||
# Assert
|
||||
assert p.tolist() == [
|
||||
(20.20791169081776, 20.978147600733806),
|
||||
(22.58003027392089, 22.518619420565898),
|
||||
]
|
||||
|
||||
|
||||
def test_transform_with_wrap():
|
||||
# Arrange
|
||||
p = ImagePath.Path([0, 1, 2, 3])
|
||||
theta = math.pi / 15
|
||||
|
||||
# Act
|
||||
# Affine transform, in-place, with wrap parameter
|
||||
p.transform(
|
||||
(math.cos(theta), math.sin(theta), 20, -math.sin(theta), math.cos(theta), 20),
|
||||
1.0,
|
||||
)
|
||||
|
||||
# Assert
|
||||
assert p.tolist() == [
|
||||
(0.20791169081775962, 20.978147600733806),
|
||||
(0.5800302739208902, 22.518619420565898),
|
||||
]
|
||||
|
||||
|
||||
def test_overflow_segfault():
|
||||
# Some Pythons fail getting the argument as an integer, and it falls
|
||||
# through to the sequence. Seeing this on 32-bit Windows.
|
||||
with pytest.raises((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):
|
||||
x[i] = b"0" * 16
|
||||
|
||||
|
||||
class evil:
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user