fix(governance): enforce SugarSubstitute quality standards
This commit is contained in:
@@ -0,0 +1,78 @@
|
||||
# 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
|
||||
Reference in New Issue
Block a user