Artificial Sweetener
f863fcf645
Add VAE encode/decode option nodes across legacy and Comfy v3 exports, with tooltip and registration coverage. Preserve cloned model graph composition for differential diffusion in tiled and regional sampling paths.
79 lines
2.7 KiB
Python
79 lines
2.7 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 VAE Encode (Options) Comfy v3 wrapper."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from importlib import import_module
|
|
|
|
from simple_syrup.nodes import tooltips
|
|
from simple_syrup.nodes.vae_options import (
|
|
VAE_ENCODE_TILE_SIZE_STEP,
|
|
VAE_OVERLAP_DEFAULT,
|
|
VAE_TEMPORAL_OVERLAP_DEFAULT,
|
|
VAE_TEMPORAL_SIZE_DEFAULT,
|
|
VAE_TILE_SIZE_DEFAULT,
|
|
VAEEncodeOptions,
|
|
)
|
|
from simple_syrup.nodes_v3.vae_encode_options import VAEEncodeOptionsV3
|
|
|
|
|
|
def test_vae_encode_options_v3_schema() -> None:
|
|
"""VAE Encode (Options) v3 schema mirrors the legacy node contract."""
|
|
|
|
schema = VAEEncodeOptionsV3.define_schema()
|
|
|
|
assert schema.node_id == "SimpleSyrup.VAEEncodeOptions"
|
|
assert schema.display_name == "VAE Encode (Options)"
|
|
assert schema.category == "SimpleSyrup/Latent"
|
|
assert schema.description == VAEEncodeOptions.DESCRIPTION
|
|
assert schema.enable_expand is True
|
|
assert [input_item.id for input_item in schema.inputs] == [
|
|
"use_tiling",
|
|
"pixels",
|
|
"vae",
|
|
"tile_size",
|
|
"overlap",
|
|
"temporal_size",
|
|
"temporal_overlap",
|
|
]
|
|
assert schema.inputs[0].io_type == "BOOLEAN"
|
|
assert schema.inputs[0].default is False
|
|
assert schema.inputs[0].tooltip == tooltips.VAE_OPTIONS_USE_TILING
|
|
assert schema.inputs[1].io_type == "IMAGE"
|
|
assert schema.inputs[1].rawLink is True
|
|
assert schema.inputs[2].io_type == "VAE"
|
|
assert schema.inputs[2].rawLink is True
|
|
assert schema.inputs[3].default == VAE_TILE_SIZE_DEFAULT
|
|
assert schema.inputs[3].step == VAE_ENCODE_TILE_SIZE_STEP
|
|
assert schema.inputs[4].default == VAE_OVERLAP_DEFAULT
|
|
assert schema.inputs[5].default == VAE_TEMPORAL_SIZE_DEFAULT
|
|
assert schema.inputs[6].default == VAE_TEMPORAL_OVERLAP_DEFAULT
|
|
assert [output.id for output in schema.outputs] == ["latent"]
|
|
assert schema.outputs[0].io_type == "LATENT"
|
|
assert schema.outputs[0].tooltip == tooltips.VAE_OPTIONS_LATENT_OUTPUT
|
|
|
|
|
|
def test_vae_encode_options_v3_execute_delegates_to_legacy_node() -> None:
|
|
"""VAE Encode (Options) v3 execution returns legacy expansion behavior."""
|
|
|
|
graph_utils = import_module("comfy_execution.graph_utils")
|
|
graph_utils.GraphBuilder.set_default_prefix("V3_ENCODE", 0, 0)
|
|
|
|
result = VAEEncodeOptionsV3.execute(
|
|
True,
|
|
["image", 0],
|
|
["loader", 2],
|
|
768,
|
|
96,
|
|
48,
|
|
12,
|
|
)
|
|
node = next(iter(result["expand"].values()))
|
|
|
|
assert node["class_type"] == "VAEEncodeTiled"
|
|
assert node["inputs"]["tile_size"] == 768
|
|
assert result["result"][0][1] == 0
|