mirror of
https://github.com/wiltodelta/remove-ai-watermarks.git
synced 2026-10-14 13:30:55 +02:00
Victor Kuznetsov
5c6588b3ef
Repair image and metadata publication contracts, detection routing, research artifact validation, release verification, and the published agent skill. Refresh the dependency lock and reconcile documentation with verified behavior. Add regression coverage and record the completed audit and corrected VideoSeal measurements. Preserve the existing Accelerate and Lightning versions. Validation: Ruff and Pyright pass; 2358 tests pass with 3 skipped. The maintenance gate exits 2 on the documented Accelerate and Lightning security findings.
81 lines
2.6 KiB
Python
81 lines
2.6 KiB
Python
from __future__ import annotations
|
|
|
|
import json
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
from click.testing import CliRunner
|
|
|
|
sys.path.insert(0, str(Path(__file__).resolve().parent.parent / "scripts"))
|
|
|
|
import synthid_routed_detector as detector
|
|
import synthid_routed_expert_bank as bank
|
|
|
|
|
|
def test_detect_path_routes_one_scored_record(monkeypatch, tmp_path: Path) -> None:
|
|
image_path = tmp_path / "image.png"
|
|
image_path.write_bytes(b"fixture")
|
|
|
|
def score_path(path: Path) -> dict[str, object]:
|
|
assert path == image_path
|
|
return {
|
|
"id": "a" * 64,
|
|
"path": str(path),
|
|
"width": 1024,
|
|
"height": 1024,
|
|
"observations": [
|
|
{"name": bank.synthid_detector.OPPONENT_REGISTERED_DETECTOR_ID, "supported": False, "score": None},
|
|
{"name": bank.synthid_detector.FINE_OPPONENT_REGISTERED_DETECTOR_ID, "supported": False, "score": None},
|
|
{
|
|
"name": bank.synthid_detector.DETECTOR_ID,
|
|
"supported": True,
|
|
"score": 0.5,
|
|
},
|
|
{
|
|
"name": bank.synthid_detector.REGISTERED_DETECTOR_ID,
|
|
"supported": True,
|
|
"score": 0.0,
|
|
},
|
|
{
|
|
"name": bank.synthid_detector.LARGE_DETECTOR_ID,
|
|
"supported": False,
|
|
"score": None,
|
|
},
|
|
],
|
|
}
|
|
|
|
monkeypatch.setattr(detector, "score_path", score_path)
|
|
|
|
result = detector.detect_path(image_path)
|
|
|
|
assert result["result"]["verdict"] == "abstain"
|
|
assert result["result"]["reason"] == "fixed_only_ambiguous"
|
|
|
|
|
|
def test_cli_writes_combined_hash_pinned_report(monkeypatch, tmp_path: Path) -> None:
|
|
image_path = tmp_path / "image.png"
|
|
report_path = tmp_path / "report.json"
|
|
image_path.write_bytes(b"fixture")
|
|
monkeypatch.setattr(
|
|
detector,
|
|
"detect_path",
|
|
lambda path: {
|
|
"id": "b" * 64,
|
|
"path": str(path),
|
|
"width": 1024,
|
|
"height": 1024,
|
|
"observations": [],
|
|
"result": {"verdict": "detected", "reason": "registered_threshold_crossed"},
|
|
},
|
|
)
|
|
|
|
result = CliRunner().invoke(
|
|
detector.main,
|
|
[str(image_path), "--report-out", str(report_path)],
|
|
)
|
|
|
|
assert result.exit_code == 0, result.output
|
|
report = json.loads(report_path.read_text(encoding="utf-8"))
|
|
assert report["counts"] == {"detected": 1, "abstain": 0}
|
|
assert report["records"][0]["id"] == "b" * 64
|