| |
| """Modern Plan keeps the Plan contract while exposing no legacy controls.""" |
|
|
| import importlib.util |
| import json |
| import pathlib |
| import sys |
| import types |
|
|
|
|
| ROOT = pathlib.Path(__file__).resolve().parents[1] |
| PACKAGE = "h3_modern_plan_unit" |
|
|
| folder_paths = types.ModuleType("folder_paths") |
| folder_paths.get_output_directory = lambda: str(ROOT) |
| folder_paths.get_temp_directory = lambda: str(ROOT) |
| folder_paths.get_input_directory = lambda: str(ROOT) |
| folder_paths.get_annotated_filepath = lambda value: str(value) |
| sys.modules["folder_paths"] = folder_paths |
|
|
| package = types.ModuleType(PACKAGE) |
| package.__path__ = [str(ROOT)] |
| sys.modules[PACKAGE] = package |
|
|
| shared_nodes = types.ModuleType(PACKAGE + ".nodes") |
| shared_nodes.MiniMaxH3MotionContext = object |
| shared_nodes._claim_inline_patch_ownership = lambda _conditioning=None: "test" |
| shared_nodes._prepare_native_guide_conditioning = lambda value: value |
| shared_nodes._resize = lambda *args: None |
| shared_nodes._streams_from_latent = lambda *args: None |
| sys.modules[shared_nodes.__name__] = shared_nodes |
|
|
| spec = importlib.util.spec_from_file_location( |
| PACKAGE + ".chain_nodes", ROOT / "chain_nodes.py") |
| chain = importlib.util.module_from_spec(spec) |
| sys.modules[spec.name] = chain |
| spec.loader.exec_module(chain) |
|
|
|
|
| required = chain.MiniMaxH3ChainPlanModern.INPUT_TYPES()["required"] |
| optional = chain.MiniMaxH3ChainPlanModern.INPUT_TYPES()["optional"] |
| assert tuple(required) == ( |
| "plan_json", "chain_policy", "run_name", "generation_fingerprint", |
| "width", "height", "encode_mode", "crop", |
| "default_duration_seconds", "default_steps", "base_seed", |
| "segment_crf", "video_blend_frames", |
| ) |
| for removed in ( |
| "context_length", "anchor_mode", "audio_mode", "audio_context_length", |
| "continuation_mode", |
| ): |
| assert removed not in required |
| assert removed not in optional |
| assert required["chain_policy"][0] == chain.CHAIN_POLICY_TYPE |
| assert tuple(optional) == ("plan_json_input", "project_assets") |
| assert chain.MiniMaxH3ChainPlanModern.RETURN_TYPES == ( |
| chain.MiniMaxH3ChainPlan.RETURN_TYPES) |
| assert chain.MiniMaxH3ChainPlanModern.RETURN_NAMES == ( |
| chain.MiniMaxH3ChainPlan.RETURN_NAMES) |
| assert (chain.CHAIN_NODE_CLASS_MAPPINGS["MiniMaxH3ChainPlanModern"] |
| is chain.MiniMaxH3ChainPlanModern) |
|
|
| plan_json = json.dumps({ |
| "prompt_prefix": "Keep identity stable.", |
| "shots": [ |
| {"id": "one", "prompt": "Opening.", "length": 73}, |
| {"id": "two", "prompt": "Continue.", "length": 73}, |
| ], |
| }) |
| policy = chain.MiniMaxH3GenerationProfile().build( |
| "Visual continuity", "Generate audio")[0] |
| modern = chain.MiniMaxH3ChainPlanModern().build( |
| plan_json=plan_json, |
| chain_policy=policy, |
| run_name="modern-plan", |
| generation_fingerprint="model-a", |
| width=960, |
| height=544, |
| encode_mode="video", |
| crop="center", |
| default_duration_seconds=15.0, |
| default_steps=20, |
| base_seed=7, |
| segment_crf=18, |
| video_blend_frames=0, |
| ) |
| legacy = chain.MiniMaxH3ChainPlan().build( |
| plan_json=plan_json, |
| run_name="modern-plan", |
| generation_fingerprint="model-a", |
| width=960, |
| height=544, |
| context_length=22, |
| encode_mode="video", |
| anchor_mode="head", |
| crop="center", |
| audio_mode="generated_audio", |
| audio_context_length=22, |
| default_duration_seconds=15.0, |
| default_steps=20, |
| base_seed=7, |
| segment_crf=18, |
| video_blend_frames=0, |
| continuation_mode="guide", |
| chain_policy=policy, |
| ) |
| assert modern == legacy |
| assert modern[0]["compatibility"]["anchor_mode"] == "head" |
| assert modern[0]["compatibility"]["context_length"] == 22 |
| assert modern[0]["compatibility"]["audio_context_length"] == 22 |
|
|
| replacement_json = json.dumps({ |
| "shots": [{"id": "external", "prompt": "External.", "length": 22}], |
| }) |
| external = chain.MiniMaxH3ChainPlanModern().build( |
| plan_json=plan_json, |
| plan_json_input=replacement_json, |
| chain_policy=policy, |
| run_name="modern-plan", |
| generation_fingerprint="model-a", |
| width=960, |
| height=544, |
| encode_mode="video", |
| crop="disabled", |
| default_duration_seconds=15.0, |
| default_steps=20, |
| base_seed=7, |
| segment_crf=18, |
| ) |
| assert external[0]["shots"][0]["id"] == "external" |
|
|
| print("Modern Plan: clean inputs and legacy-equivalent Plan output pass") |
|
|
