IxMxAMAR
15e4b0da4e
Critical:
- rename nodes/ -> mp_nodes/, shared/ -> mp_shared/ to avoid shadowing
ComfyUI core's nodes.py via sys.path injection (root cause of v0.1.1
'nodes is not a package' startup failure)
- http_post now streams response and enforces 50MB cap mid-stream (DoS)
- AES key derivation: HKDF -> PBKDF2-HMAC-SHA256 600k iters + random salt
(work factor against weak-key brute force; ciphertext format changed)
- fs_mkdir/copy/move/delete confined to ComfyUI input/output/temp + ~ by
default; UTILITY_MEGAPACK_ALLOW_ARBITRARY_PATHS=1 to override
UX / correctness:
- combo settings now declare options array (themes dropdown was empty)
- LATENT default 1x4x8x8 zeros -> {} (avoids Flux/SD3 shape crashes)
- @op decorator rejects reserved widget names mode/theme at register time
- _pad_outputs docstring updated to match relaxed opaque-type behavior
107 lines
3.5 KiB
Python
107 lines
3.5 KiB
Python
"""Tests for the Text & Parsing category."""
|
|
|
|
import pytest
|
|
|
|
from mp_nodes.programming import ProgrammingNode
|
|
|
|
|
|
def run(mode, **kwargs):
|
|
return ProgrammingNode().process(mode=mode, theme="(use pack default)", **kwargs)
|
|
|
|
|
|
class TestRegex:
|
|
def test_match_true(self):
|
|
assert run("regex_match", text="hello world", pattern=r"world")[3] is True
|
|
|
|
def test_match_false(self):
|
|
assert run("regex_match", text="hello", pattern=r"world")[3] is False
|
|
|
|
def test_extract_no_match_returns_empty(self):
|
|
assert run("regex_extract", text="abc", pattern=r"\d+")[0] == ""
|
|
|
|
def test_extract_default_group_zero(self):
|
|
assert run("regex_extract", text="abc123def", pattern=r"\d+")[0] == "123"
|
|
|
|
def test_extract_named_group(self):
|
|
out = run("regex_extract", text="rev1.2.3", pattern=r"(\d+)\.(\d+)\.(\d+)", group=2)[0]
|
|
assert out == "2"
|
|
|
|
def test_extract_out_of_range_group(self):
|
|
# group 5 doesn't exist
|
|
assert run("regex_extract", text="abc", pattern=r"a(b)c", group=5)[0] == ""
|
|
|
|
def test_replace(self):
|
|
assert run("regex_replace", text="foo bar foo", pattern=r"foo", replacement="baz")[0] == "baz bar baz"
|
|
|
|
def test_split_default_whitespace(self):
|
|
assert run("regex_split", text="a b c", pattern=r"\s+")[4] == ["a", "b", "c"]
|
|
|
|
def test_split_max_splits(self):
|
|
out = run("regex_split", text="a,b,c,d", pattern=",", max_splits=2)[4]
|
|
assert out == ["a", "b", "c,d"]
|
|
|
|
|
|
class TestTemplate:
|
|
def test_basic_render(self):
|
|
out = run("template_render", template="hello {{name}}", vars_json='{"name": "world"}')[0]
|
|
assert out == "hello world"
|
|
|
|
def test_loops(self):
|
|
out = run(
|
|
"template_render",
|
|
template="{% for x in items %}{{x}},{% endfor %}",
|
|
vars_json='{"items": [1, 2, 3]}',
|
|
)[0]
|
|
assert out == "1,2,3,"
|
|
|
|
def test_strict_undefined_raises(self):
|
|
with pytest.raises(RuntimeError):
|
|
run("template_render", template="{{nope}}", vars_json="{}")
|
|
|
|
def test_invalid_vars_json_raises(self):
|
|
with pytest.raises(RuntimeError):
|
|
run("template_render", template="x", vars_json="not json")
|
|
|
|
|
|
class TestStringOps:
|
|
def test_concat(self):
|
|
assert run("string_concat", a="foo", b="bar", sep="-")[0] == "foo-bar"
|
|
|
|
def test_concat_no_sep(self):
|
|
assert run("string_concat", a="foo", b="bar", sep="")[0] == "foobar"
|
|
|
|
def test_upper(self):
|
|
assert run("string_upper", text="hello")[0] == "HELLO"
|
|
|
|
def test_lower(self):
|
|
assert run("string_lower", text="HELLO")[0] == "hello"
|
|
|
|
def test_trim(self):
|
|
assert run("string_trim", text=" hi ")[0] == "hi"
|
|
|
|
def test_tokenize_words(self):
|
|
assert run("tokenize_words", text="hello world foo")[4] == ["hello", "world", "foo"]
|
|
|
|
|
|
class TestYAML:
|
|
def test_parse_dict(self):
|
|
out = run("yaml_parse", text="a: 1\nb: 2\n")[5]
|
|
assert out == {"a": 1, "b": 2}
|
|
|
|
def test_parse_empty(self):
|
|
assert run("yaml_parse", text="")[5] == {}
|
|
|
|
def test_parse_scalar_wraps_under_value(self):
|
|
assert run("yaml_parse", text="42")[5] == {"value": 42}
|
|
|
|
def test_dump(self):
|
|
out = run("yaml_dump", data_json='{"a": 1}')[0]
|
|
assert "a: 1" in out
|
|
|
|
def test_dump_round_trip(self):
|
|
original = {"alpha": 1, "beta": [1, 2, 3]}
|
|
import json
|
|
dumped = run("yaml_dump", data_json=json.dumps(original))[0]
|
|
parsed = run("yaml_parse", text=dumped)[5]
|
|
assert parsed == original
|