Vito Sansevero
38f820c8f5
- VALIDATE_INPUTS lets unknown family labels reach create() - no size is listed under two chips; SD 1.5 3:4 gets true 3:4 sizes - model change before the families payload loads is applied on load - custom ratio: clear error on chip click, cap sides at 8192, accept X - real-host assertions for model change and wired-in state
75 lines
2.6 KiB
Python
75 lines
2.6 KiB
Python
import asyncio
|
|
import json
|
|
import logging
|
|
from unittest import mock
|
|
|
|
from kikotools.tools.model_resolution_picker import families
|
|
from kikotools.tools.model_resolution_picker.node import (
|
|
ModelResolutionPickerNode,
|
|
get_families,
|
|
)
|
|
|
|
|
|
def test_inputs_and_outputs():
|
|
inputs = ModelResolutionPickerNode.INPUT_TYPES()["required"]
|
|
assert inputs["model"][1]["default"] == "SDXL"
|
|
assert "HunyuanImage 2.1" in inputs["model"][0]
|
|
width = inputs["width"][1]
|
|
assert (width["default"], width["min"], width["max"], width["step"]) == (
|
|
1024,
|
|
64,
|
|
8192,
|
|
8,
|
|
)
|
|
assert ModelResolutionPickerNode.RETURN_TYPES == ("LATENT", "INT", "INT")
|
|
assert ModelResolutionPickerNode.CATEGORY == "🫶 ComfyAssets/🖼️ Resolution"
|
|
|
|
|
|
def test_execution_rounds_to_family_step_and_marks_latent():
|
|
latent, w, h = ModelResolutionPickerNode().create("HunyuanImage 2.1", 70, 1000, 2)
|
|
assert (w, h) == (64, 992)
|
|
assert latent["samples"].shape == (2, 4, 124, 8)
|
|
assert latent["downscale_ratio_spacial"] == 8
|
|
|
|
|
|
def test_exact_sizes_pass_through():
|
|
_, w, h = ModelResolutionPickerNode().create("Flux.2 / Klein / MOSS", 1248, 832, 1)
|
|
assert (w, h) == (1248, 832)
|
|
|
|
|
|
def test_unknown_label_keeps_size_and_warns(caplog):
|
|
with caplog.at_level(logging.WARNING):
|
|
_, w, h = ModelResolutionPickerNode().create("Renamed Family", 1003, 1003, 1)
|
|
assert (w, h) == (1000, 1000) # fallback step 8
|
|
assert "Renamed Family" in caplog.text
|
|
|
|
|
|
def test_validate_inputs_accepts_any_label():
|
|
assert ModelResolutionPickerNode.VALIDATE_INPUTS("Renamed Family") is True
|
|
|
|
|
|
def test_alias_label_resolves_to_its_family():
|
|
sdxl = families.resolve_family("SDXL")
|
|
patched = sdxl._replace(aliases=("SDXL 1.0",))
|
|
others = tuple(f for f in families.FAMILIES if f.label != "SDXL")
|
|
with mock.patch.object(families, "FAMILIES", (patched,) + others):
|
|
assert families.resolve_family("SDXL 1.0").label == "SDXL"
|
|
latent, w, h = ModelResolutionPickerNode().create("SDXL 1.0", 1216, 832, 1)
|
|
assert (w, h) == (1216, 832)
|
|
assert latent["samples"].shape[-2:] == (104, 152)
|
|
|
|
|
|
def test_families_route_returns_payload():
|
|
response = asyncio.run(get_families(None))
|
|
assert response.status == 200
|
|
assert json.loads(response.text)["default"] == "SDXL"
|
|
|
|
|
|
def test_registered_in_kikotools():
|
|
from kikotools import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
|
|
|
|
assert NODE_CLASS_MAPPINGS["ModelResolutionPicker"] is ModelResolutionPickerNode
|
|
assert (
|
|
NODE_DISPLAY_NAME_MAPPINGS["ModelResolutionPicker"] == "Model Resolution Picker"
|
|
)
|