tests/test_cli.py

from pathlib import Path

from PIL import Image

from cli import (
    _default_panorama_inspection_output_path,
    main,
    make_panorama_inspection_image,
)


def _solid(color):
    return Image.new("RGB", (2, 2), color)


def test_panorama_inspection_moves_corners_to_center_quadrants():
    image = Image.new("RGB", (4, 4))
    image.paste(_solid((255, 0, 0)), (0, 0))
    image.paste(_solid((0, 255, 0)), (2, 0))
    image.paste(_solid((0, 0, 255)), (0, 2))
    image.paste(_solid((255, 255, 0)), (2, 2))

    inspected = make_panorama_inspection_image(image)

    assert inspected.getpixel((0, 0)) == (255, 255, 0)
    assert inspected.getpixel((3, 0)) == (0, 0, 255)
    assert inspected.getpixel((0, 3)) == (0, 255, 0)
    assert inspected.getpixel((3, 3)) == (255, 0, 0)


def test_inspect_panorama_command_writes_default_output(tmp_path):
    input_path = tmp_path / "output.png"
    Image.new("RGB", (4, 4), (255, 0, 0)).save(input_path)

    main(["inspect-panorama", str(input_path)])

    assert _default_panorama_inspection_output_path(input_path).exists()


def test_default_panorama_inspection_output_path():
    assert _default_panorama_inspection_output_path(Path("output.png")) == Path("output_panorama_seams.png")