File size: 2,432 Bytes
821976c 2604501 821976c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 | from types import SimpleNamespace
from videoclean.cli import _build_parser
from videoclean.engine import WipeEngine
def test_cli_defaults_to_auto_detection():
args = _build_parser().parse_args(["clean", "input.mp4"])
assert args.detect_mode == "auto"
def test_explicit_modes_remain_available():
for mode in ("fast", "balanced", "sensitive"):
args = _build_parser().parse_args(["clean", "input.mp4", "--detect-mode", mode])
assert args.detect_mode == mode
def test_auto_mode_runs_balanced_then_sensitive(monkeypatch, tmp_path):
empty = SimpleNamespace(candidates=[])
found = SimpleNamespace(candidates=[])
calls = []
def fake_detect(*args, **kwargs):
calls.append(kwargs["sample_count"])
return empty if len(calls) == 1 else found
monkeypatch.setattr("videoclean.detect.detect_clean_candidates", fake_detect)
monkeypatch.setattr("videoclean.detect._default_detector", lambda: object())
monkeypatch.setattr("videoclean.detect.write_clean_artifacts", lambda *args: None)
monkeypatch.setattr("videoclean.engine.WipeEngine._build_recognizer", staticmethod(lambda mode: None))
engine = WipeEngine(task="clean", detect_mode="auto", ocr="off")
result, selected, snapshot, directed = engine._detect_clean(
"input.mp4", object(), [], None, [], None, "off", str(tmp_path), False, "dbnet"
)
assert calls == [24, 80]
assert snapshot["detect_mode"] == "auto"
assert snapshot["effective_detect_mode"] == "sensitive"
assert selected == set()
assert directed is False
def test_explicit_fast_does_not_retry(monkeypatch, tmp_path):
calls = []
def fake_detect(*args, **kwargs):
calls.append(kwargs["sample_count"])
return SimpleNamespace(candidates=[])
monkeypatch.setattr("videoclean.detect.detect_clean_candidates", fake_detect)
monkeypatch.setattr("videoclean.detect._default_detector", lambda: object())
monkeypatch.setattr("videoclean.detect.write_clean_artifacts", lambda *args: None)
monkeypatch.setattr("videoclean.engine.WipeEngine._build_recognizer", staticmethod(lambda mode: None))
engine = WipeEngine(task="clean", detect_mode="fast", ocr="off")
_, _, snapshot, _ = engine._detect_clean(
"input.mp4", object(), [], None, [], None, "off", str(tmp_path), False, "dbnet"
)
assert calls == [24]
assert snapshot["effective_detect_mode"] == "fast"
|