f5908006eb
Add nodes/category_splitter_node.py: CategorySplitterNode, a thin adapter that
delegates to split_into_categories (MASTER_SPEC §20).
- Input resolved_tags (RESOLVED_TAGS); outputs (category_map, warnings, errors):
the CATEGORY_MAP (per-category grouping, correct assignment + order) plus
surfaced diagnostics. No compiler logic in the node.
- The per-category prompt strings are produced downstream by the Prompt Builder
node (§19); this node produces the intermediate Category Map that feeds it.
- Registered as SceneCompilerCategorySplitter ("Category Splitter").
Tests: metadata/inputs, real split into a CategoryMap, delegation + SC0008
surfacing (monkeypatched), and registration.
Co-authored-by: kunail0804 <kunail0804@gmail.com>
Co-authored-by: Claude Opus 4.8 <noreply@anthropic.com>
46 lines
1.8 KiB
Python
46 lines
1.8 KiB
Python
"""Tests for the Category Splitter ComfyUI node (issue #22)."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import nodes
|
|
import nodes.category_splitter_node as node_module
|
|
from compiler.common.message_codes import message
|
|
from compiler.common.result import CompilerResult
|
|
from nodes.category_splitter_node import CategorySplitterNode
|
|
from schemas.models import CategoryMap, ResolvedTag
|
|
|
|
|
|
def rtag(name: str, category: str) -> ResolvedTag:
|
|
return ResolvedTag(tag=name, category=category, source_concept=name, knowledge_base_entry=name)
|
|
|
|
|
|
def test_node_metadata() -> None:
|
|
assert CategorySplitterNode.RETURN_TYPES == ("CATEGORY_MAP", "STRING", "STRING")
|
|
assert CategorySplitterNode.RETURN_NAMES == ("category_map", "warnings", "errors")
|
|
assert "resolved_tags" in CategorySplitterNode.INPUT_TYPES()["required"]
|
|
|
|
|
|
def test_splits_with_real_module() -> None:
|
|
tags = (rtag("1girl", "character"), rtag("long hair", "hair"))
|
|
category_map, warnings, errors = CategorySplitterNode().run(tags)
|
|
assert isinstance(category_map, CategoryMap)
|
|
assert [t.tag for t in category_map.tags_for("character")] == ["1girl"]
|
|
assert [t.tag for t in category_map.tags_for("hair")] == ["long hair"]
|
|
assert (warnings, errors) == ("", "")
|
|
|
|
|
|
def test_delegates_and_surfaces_errors(monkeypatch) -> None:
|
|
monkeypatch.setattr(
|
|
node_module,
|
|
"split_into_categories",
|
|
lambda tags: CompilerResult().add_error(message("SC0008", "bad category")),
|
|
)
|
|
category_map, warnings, errors = CategorySplitterNode().run(())
|
|
assert category_map is None
|
|
assert "SC0008: bad category" in errors
|
|
|
|
|
|
def test_registered_for_comfyui() -> None:
|
|
assert nodes.NODE_CLASS_MAPPINGS["SceneCompilerCategorySplitter"] is CategorySplitterNode
|
|
assert nodes.NODE_DISPLAY_NAME_MAPPINGS["SceneCompilerCategorySplitter"] == "Category Splitter"
|