mirror of
https://github.com/wiltodelta/remove-ai-watermarks.git
synced 2026-05-20 20:04:40 +02:00
test-user
7eb32fedee
- Expand ruff rules (B, S, SIM, RET, COM, C4, G, PT, PIE, T20, DTZ, ICN, TCH, RUF, ANN) - Switch pyright to strict mode with relaxed test environment - Replace try-except-pass with contextlib.suppress throughout - Move type-only imports into TYPE_CHECKING blocks - Replace ambiguous Unicode chars (en dash, multiplication sign, Greek alpha) with ASCII - Move color-matcher from base deps to [gpu], remove unused requests dep - Add pyright to dev deps, update dependabot to uv ecosystem - Fix hardcoded version in test_version, unused unpacked vars in tests - Update maintain.sh, CLAUDE.md, .gitignore, .claude/settings.json - Remove obsolete .agents/rules/project.md - Upgrade all dependencies (Pygments vulnerability fix) Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
67 lines
1.9 KiB
Python
67 lines
1.9 KiB
Python
"""Shared fixtures for remove-ai-watermarks test suite."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
if TYPE_CHECKING:
|
|
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 200x200 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 1200x1200 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
|