mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-07-18 20:38:15 +08:00
50 lines
1.3 KiB
Python
50 lines
1.3 KiB
Python
"""Shared fixtures for the execution validation tests.
|
|
|
|
Forces CPU before any comfy import so the tests in this directory collect
|
|
and run on machines without CUDA/MPS; no GPU code path is exercised.
|
|
"""
|
|
import pytest
|
|
|
|
from comfy.cli_args import args as cli_args
|
|
|
|
cli_args.cpu = True
|
|
|
|
import nodes # noqa: E402
|
|
|
|
|
|
class _StubBase:
|
|
"""Minimal node base for stubs registered in NODE_CLASS_MAPPINGS."""
|
|
|
|
RETURN_TYPES = ()
|
|
FUNCTION = "run"
|
|
CATEGORY = "_test"
|
|
OUTPUT_NODE = False
|
|
|
|
def run(self, **kwargs):
|
|
"""Never executed; validation stops before node execution."""
|
|
return ()
|
|
|
|
|
|
def _make_stub(name, input_types, return_types=(), output_node=False, validate=None):
|
|
"""Build a legacy-style node class with the given validation surface."""
|
|
attrs = {
|
|
"RETURN_TYPES": return_types,
|
|
"OUTPUT_NODE": output_node,
|
|
"INPUT_TYPES": classmethod(lambda cls: input_types),
|
|
}
|
|
if validate is not None:
|
|
attrs["VALIDATE_INPUTS"] = classmethod(validate)
|
|
return type(name, (_StubBase,), attrs)
|
|
|
|
|
|
@pytest.fixture
|
|
def stub_nodes(monkeypatch):
|
|
"""Register stub node classes in nodes.NODE_CLASS_MAPPINGS; auto-clean."""
|
|
|
|
def register(name, **kwargs):
|
|
cls = _make_stub(name, **kwargs)
|
|
monkeypatch.setitem(nodes.NODE_CLASS_MAPPINGS, name, cls)
|
|
return cls
|
|
|
|
return register
|