Skip to content

Commit d2f53c6

Browse files
committed
Return LUT from build_default_lut()
1 parent ca21683 commit d2f53c6

File tree

2 files changed

+7
-1
lines changed

2 files changed

+7
-1
lines changed

Tests/test_imagemorph.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -281,6 +281,11 @@ def test_pattern_syntax_error(pattern: str) -> None:
281281
lb.build_lut()
282282

283283

284+
def test_build_default_lut() -> None:
285+
lb = ImageMorph.LutBuilder(op_name="corner")
286+
assert lb.build_default_lut() == lb.lut
287+
288+
284289
def test_load_invalid_mrl() -> None:
285290
# Arrange
286291
invalid_mrl = "Tests/images/hopper.png"

src/PIL/ImageMorph.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -92,10 +92,11 @@ def __init__(
9292
def add_patterns(self, patterns: list[str]) -> None:
9393
self.patterns += patterns
9494

95-
def build_default_lut(self) -> None:
95+
def build_default_lut(self) -> bytearray:
9696
symbols = [0, 1]
9797
m = 1 << 4 # pos of current pixel
9898
self.lut = bytearray(symbols[(i & m) > 0] for i in range(LUT_SIZE))
99+
return self.lut
99100

100101
def get_lut(self) -> bytearray | None:
101102
return self.lut

0 commit comments

Comments
 (0)