15 lines
498 B
Python
15 lines
498 B
Python
import pytest
|
|
from mjr_am_backend.features.metadata.parsing_utils import looks_like_comfyui_prompt_graph
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_prompt_graph_accepts_colon_ids():
|
|
prompt_graph = {
|
|
"91:1": {"class_type": "LoadImage", "inputs": {}},
|
|
"92:2": {"class_type": "TextEncode", "inputs": {}},
|
|
"93:3": {"class_type": "KSampler", "inputs": {}},
|
|
"94:4": {"class_type": "SaveImage", "inputs": {}},
|
|
}
|
|
|
|
assert looks_like_comfyui_prompt_graph(prompt_graph)
|