Claude
dc4e51563c
Co-Authored-By: Claude Sonnet 5.5 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_016acRss69eH1235eRdFTZHY
28 lines
1.2 KiB
Python
28 lines
1.2 KiB
Python
from core import legacy
|
|
|
|
|
|
def test_old_node_ids_and_keys_are_renamed():
|
|
d = {"slots": [{"node": "FunPackSampler", "key": "funpack.clock"}], "conditioning_slot": "funpack"}
|
|
assert legacy.data(d) == {"slots": [{"node": "FunUISampler", "key": "funui.clock"}], "conditioning_slot": "funui"}
|
|
|
|
|
|
def test_prose_and_other_packs_are_left_alone():
|
|
assert legacy.name("FunPack is nice") == "FunPack is nice"
|
|
assert legacy.name("funpackage") == "funpackage"
|
|
assert legacy.name("KSampler") == "KSampler"
|
|
|
|
|
|
def test_a_project_saved_under_the_old_name_opens_with_the_new_one():
|
|
from core import projects
|
|
old = {"name": "Old", "conditioning_slot": "funpack", "sampler_slot": "funpack",
|
|
"models": {"slots": [{"id": "sampler", "node": "FunPackSampler", "inputs": {"seed": 1}}]}}
|
|
p = projects.Project.from_dict(old)
|
|
assert p.conditioning_slot == "funui" and p.sampler_slot == "funui"
|
|
assert p.models["slots"][0]["node"] == "FunUISampler"
|
|
|
|
|
|
def test_an_imported_workflow_with_old_node_ids_is_read_with_the_new_ones():
|
|
from core import workflow_import
|
|
wf = {"1": {"class_type": "FunPackSampler", "inputs": {"seed": 1}}}
|
|
assert workflow_import._from_api(legacy.data(wf), [])[0]["node"] == "FunUISampler"
|