digital-garbageandClaude Sonnet 5.5
8bbed2b1f8
Co-Authored-By: Claude Sonnet 5.5 <noreply@anthropic.com>
130 lines
5.2 KiB
Python
130 lines
5.2 KiB
Python
"""Cross-prompt word frequency as a second score for where the camera aims."""
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
|
|
|
|
import focus_memory as fm # noqa: E402
|
|
import shot_camera as sc # noqa: E402
|
|
|
|
|
|
@pytest.fixture
|
|
def store(tmp_path, monkeypatch):
|
|
monkeypatch.setattr(fm, "_path", lambda: str(tmp_path / "focus_memory.json"))
|
|
return tmp_path
|
|
|
|
|
|
def test_silent_until_enough_prompts_then_more_seen_means_more_bonus(store):
|
|
assert fm.prior() == {}
|
|
fm.observe("a", ["chin", "lamp"])
|
|
fm.observe("b", ["chin"])
|
|
assert fm.prior() == {} # two prompts: not a habit yet
|
|
fm.observe("c", ["chin", "door"])
|
|
p = fm.prior()
|
|
assert p["chin"] == pytest.approx(fm.FREQ_W) and 0 < p["lamp"] < p["chin"]
|
|
assert p["lamp"] == p["door"]
|
|
|
|
|
|
def test_the_same_prompt_again_counts_nothing(store):
|
|
for _ in range(5):
|
|
fm.observe("same", ["chin"])
|
|
assert fm.observe("same", ["chin"]) == 1 and fm.prior() == {}
|
|
|
|
|
|
def test_nothing_to_count_keeps_nothing(store):
|
|
assert fm.observe("a", []) == 0 and fm.prior() == {}
|
|
|
|
|
|
def test_the_memory_is_its_own_file_named_by_the_environment_or_the_user_dir(monkeypatch, tmp_path):
|
|
monkeypatch.setenv("SHOT_CAMERA_MEMORY", str(tmp_path / "mine.json"))
|
|
assert fm._path() == str(tmp_path / "mine.json")
|
|
fm.observe("a", ["chin"])
|
|
assert (tmp_path / "mine.json").is_file()
|
|
monkeypatch.delenv("SHOT_CAMERA_MEMORY")
|
|
assert fm._path().endswith("focus_memory.json") and "shot_camera" in fm._path()
|
|
|
|
|
|
def test_shot_camera_and_focus_memory_need_nothing_from_the_host_pack():
|
|
import re
|
|
for name in ("shot_camera.py", "focus_memory.py"):
|
|
src = (Path(__file__).resolve().parents[1] / name).read_text(encoding="utf-8")
|
|
assert not re.search(r"^\s*(?:from|import)\s+(?:conditioning|samplers|templates|movie_editor)", src, re.M)
|
|
|
|
|
|
def test_a_recurring_word_wins_between_two_otherwise_equal_targets():
|
|
pytest.importorskip("spacy")
|
|
P = "Intro. [Shot 1] <Subject 1> raises the lamp and lowers the mirror."
|
|
plain = sc.add_camera_moves(P)[1][0]["target"]
|
|
lamp = sc.add_camera_moves(P, prior={"lamp": 0.6})[1][0]["target"]
|
|
mirror = sc.add_camera_moves(P, prior={"mirror": 0.6})[1][0]["target"]
|
|
assert "lamp" in lamp or "->" in lamp
|
|
assert "mirror" in mirror or "->" in mirror
|
|
assert plain
|
|
|
|
|
|
def test_the_prior_never_outranks_an_owned_part():
|
|
pytest.importorskip("spacy")
|
|
P = "Intro. [Shot 1] <Subject 1> touches <Subject 1>'s chin and glances at the lamp."
|
|
_o, rep = sc.add_camera_moves(P, prior={"lamp": sc_max()})
|
|
assert "chin" in rep[0]["target"]
|
|
|
|
|
|
def sc_max():
|
|
return fm.FREQ_W
|
|
|
|
|
|
def test_regenerating_with_other_cuts_and_views_is_still_one_prompt():
|
|
base = "Intro. [Shot 1] <Subject 1> waves. [Shot 2] <Subject 2> nods."
|
|
varied = ("Intro. [Shot 1] <Subject 1> waves. [Shot 2] At 00:04.000, the shot changes to a new view. "
|
|
"Side view. <Subject 2> nods.")
|
|
assert sc.content_fingerprint(base) == sc.content_fingerprint(varied)
|
|
assert sc.content_fingerprint(base) != sc.content_fingerprint(base + " More.")
|
|
|
|
|
|
def test_add_camera_moves_reports_the_words_each_shot_dwells_on():
|
|
pytest.importorskip("spacy")
|
|
_o, rep = sc.add_camera_moves("Intro. [Shot 1] <Subject 1> touches <Subject 1>'s chin.")
|
|
assert rep[0]["lemmas"] == ["chin"]
|
|
|
|
|
|
def test_refiner_wrapper_counts_prompts_and_uses_the_prior(store, monkeypatch):
|
|
pytest.importorskip("spacy")
|
|
import conditioning
|
|
R = conditioning.FunPackVideoRefinerV2
|
|
monkeypatch.setattr(fm, "_path", lambda: str(store / "focus_memory.json"))
|
|
for i in range(3):
|
|
R._v2_camera_moves(f"Intro. [Shot 1] <Subject 1> touches <Subject 1>'s chin{i and ' again' * i}.",
|
|
"prompt", 1, 1.0)
|
|
assert fm.prior().get("chin", 0) > 0
|
|
|
|
|
|
def test_a_habit_does_not_win_every_shot_it_merely_appears_in():
|
|
"""100 videos with bananas, then a shot with a banana AND a cherry: the cherry still gets
|
|
the camera some of the time, and only between equals."""
|
|
pytest.importorskip("spacy")
|
|
P = "Intro. [Shot 1] <Subject 1> holds a banana and a cherry."
|
|
heard = {"banana": 0, "cherry": 0}
|
|
for seed in range(200):
|
|
_o, rep = sc.add_camera_moves(P, seed=seed, chance=1.0, prior={"banana": fm.FREQ_W})
|
|
t = rep[0]["target"]
|
|
heard["banana"] += "banana" in t
|
|
heard["cherry"] += "cherry" in t
|
|
assert heard["banana"] > heard["cherry"] > 20, heard # favoured, not decisive
|
|
|
|
|
|
def test_a_habit_never_beats_a_real_target():
|
|
pytest.importorskip("spacy")
|
|
P = "Intro. [Shot 1] <Subject 1> touches <Subject 1>'s chin and glances at a banana."
|
|
for seed in range(60):
|
|
_o, rep = sc.add_camera_moves(P, seed=seed, chance=1.0, prior={"banana": fm.FREQ_W})
|
|
assert "chin" in rep[0]["target"], rep[0]["target"]
|
|
|
|
|
|
def test_a_word_the_shot_keeps_returning_to_is_what_it_is_about():
|
|
pytest.importorskip("spacy")
|
|
P = "Intro. [Shot 1] <Subject 1> lifts the cherry, then the lamp, then looks at the cherry again."
|
|
hits = sum("cherry" in sc.add_camera_moves(P, seed=s, chance=1.0)[1][0]["target"] for s in range(80))
|
|
assert hits > 50, hits
|