xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-08-28 03:45:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

Using the same python executable to run inference tests, rather than system python

Browse Source
This commit is contained in:
mitchell.ludwig 2025-06-14 21:27:55 -06:00
parent 53e8d8193c
commit a4be4c1a69
3 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/inference/extra_model_paths.yaml
View File

@ -1,4 +1,4 @@
# Config for testing nodes # Config for testing nodes
testing: testing:
custom_nodes: tests/inference/testing_nodes custom_nodes: testing_nodes

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)
class RunResult: class RunResult:
def __init__(self, prompt_id: str): def __init__(self, prompt_id: str):
@ -125,7 +129,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"]),