mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-11-10 19:56:47 +03:00
335 lines
9.1 KiB
Python
335 lines
9.1 KiB
Python
# Test the ImageMorphology functionality
|
|
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from PIL import Image, ImageMorph, _imagingmorph
|
|
|
|
from .helper import assert_image_equal_tofile, hopper
|
|
|
|
|
|
def string_to_img(image_string: str) -> Image.Image:
|
|
"""Turn a string image representation into a binary image"""
|
|
rows = [s for s in image_string.replace(" ", "").split("\n") if len(s)]
|
|
height = len(rows)
|
|
width = len(rows[0])
|
|
im = Image.new("L", (width, height))
|
|
for i in range(width):
|
|
for j in range(height):
|
|
c = rows[j][i]
|
|
v = c in "X1"
|
|
im.putpixel((i, j), v)
|
|
|
|
return im
|
|
|
|
|
|
A = string_to_img(
|
|
"""
|
|
.......
|
|
.......
|
|
..111..
|
|
..111..
|
|
..111..
|
|
.......
|
|
.......
|
|
"""
|
|
)
|
|
|
|
|
|
def img_to_string(im: Image.Image) -> str:
|
|
"""Turn a (small) binary image into a string representation"""
|
|
chars = ".1"
|
|
width, height = im.size
|
|
return "\n".join(
|
|
"".join(chars[im.getpixel((c, r)) > 0] for c in range(width))
|
|
for r in range(height)
|
|
)
|
|
|
|
|
|
def img_string_normalize(im: str) -> str:
|
|
return img_to_string(string_to_img(im))
|
|
|
|
|
|
def assert_img_equal_img_string(a: Image.Image, b_string: str) -> None:
|
|
assert img_to_string(a) == img_string_normalize(b_string)
|
|
|
|
|
|
def test_str_to_img() -> None:
|
|
assert_image_equal_tofile(A, "Tests/images/morph_a.png")
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"op", ("corner", "dilation4", "dilation8", "erosion4", "erosion8", "edge")
|
|
)
|
|
def test_lut(op: str) -> None:
|
|
lb = ImageMorph.LutBuilder(op_name=op)
|
|
assert lb.get_lut() is None
|
|
|
|
lut = lb.build_lut()
|
|
with open(f"Tests/images/{op}.lut", "rb") as f:
|
|
assert lut == bytearray(f.read())
|
|
|
|
|
|
def test_no_operator_loaded() -> None:
|
|
mop = ImageMorph.MorphOp()
|
|
with pytest.raises(Exception) as e:
|
|
mop.apply(None)
|
|
assert str(e.value) == "No operator loaded"
|
|
with pytest.raises(Exception) as e:
|
|
mop.match(None)
|
|
assert str(e.value) == "No operator loaded"
|
|
with pytest.raises(Exception) as e:
|
|
mop.save_lut(None)
|
|
assert str(e.value) == "No operator loaded"
|
|
|
|
|
|
# Test the named patterns
|
|
def test_erosion8() -> None:
|
|
# erosion8
|
|
mop = ImageMorph.MorphOp(op_name="erosion8")
|
|
count, Aout = mop.apply(A)
|
|
assert count == 8
|
|
assert_img_equal_img_string(
|
|
Aout,
|
|
"""
|
|
.......
|
|
.......
|
|
.......
|
|
...1...
|
|
.......
|
|
.......
|
|
.......
|
|
""",
|
|
)
|
|
|
|
|
|
def test_dialation8() -> None:
|
|
# dialation8
|
|
mop = ImageMorph.MorphOp(op_name="dilation8")
|
|
count, Aout = mop.apply(A)
|
|
assert count == 16
|
|
assert_img_equal_img_string(
|
|
Aout,
|
|
"""
|
|
.......
|
|
.11111.
|
|
.11111.
|
|
.11111.
|
|
.11111.
|
|
.11111.
|
|
.......
|
|
""",
|
|
)
|
|
|
|
|
|
def test_erosion4() -> None:
|
|
# erosion4
|
|
mop = ImageMorph.MorphOp(op_name="dilation4")
|
|
count, Aout = mop.apply(A)
|
|
assert count == 12
|
|
assert_img_equal_img_string(
|
|
Aout,
|
|
"""
|
|
.......
|
|
..111..
|
|
.11111.
|
|
.11111.
|
|
.11111.
|
|
..111..
|
|
.......
|
|
""",
|
|
)
|
|
|
|
|
|
def test_edge() -> None:
|
|
# edge
|
|
mop = ImageMorph.MorphOp(op_name="edge")
|
|
count, Aout = mop.apply(A)
|
|
assert count == 1
|
|
assert_img_equal_img_string(
|
|
Aout,
|
|
"""
|
|
.......
|
|
.......
|
|
..111..
|
|
..1.1..
|
|
..111..
|
|
.......
|
|
.......
|
|
""",
|
|
)
|
|
|
|
|
|
def test_corner() -> None:
|
|
# Create a corner detector pattern
|
|
mop = ImageMorph.MorphOp(patterns=["1:(... ... ...)->0", "4:(00. 01. ...)->1"])
|
|
count, Aout = mop.apply(A)
|
|
assert count == 5
|
|
assert_img_equal_img_string(
|
|
Aout,
|
|
"""
|
|
.......
|
|
.......
|
|
..1.1..
|
|
.......
|
|
..1.1..
|
|
.......
|
|
.......
|
|
""",
|
|
)
|
|
|
|
# Test the coordinate counting with the same operator
|
|
coords = mop.match(A)
|
|
assert len(coords) == 4
|
|
assert tuple(coords) == ((2, 2), (4, 2), (2, 4), (4, 4))
|
|
|
|
coords = mop.get_on_pixels(Aout)
|
|
assert len(coords) == 4
|
|
assert tuple(coords) == ((2, 2), (4, 2), (2, 4), (4, 4))
|
|
|
|
|
|
def test_mirroring() -> None:
|
|
# Test 'M' for mirroring
|
|
mop = ImageMorph.MorphOp(patterns=["1:(... ... ...)->0", "M:(00. 01. ...)->1"])
|
|
count, Aout = mop.apply(A)
|
|
assert count == 7
|
|
assert_img_equal_img_string(
|
|
Aout,
|
|
"""
|
|
.......
|
|
.......
|
|
..1.1..
|
|
.......
|
|
.......
|
|
.......
|
|
.......
|
|
""",
|
|
)
|
|
|
|
|
|
def test_negate() -> None:
|
|
# Test 'N' for negate
|
|
mop = ImageMorph.MorphOp(patterns=["1:(... ... ...)->0", "N:(00. 01. ...)->1"])
|
|
count, Aout = mop.apply(A)
|
|
assert count == 8
|
|
assert_img_equal_img_string(
|
|
Aout,
|
|
"""
|
|
.......
|
|
.......
|
|
..1....
|
|
.......
|
|
.......
|
|
.......
|
|
.......
|
|
""",
|
|
)
|
|
|
|
|
|
def test_incorrect_mode() -> None:
|
|
im = hopper("RGB")
|
|
mop = ImageMorph.MorphOp(op_name="erosion8")
|
|
|
|
with pytest.raises(ValueError) as e:
|
|
mop.apply(im)
|
|
assert str(e.value) == "Image mode must be L"
|
|
with pytest.raises(ValueError) as e:
|
|
mop.match(im)
|
|
assert str(e.value) == "Image mode must be L"
|
|
with pytest.raises(ValueError) as e:
|
|
mop.get_on_pixels(im)
|
|
assert str(e.value) == "Image mode must be L"
|
|
|
|
|
|
def test_add_patterns() -> None:
|
|
# Arrange
|
|
lb = ImageMorph.LutBuilder(op_name="corner")
|
|
assert lb.patterns == ["1:(... ... ...)->0", "4:(00. 01. ...)->1"]
|
|
new_patterns = ["M:(00. 01. ...)->1", "N:(00. 01. ...)->1"]
|
|
|
|
# Act
|
|
lb.add_patterns(new_patterns)
|
|
|
|
# Assert
|
|
assert lb.patterns == [
|
|
"1:(... ... ...)->0",
|
|
"4:(00. 01. ...)->1",
|
|
"M:(00. 01. ...)->1",
|
|
"N:(00. 01. ...)->1",
|
|
]
|
|
|
|
|
|
def test_unknown_pattern() -> None:
|
|
with pytest.raises(Exception):
|
|
ImageMorph.LutBuilder(op_name="unknown")
|
|
|
|
|
|
def test_pattern_syntax_error() -> None:
|
|
# Arrange
|
|
lb = ImageMorph.LutBuilder(op_name="corner")
|
|
new_patterns = ["a pattern with a syntax error"]
|
|
lb.add_patterns(new_patterns)
|
|
|
|
# Act / Assert
|
|
with pytest.raises(Exception) as e:
|
|
lb.build_lut()
|
|
assert str(e.value) == 'Syntax error in pattern "a pattern with a syntax error"'
|
|
|
|
|
|
def test_load_invalid_mrl() -> None:
|
|
# Arrange
|
|
invalid_mrl = "Tests/images/hopper.png"
|
|
mop = ImageMorph.MorphOp()
|
|
|
|
# Act / Assert
|
|
with pytest.raises(Exception) as e:
|
|
mop.load_lut(invalid_mrl)
|
|
assert str(e.value) == "Wrong size operator file!"
|
|
|
|
|
|
def test_roundtrip_mrl(tmp_path: Path) -> None:
|
|
# Arrange
|
|
tempfile = str(tmp_path / "temp.mrl")
|
|
mop = ImageMorph.MorphOp(op_name="corner")
|
|
initial_lut = mop.lut
|
|
|
|
# Act
|
|
mop.save_lut(tempfile)
|
|
mop.load_lut(tempfile)
|
|
|
|
# Act / Assert
|
|
assert mop.lut == initial_lut
|
|
|
|
|
|
def test_set_lut() -> None:
|
|
# Arrange
|
|
lb = ImageMorph.LutBuilder(op_name="corner")
|
|
lut = lb.build_lut()
|
|
mop = ImageMorph.MorphOp()
|
|
|
|
# Act
|
|
mop.set_lut(lut)
|
|
|
|
# Assert
|
|
assert mop.lut == lut
|
|
|
|
|
|
def test_wrong_mode() -> None:
|
|
lut = ImageMorph.LutBuilder(op_name="corner").build_lut()
|
|
imrgb = Image.new("RGB", (10, 10))
|
|
iml = Image.new("L", (10, 10))
|
|
|
|
with pytest.raises(RuntimeError):
|
|
_imagingmorph.apply(bytes(lut), imrgb.im.id, iml.im.id)
|
|
|
|
with pytest.raises(RuntimeError):
|
|
_imagingmorph.apply(bytes(lut), iml.im.id, imrgb.im.id)
|
|
|
|
with pytest.raises(RuntimeError):
|
|
_imagingmorph.match(bytes(lut), imrgb.im.id)
|
|
|
|
# Should not raise
|
|
_imagingmorph.match(bytes(lut), iml.im.id)
|