mirror of
https://github.com/wiltodelta/remove-ai-watermarks.git
synced 2026-05-21 12:16:46 +02:00
test-user
e5d8970add
- CLI with visible, invisible, all, metadata, and batch commands - Gemini watermark removal via reverse alpha blending - Invisible watermark removal via diffusion regeneration (SynthID, TreeRing) - AI metadata stripping (EXIF, PNG text, C2PA) - Face protection (YOLO/Haar) and analog humanizer - 137 tests covering all CLI modes and core engines - Ruff and Pyright clean
64 lines
1.9 KiB
Python
64 lines
1.9 KiB
Python
"""Shared fixtures for remove-ai-watermarks test suite."""
|
||
|
||
from __future__ import annotations
|
||
|
||
from pathlib import Path
|
||
|
||
import cv2
|
||
import numpy as np
|
||
import pytest
|
||
from PIL import Image
|
||
from PIL.PngImagePlugin import PngInfo
|
||
|
||
|
||
@pytest.fixture()
|
||
def tmp_image_path(tmp_path: Path) -> Path:
|
||
"""Create a minimal 200×200 test PNG image and return its path."""
|
||
img = np.random.randint(0, 255, (200, 200, 3), dtype=np.uint8)
|
||
path = tmp_path / "test_image.png"
|
||
cv2.imwrite(str(path), img)
|
||
return path
|
||
|
||
|
||
@pytest.fixture()
|
||
def tmp_large_image_path(tmp_path: Path) -> Path:
|
||
"""Create a 1200×1200 test PNG image (triggers large watermark branch)."""
|
||
img = np.random.randint(0, 255, (1200, 1200, 3), dtype=np.uint8)
|
||
path = tmp_path / "test_large.png"
|
||
cv2.imwrite(str(path), img)
|
||
return path
|
||
|
||
|
||
@pytest.fixture()
|
||
def tmp_jpeg_path(tmp_path: Path) -> Path:
|
||
"""Create a minimal JPEG test image."""
|
||
img = np.random.randint(0, 255, (100, 100, 3), dtype=np.uint8)
|
||
path = tmp_path / "test_image.jpg"
|
||
cv2.imwrite(str(path), img)
|
||
return path
|
||
|
||
|
||
@pytest.fixture()
|
||
def tmp_png_with_ai_metadata(tmp_path: Path) -> Path:
|
||
"""Create a PNG with AI-related metadata keys."""
|
||
img = Image.new("RGB", (64, 64), color=(128, 128, 128))
|
||
pnginfo = PngInfo()
|
||
pnginfo.add_text("parameters", "Steps: 20, Sampler: Euler, CFG scale: 7")
|
||
pnginfo.add_text("prompt", "a beautiful landscape")
|
||
pnginfo.add_text("Author", "Test Author")
|
||
path = tmp_path / "ai_metadata.png"
|
||
img.save(path, pnginfo=pnginfo)
|
||
return path
|
||
|
||
|
||
@pytest.fixture()
|
||
def tmp_clean_png(tmp_path: Path) -> Path:
|
||
"""Create a PNG with no AI metadata."""
|
||
img = Image.new("RGB", (64, 64), color=(200, 100, 50))
|
||
pnginfo = PngInfo()
|
||
pnginfo.add_text("Author", "Human Artist")
|
||
pnginfo.add_text("Title", "Test Artwork")
|
||
path = tmp_path / "clean.png"
|
||
img.save(path, pnginfo=pnginfo)
|
||
return path
|