mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-07-18 20:38:15 +08:00
61 lines
2.0 KiB
Python
61 lines
2.0 KiB
Python
import subprocess
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
|
|
ROOT = Path(__file__).resolve().parents[2]
|
|
IMPORT_CLI = "import comfy.options; comfy.options.enable_args_parsing(); import comfy.cli_args"
|
|
PRINT_CONTINUOUS_BATCHING = IMPORT_CLI + "; print(comfy.cli_args.args.continuous_batching)"
|
|
PRINT_DYNAMIC_VRAM = IMPORT_CLI + "; print(comfy.cli_args.enables_dynamic_vram())"
|
|
|
|
|
|
def run_cli(*arguments, script=IMPORT_CLI):
|
|
return subprocess.run(
|
|
[sys.executable, "-c", script, *arguments],
|
|
cwd=ROOT,
|
|
capture_output=True,
|
|
text=True,
|
|
check=False,
|
|
)
|
|
|
|
|
|
def test_continuous_batching_defaults_to_disabled():
|
|
result = run_cli(script=PRINT_CONTINUOUS_BATCHING)
|
|
assert result.returncode == 0
|
|
assert result.stdout.strip() == "0"
|
|
|
|
|
|
def test_continuous_batching_stores_the_prompt_limit():
|
|
result = run_cli("--continuous-batching", "4", script=PRINT_CONTINUOUS_BATCHING)
|
|
assert result.returncode == 0
|
|
assert result.stdout.strip() == "4"
|
|
|
|
|
|
def test_continuous_batching_rejects_negative_limit():
|
|
result = run_cli("--continuous-batching", "-1")
|
|
assert result.returncode == 2
|
|
assert "must be zero or greater" in result.stderr
|
|
|
|
|
|
def test_multi_prompt_continuous_batching_rejects_cache_none():
|
|
result = run_cli("--continuous-batching", "2", "--cache-none")
|
|
assert result.returncode == 2
|
|
assert "incompatible with --cache-none" in result.stderr
|
|
|
|
|
|
def test_single_prompt_continuous_batching_allows_cache_none():
|
|
result = run_cli("--continuous-batching", "1", "--cache-none")
|
|
assert result.returncode == 0
|
|
|
|
|
|
def test_continuous_batching_disables_automatic_dynamic_vram():
|
|
result = run_cli("--continuous-batching", "1", script=PRINT_DYNAMIC_VRAM)
|
|
assert result.returncode == 0
|
|
assert result.stdout.strip() == "False"
|
|
|
|
|
|
def test_continuous_batching_rejects_explicit_dynamic_vram():
|
|
result = run_cli("--continuous-batching", "2", "--enable-dynamic-vram")
|
|
assert result.returncode == 2
|
|
assert "incompatible with --enable-dynamic-vram" in result.stderr
|