86 lines
2.9 KiB
Python
86 lines
2.9 KiB
Python
# SimpleSyrup - workflow-focused ComfyUI extensions for image generation
|
|
# Copyright (C) 2026 Artificial Sweetener and contributors
|
|
# SPDX-License-Identifier: AGPL-3.0-or-later
|
|
|
|
"""Tests for the Simple Preview SEGS ComfyUI node."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import ClassVar
|
|
|
|
import torch
|
|
from pytest import MonkeyPatch
|
|
|
|
from simple_syrup.nodes.simple_preview_segs import SimplePreviewSEGS
|
|
from simple_syrup.runtime.seg_preview_assets import SegPreviewPublication
|
|
from simple_syrup.services.simple_preview_segs_service import SegPreviewDocument
|
|
|
|
|
|
def test_node_declares_terminal_passthrough_preview_contract() -> None:
|
|
"""The inspector remains insertable inline and executable as a terminal node."""
|
|
|
|
inputs = SimplePreviewSEGS.INPUT_TYPES()["required"]
|
|
|
|
assert tuple(inputs) == ("image", "segs")
|
|
assert SimplePreviewSEGS.RETURN_TYPES == ("SEGS",)
|
|
assert SimplePreviewSEGS.RETURN_NAMES == ("segs",)
|
|
assert SimplePreviewSEGS.OUTPUT_NODE is True
|
|
|
|
|
|
def test_node_publishes_manifest_and_returns_original_segs_object(
|
|
monkeypatch: MonkeyPatch,
|
|
) -> None:
|
|
"""Preview transport never replaces or mutates the connected SEGS payload."""
|
|
|
|
segs: object = ((4, 5), [])
|
|
document = SegPreviewDocument(
|
|
source_width=5,
|
|
source_height=4,
|
|
preview_width=5,
|
|
preview_height=4,
|
|
image=torch.zeros((1, 4, 5, 3)),
|
|
atlas=torch.zeros((1, 1, 1, 3)),
|
|
region_images=(),
|
|
regions=(),
|
|
)
|
|
|
|
class _Service:
|
|
"""Return the fixed preview document."""
|
|
|
|
def build(self, *, image: object, segs: object) -> SegPreviewDocument:
|
|
"""Validate inputs before returning the document."""
|
|
|
|
assert isinstance(image, torch.Tensor)
|
|
assert segs == ((4, 5), ())
|
|
return document
|
|
|
|
class _Publisher:
|
|
"""Return one deterministic frontend manifest."""
|
|
|
|
published: ClassVar[list[SegPreviewDocument]] = []
|
|
|
|
def publish(self, value: SegPreviewDocument) -> SegPreviewPublication:
|
|
"""Record and publish the document."""
|
|
|
|
self.published.append(value)
|
|
return SegPreviewPublication(
|
|
manifest={"version": 1, "regions": []},
|
|
images=({"filename": "region.png", "subfolder": "", "type": "temp"},),
|
|
)
|
|
|
|
node = SimplePreviewSEGS()
|
|
monkeypatch.setattr(SimplePreviewSEGS, "service_class", _Service)
|
|
monkeypatch.setattr(SimplePreviewSEGS, "publisher_class", _Publisher)
|
|
|
|
result = node.preview(image=torch.zeros((1, 4, 5, 3)), segs=segs)
|
|
|
|
node_result = result["result"]
|
|
assert isinstance(node_result, tuple)
|
|
assert node_result == (segs,)
|
|
assert node_result[0] is segs
|
|
assert result["ui"] == {
|
|
"images": [{"filename": "region.png", "subfolder": "", "type": "temp"}],
|
|
"simple_syrup_segs_preview": [{"version": 1, "regions": []}],
|
|
}
|
|
assert _Publisher.published == [document]
|