49 lines
1.9 KiB
Python
49 lines
1.9 KiB
Python
import os
|
|
import sys
|
|
|
|
_here = os.path.abspath(os.path.dirname(__file__))
|
|
_project_root = os.path.abspath(os.path.join(_here, os.pardir))
|
|
_comfy_root = os.path.abspath(os.path.join(_project_root, os.pardir, os.pardir))
|
|
|
|
# Add project root and ComfyUI root to sys.path
|
|
# This ensures import comfy_api and import comfy work.
|
|
for p in [_project_root, _comfy_root]:
|
|
if p not in sys.path:
|
|
sys.path.insert(0, p)
|
|
# also make it visible to subprocesses that inspect PYTHONPATH
|
|
os.environ["PYTHONPATH"] = p + os.pathsep + os.environ.get("PYTHONPATH", "")
|
|
|
|
# Special case: 'more_math' root folder has an __init__.py which conflicts with the 'more_math' subfolder.
|
|
# We want 'import more_math' to resolve to the subfolder package.
|
|
if "more_math" not in sys.modules or "custom_nodes" in (getattr(sys.modules["more_math"], "__file__", "") or ""):
|
|
import importlib.util
|
|
_subfolder = os.path.join(_project_root, "more_math")
|
|
_spec = importlib.util.spec_from_file_location("more_math", os.path.join(_subfolder, "__init__.py"))
|
|
if _spec:
|
|
_m = importlib.util.module_from_spec(_spec)
|
|
sys.modules["more_math"] = _m
|
|
_spec.loader.exec_module(_m)
|
|
|
|
# Redirect comfy_api.Model to comfy_api.latest.io.Model if needed
|
|
# as requested by user to support the 'latest' API position.
|
|
try:
|
|
import comfy_api
|
|
import comfy_api.latest as latest
|
|
if not hasattr(comfy_api, "Model") and hasattr(latest, "io") and hasattr(latest.io, "Model"):
|
|
comfy_api.Model = latest.io.Model
|
|
comfy_api.io = latest.io
|
|
except (ImportError, AttributeError):
|
|
pass
|
|
|
|
import pytest
|
|
|
|
|
|
@pytest.fixture(scope="session", autouse=True)
|
|
def session_setup():
|
|
"""
|
|
Session-scoped autouse fixture kept for future per-session initialization.
|
|
Top-level sys.path modification already runs during collection, so this fixture
|
|
ensures any further session setup can be added without changing tests.
|
|
"""
|
|
yield
|