digital-garbageandClaude Sonnet 5.5
f8f612277e
Co-Authored-By: Claude Sonnet 5.5 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01Tw9fjV3UWsHHjJiAXix29B
96 lines
5.5 KiB
Python
96 lines
5.5 KiB
Python
"""Reframe within the shot: a kept shortcut boundary turns the camera in one continuous take, never a cut."""
|
|
|
|
import pytest
|
|
|
|
from .. import engine, memory, nodes
|
|
|
|
DANCE = "She sways her hips from side to side, smiling at the room."
|
|
WAVE = "She raises her hand and waves at the camera with her fingers."
|
|
|
|
|
|
def test_reframe_turns_the_camera_between_shortcuts_without_a_cut(tmp_path, monkeypatch):
|
|
monkeypatch.setenv("SHOT_CAMERA_MEMORY", str(tmp_path / "m.json"))
|
|
text = f"[Shot 1] {DANCE} {WAVE}"
|
|
out, said, chose = nodes.rewrite(text, {"reframe_same_shot": True, "reframe_chance": 1}, seconds=10, pieces=[DANCE, WAVE])
|
|
assert out.count("[Shot ") == 1 and "cuts to" not in out, "one take: no new shot, no cut words"
|
|
head, tail = out.split("Without a cut, the camera ")
|
|
assert DANCE in head and WAVE in tail, "the reframe sits between the two actions, in order"
|
|
assert "hand" in tail.split(".")[0] or "fingers" in tail.split(".")[0], "moves to what the next action is about"
|
|
assert "reframes: shot 1 to " in said
|
|
assert chose["arms"][0] == "reframe:yes" and chose["arms"][1] in ("reframe:keep", "reframe:turn")
|
|
|
|
|
|
def test_every_reframe_moves_to_a_new_focus_and_the_view_may_stay():
|
|
pytest.importorskip("spacy")
|
|
text = f"[Shot 1] {DANCE} {DANCE} {DANCE} {DANCE}"
|
|
out, info = engine.add_reframes(text, seed="s", chance=1, pieces=[DANCE])
|
|
targets = [a["target"] for a in info["added"]]
|
|
assert len(targets) == 3 and all(t for t in targets)
|
|
assert all(a != b for a, b in zip(targets, targets[1:])), "the focus changes at every reframe"
|
|
assert all(v != "View from behind" for v in (a["view"] for a in info["added"]) if v), "a smiling face is not turned away from"
|
|
kept = sum(1 for seed in range(40) for a in engine.add_reframes(text, seed=seed, chance=1, pieces=[DANCE])[1]["added"] if a["view"] is None)
|
|
assert kept > 0, "keeping the view is allowed"
|
|
|
|
|
|
def test_at_one_every_boundary_reframes_even_with_nothing_new_to_aim_at():
|
|
plain = "She smiles warmly at the room again."
|
|
out, info = engine.add_reframes(f"[Shot 1] {plain} {plain}", seed=1, chance=1, pieces=[plain])
|
|
assert out.count("Without a cut") == 1 and info["added"][0]["target"] is None, "1.0 means every boundary: the camera still moves"
|
|
|
|
|
|
def test_zero_chance_adds_nothing_and_no_boundary_says_so(tmp_path, monkeypatch):
|
|
monkeypatch.setenv("SHOT_CAMERA_MEMORY", str(tmp_path / "m.json"))
|
|
text = f"[Shot 1] {DANCE} {WAVE}"
|
|
assert engine.add_reframes(text, seed=1, chance=0, pieces=[DANCE, WAVE])[0] == text
|
|
out, said, _ = nodes.rewrite("[Shot 1] A woman walks.", {"reframe_same_shot": True, "reframe_chance": 1}, seconds=5, pieces=[])
|
|
assert out == "[Shot 1] A woman walks." and "no shortcut boundary inside a shot" in said
|
|
|
|
|
|
def test_a_cut_boundary_is_not_also_reframed(tmp_path, monkeypatch):
|
|
monkeypatch.setenv("SHOT_CAMERA_MEMORY", str(tmp_path / "m.json"))
|
|
text = f"[Shot 1] {DANCE} {WAVE}"
|
|
out, _, _ = nodes.rewrite(text, {"cut_same_shot": True, "shot_cuts_chance": 1, "reframe_same_shot": True, "reframe_chance": 1},
|
|
seconds=10, pieces=[DANCE, WAVE])
|
|
assert out.count("[Shot ") == 2 and "Without a cut" not in out
|
|
|
|
|
|
def test_the_reframe_is_not_content_and_does_not_block_the_opening_view():
|
|
text = f"[Shot 1] A woman walks in. [Shot 2] {DANCE} {WAVE}"
|
|
out, _ = engine.add_reframes(text, seed=2, chance=1, pieces=[DANCE, WAVE])
|
|
assert engine.content_fingerprint(out) == engine.content_fingerprint(text), "same prompt, same seed next run"
|
|
shot2 = engine.shot_texts(out)[2]
|
|
assert "Without a cut" not in shot2
|
|
_, added = engine.add_shot_views(out, seed=0, chance=1)
|
|
assert [a["shot"] for a in added] == [2], "a mid-shot reframe is not the view the shot opens on"
|
|
|
|
|
|
def test_ratings_tilt_the_reframe_chance(tmp_path, monkeypatch):
|
|
monkeypatch.setenv("SHOT_CAMERA_MEMORY", str(tmp_path / "m.json"))
|
|
assert memory.reframe_chance(0.5) == 0.5
|
|
memory.record_run("r", {"views": [], "arms": ["reframe:yes"]})
|
|
memory.on_rating("r", "liked")
|
|
assert memory.reframe_chance(0.5) > 0.5
|
|
assert memory.reframe_chance(1.0) == 1.0, "always stays always"
|
|
|
|
|
|
def test_cut_within_the_shot_at_zero_leaves_every_boundary_to_reframe(tmp_path, monkeypatch):
|
|
monkeypatch.setenv("SHOT_CAMERA_MEMORY", str(tmp_path / "m.json"))
|
|
text = f"[Shot 1] {DANCE} {WAVE} {DANCE}"
|
|
out, _, _ = nodes.rewrite(text, {"cut_same_shot": True, "shot_cuts_chance": 0, "reframe_same_shot": True, "reframe_chance": 1},
|
|
seconds=10, pieces=[DANCE, WAVE])
|
|
assert out.count("[Shot ") == 1 and out.count("Without a cut") == 2
|
|
|
|
|
|
def test_only_one_is_always_and_only_zero_is_never(tmp_path, monkeypatch):
|
|
monkeypatch.setenv("SHOT_CAMERA_MEMORY", str(tmp_path / "m.json"))
|
|
for _ in range(30): # a long run of likes: the strongest possible lean
|
|
memory.record_run("r", {"views": [], "arms": ["reframe:yes", "split:yes", "move:yes"]})
|
|
memory.on_rating("r", "liked")
|
|
memory.record_run("q", {"views": [], "arms": ["reframe:no", "split:no", "move:no"]})
|
|
memory.on_rating("q", "disliked")
|
|
for fn in (memory.reframe_chance, memory.split_chance, memory.effective_chance):
|
|
assert fn(1.0) == 1.0 and fn(0.0) == 0.0
|
|
assert 0.0 < fn(0.6) < 1.0 and 0.0 < fn(0.05) < 1.0, f"{fn.__name__}: in between stays occasionally"
|
|
assert fn(0.6) > 0.6, "the likes still lean it up"
|
|
assert 0.0 < memory.detail_chance(0.9, 100, 0) < 1.0 and memory.detail_chance(1.0, 0, 100) == 1.0
|