40 lines
1.1 KiB
Python
40 lines
1.1 KiB
Python
# SimpleSyrup - workflow-focused ComfyUI extensions for image generation
|
|
# Copyright (C) 2026 Artificial Sweetener and contributors
|
|
# SPDX-License-Identifier: AGPL-3.0-or-later
|
|
|
|
"""Pytest configuration for SimpleSyrup tests."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
CUDA_TESTS_ENABLED = os.environ.get("SIMPLE_SYRUP_TEST_CUDA") == "1"
|
|
if not CUDA_TESTS_ENABLED:
|
|
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
|
|
|
|
PROJECT_ROOT = Path(__file__).resolve().parents[1]
|
|
TESTS_ROOT = PROJECT_ROOT / "tests"
|
|
COMFY_ROOT = PROJECT_ROOT.parents[1]
|
|
CUSTOM_NODES_ROOT = PROJECT_ROOT.parent
|
|
|
|
for path in reversed((TESTS_ROOT, PROJECT_ROOT, COMFY_ROOT, CUSTOM_NODES_ROOT)):
|
|
path_text = str(path)
|
|
if path_text not in sys.path:
|
|
sys.path.insert(0, path_text)
|
|
|
|
if not CUDA_TESTS_ENABLED:
|
|
from comfy.cli_args import args
|
|
|
|
args.cpu = True
|
|
|
|
|
|
def pytest_xdist_auto_num_workers(config: pytest.Config) -> int:
|
|
"""Bound Windows collection concurrency for the installed Comfy runtime."""
|
|
|
|
del config
|
|
return 4
|