Shaman
89 lines
3.2 KiB
Python
89 lines
3.2 KiB
Python
import pytest
|
|
|
|
from app.config import Settings
|
|
from app.errors import InvalidRequest
|
|
from app.injector import build_workflow, new_seed, parse_size
|
|
from app.templates import load_registry
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def registry():
|
|
return load_registry(Settings(_env_file=None))
|
|
|
|
|
|
def test_parse_size():
|
|
assert parse_size("1024x1024") == (1024, 1024)
|
|
assert parse_size("832X1216") == (832, 1216)
|
|
|
|
|
|
@pytest.mark.parametrize("bad", ["1024", "ax1024", "1024x", "0x100", "1024x-2", "wide"])
|
|
def test_parse_size_invalid(bad):
|
|
with pytest.raises(InvalidRequest):
|
|
parse_size(bad)
|
|
|
|
|
|
def test_seed_random_differs():
|
|
assert new_seed(None) != new_seed(None)
|
|
|
|
|
|
def test_seed_fixed_and_range():
|
|
assert new_seed(42) == 42
|
|
with pytest.raises(InvalidRequest):
|
|
new_seed(-1)
|
|
|
|
|
|
def test_create_injection_and_template_untouched(registry):
|
|
create = registry["z-image-turbo"].graph
|
|
graph = build_workflow(create, {"prompt": "a cat", "seed": 77, "negative": "", "width": 832, "height": 1216})
|
|
assert graph["83:27"]["inputs"]["text"] == "a cat"
|
|
assert graph["83:13"]["inputs"]["width"] == 832
|
|
assert graph["83:13"]["inputs"]["height"] == 1216
|
|
assert graph["83:3"]["inputs"]["seed"] == 77
|
|
# template still holds placeholders (deep copy, no mutation)
|
|
assert create["83:27"]["inputs"]["text"] == "{prompt}"
|
|
assert create["83:13"]["inputs"]["width"] == "{width}"
|
|
assert create["83:3"]["inputs"]["seed"] == "{seed}"
|
|
|
|
|
|
def test_injection_is_typed(registry):
|
|
graph = build_workflow(registry["z-image-turbo"].graph, {"prompt": "x", "seed": 7, "negative": "", "width": 64, "height": 128})
|
|
assert isinstance(graph["83:3"]["inputs"]["seed"], int) # numeric token -> int, not "{seed}"
|
|
assert isinstance(graph["83:13"]["inputs"]["width"], int)
|
|
assert isinstance(graph["83:27"]["inputs"]["text"], str)
|
|
|
|
|
|
def test_embedded_token_is_string_interpolated():
|
|
graph = {"1": {"class_type": "X", "inputs": {"text": "prefix {prompt} suffix"}}}
|
|
out = build_workflow(graph, {"prompt": "hi"})
|
|
assert out["1"]["inputs"]["text"] == "prefix hi suffix"
|
|
|
|
|
|
def test_edit_injection(registry):
|
|
edit = registry["qwen-image-edit"].graph
|
|
graph = build_workflow(edit, {"prompt": "make it night", "seed": 5, "negative": "", "image": "adapter_x.png"})
|
|
assert graph["115:111"]["inputs"]["prompt"] == "make it night"
|
|
assert graph["115:110"]["inputs"]["prompt"] == ""
|
|
assert graph["78"]["inputs"]["image"] == "adapter_x.png"
|
|
assert graph["115:3"]["inputs"]["seed"] == 5
|
|
|
|
|
|
def test_edit_seed_varies_per_call(registry):
|
|
edit = registry["qwen-image-edit"].graph
|
|
seeds = set()
|
|
for _ in range(3):
|
|
g = build_workflow(edit, {"prompt": "p", "seed": new_seed(None), "negative": "", "image": "a.png"})
|
|
seeds.add(g["115:3"]["inputs"]["seed"])
|
|
assert len(seeds) == 3
|
|
|
|
|
|
def test_unrecognized_token_rejected():
|
|
graph = {"1": {"class_type": "X", "inputs": {"text": "{foo}"}}}
|
|
with pytest.raises(InvalidRequest, match="unrecognized token"):
|
|
build_workflow(graph, {"prompt": "p"})
|
|
|
|
|
|
def test_unfilled_token_rejected():
|
|
graph = {"1": {"class_type": "X", "inputs": {"seed": "{seed}"}}}
|
|
with pytest.raises(InvalidRequest, match="no value supplied"):
|
|
build_workflow(graph, {"prompt": "p"})
|