Vincent Bourdon
52 lines
1.9 KiB
Python
52 lines
1.9 KiB
Python
import io
|
|
from pathlib import Path
|
|
from PIL import Image, ImageDraw
|
|
from logimage.infrastructure.image.pillow_converter import PillowImageConverter
|
|
from logimage.infrastructure.image.local_file_source import LocalFileImageSource
|
|
from tests.helpers import count_blocks
|
|
|
|
|
|
def _make_icon_bytes() -> bytes:
|
|
"""Maison simple : rectangle + triangle sur fond blanc."""
|
|
img = Image.new("RGB", (100, 100), color=(255, 255, 255))
|
|
draw = ImageDraw.Draw(img)
|
|
# Corps de la maison
|
|
draw.rectangle([25, 50, 75, 90], fill=(0, 0, 0))
|
|
# Toit (triangle approximatif)
|
|
draw.polygon([(50, 10), (20, 50), (80, 50)], fill=(0, 0, 0))
|
|
buf = io.BytesIO()
|
|
img.save(buf, format="PNG")
|
|
return buf.getvalue()
|
|
|
|
|
|
def test_house_icon_produces_playable_grid() -> None:
|
|
"""Une icône maison simple doit donner une grille jouable (≤ 6 blocs)."""
|
|
converter = PillowImageConverter(max_clue_blocks=6)
|
|
grid = converter.to_grid(_make_icon_bytes(), width=20, height=20)
|
|
|
|
for i in range(grid.height):
|
|
blocks = count_blocks(grid.row(i))
|
|
assert blocks <= 6, f"Ligne {i} a {blocks} blocs"
|
|
|
|
for j in range(grid.width):
|
|
blocks = count_blocks(grid.col(j))
|
|
assert blocks <= 6, f"Colonne {j} a {blocks} blocs"
|
|
|
|
|
|
def test_local_source_feeds_converter(tmp_path: Path) -> None:
|
|
"""LocalFileImageSource + PillowImageConverter produisent une grille valide."""
|
|
icon_path = tmp_path / "house.png"
|
|
icon_path.write_bytes(_make_icon_bytes())
|
|
|
|
source = LocalFileImageSource(icon_path)
|
|
image_data = source.fetch()
|
|
|
|
converter = PillowImageConverter(max_clue_blocks=6)
|
|
grid = converter.to_grid(image_data.content, width=15, height=15)
|
|
|
|
assert len(image_data.content) > 0
|
|
assert grid.width == 15
|
|
assert grid.height == 15
|
|
total_filled = sum(cell for r in range(grid.height) for cell in grid.row(r))
|
|
assert total_filled > 0, "Pipeline should produce some filled cells for house icon"
|