Report anatomy guard coverage in H3 plans

This commit is contained in:
2026-08-27 07:16:25 +00:00
parent 7ad087f781
commit d24f9b262f
2 changed files with 38 additions and 15 deletions
+7 -3
View File
@@ -437,7 +437,7 @@ class DumasH3LongVideosHelperTests(unittest.TestCase):
module.dialogue_filler_warnings = lambda *_args, **_kwargs: []
module.distribute_generations = lambda anchor, beats, *_args, **_kwargs: [
f"[Generation 1] {anchor}. {beats[0]}",
f"[Generation 2] {anchor}. {beats[1]}",
f"[Generation 2] {anchor}. {beats[1]}{module.ANATOMY_STATE}",
]
module.continuity_warnings = lambda _gens: []
module._empty_av_latent = lambda *_args, **_kwargs: ({"samples": "latent"},)
@@ -459,13 +459,17 @@ class DumasH3LongVideosHelperTests(unittest.TestCase):
plan_only=True,
)
self.assertEqual(
self.assertIn(
"[Generation 1] editorial room, soft practical lighting. Francine stands alone.",
result[3],
"[Generation 1] editorial room, soft practical lighting. Francine stands alone.\n---\n"
)
self.assertIn(
"[Generation 2] editorial room, soft practical lighting. Francine and Frankie walk together.",
result[3],
)
self.assertIn("2 beat(s)", result[2])
self.assertIn("2 shot(s)", result[2])
self.assertIn("ANATOMY -- guard injected on shot(s) 2", result[2])
self.assertEqual(result[-2:], ([], []))
finally:
module.torch = original_torch