xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-08-23 06:04:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge a4be4c1a69e01779c7b2054d80e1959de4018003 into 4368d8f87f580f526e8b2bc43bf0ac88e2b67033

Browse Source
This commit is contained in:
Mitchell Ludwig 2025-09-03 17:09:20 -06:00 committed by GitHub
commit 1421f90ef0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/inference/test_execution.py
View File

@ -14,6 +14,10 @@ import urllib.request
import urllib.parse import urllib.parse
import urllib.error import urllib.error
from comfy_execution.graph_utils import GraphBuilder, Node from comfy_execution.graph_utils import GraphBuilder, Node
import os
import sys
PYTHON_EXECUTABLE_PATH = os.path.realpath(sys.executable)
def run_warmup(client, prefix="warmup"): def run_warmup(client, prefix="warmup"):
"""Run a simple workflow to warm up the server.""" """Run a simple workflow to warm up the server."""
@ -145,7 +149,7 @@ class TestExecution:
def _server(self, args_pytest, request): def _server(self, args_pytest, request):
# Start server # Start server
pargs = [ pargs = [
'python','main.py', PYTHON_EXECUTABLE_PATH,'main.py',
'--output-directory', args_pytest["output_dir"], '--output-directory', args_pytest["output_dir"],
'--listen', args_pytest["listen"], '--listen', args_pytest["listen"],
'--port', str(args_pytest["port"]), '--port', str(args_pytest["port"]),

5
tests/inference/test_inference.py
View File

@ -14,6 +14,8 @@ import websocket #NOTE: websocket-client (https://github.com/websocket-client/we
import uuid import uuid
import urllib.request import urllib.request
import urllib.parse import urllib.parse
import sys
from comfy.samplers import KSampler from comfy.samplers import KSampler
@ -21,6 +23,7 @@ from comfy.samplers import KSampler
""" """
These tests generate and save images through a range of parameters These tests generate and save images through a range of parameters
""" """
PYTHON_EXECUTABLE_PATH = os.path.realpath(sys.executable)
class ComfyGraph: class ComfyGraph:
def __init__(self, def __init__(self,
@ -152,7 +155,7 @@ class TestInference:
def _server(self, args_pytest): def _server(self, args_pytest):
# Start server # Start server
p = subprocess.Popen([ p = subprocess.Popen([
'python','main.py', PYTHON_EXECUTABLE_PATH,'main.py',
'--output-directory', args_pytest["output_dir"], '--output-directory', args_pytest["output_dir"],
'--listen', args_pytest["listen"], '--listen', args_pytest["listen"],
'--port', str(args_pytest["port"]), '--port', str(args_pytest["port"]),