Katsuhide Tsuruta
2c52b46f98
- 20 tests covering node registration, loader, decode, encode, and model wrapper - conftest.py mocks ComfyUI-specific modules (folder_paths, comfy.model_management) - Use absolute import in __init__.py for test compatibility - Setup uv for dependency management
21 lines
745 B
Python
21 lines
745 B
Python
"""Pre-mock ComfyUI modules before any imports happen."""
|
|
import sys
|
|
import types
|
|
|
|
# Mock ComfyUI-specific modules that don't exist outside ComfyUI
|
|
folder_paths = types.ModuleType("folder_paths")
|
|
folder_paths.models_dir = "/tmp/comfyui_test_models"
|
|
|
|
comfy = types.ModuleType("comfy")
|
|
comfy_mm = types.ModuleType("comfy.model_management")
|
|
comfy_mm.vae_device = lambda: __import__("torch").device("cpu")
|
|
comfy_mm.vae_dtype = lambda device=None: __import__("torch").float32
|
|
comfy_mm.vae_offload_device = lambda: __import__("torch").device("cpu")
|
|
comfy.model_management = comfy_mm
|
|
|
|
sys.modules["folder_paths"] = folder_paths
|
|
sys.modules["comfy"] = comfy
|
|
sys.modules["comfy.model_management"] = comfy_mm
|
|
|
|
collect_ignore = ["__init__.py", "nodes.py"]
|