wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-23 12:50:18 +08:00
Code Issues Actions 15 Packages Projects Releases Wiki Activity

Only run this test with CUDA. Fix CPU mode for testing later

Browse Source
This commit is contained in:
doctorpangloss 2024-05-08 17:03:45 -07:00
parent 464c132c50
commit d6c374942e
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/unit/test_openapi_nodes.py
View File

@ -11,7 +11,6 @@ from PIL import Image
from freezegun import freeze_time from freezegun import freeze_time
from comfy.cmd import folder_paths from comfy.cmd import folder_paths
from comfy.nodes.base_nodes import LoadImage
from comfy_extras.nodes.nodes_open_api import SaveImagesResponse, IntRequestParameter, FloatRequestParameter, \ from comfy_extras.nodes.nodes_open_api import SaveImagesResponse, IntRequestParameter, FloatRequestParameter, \
StringRequestParameter, HashImage, StringPosixPathJoin, LegacyOutputURIs, DevNullUris, StringJoin, StringToUri, \ StringRequestParameter, HashImage, StringPosixPathJoin, LegacyOutputURIs, DevNullUris, StringJoin, StringToUri, \
UriFormat, ImageExifMerge, ImageExifCreationDateAndBatchNumber, ImageExif, ImageExifUncommon, \ UriFormat, ImageExifMerge, ImageExifCreationDateAndBatchNumber, ImageExif, ImageExifUncommon, \
@ -257,6 +256,7 @@ def test_posix_join_curly_brackets():
assert joined_path == "a_{test}/b/c" assert joined_path == "a_{test}/b/c"
@pytest.mark.skipif(not torch.cuda.is_available(), reason="CUDA is not available")
def test_file_request_parameter(use_temporary_input_directory): def test_file_request_parameter(use_temporary_input_directory):
_image_1x1_px = np.array([[[255, 0, 0]]], dtype=np.uint8) _image_1x1_px = np.array([[[255, 0, 0]]], dtype=np.uint8)
image_path = os.path.join(use_temporary_input_directory, "test_image.png") image_path = os.path.join(use_temporary_input_directory, "test_image.png")
@ -266,6 +266,7 @@ def test_file_request_parameter(use_temporary_input_directory):
n = ImageRequestParameter() n = ImageRequestParameter()
loaded_image, = n.execute(uri=image_path) loaded_image, = n.execute(uri=image_path)
assert loaded_image.shape == (1, 1, 1, 3) assert loaded_image.shape == (1, 1, 1, 3)
from comfy.nodes.base_nodes import LoadImage
load_image_node = LoadImage() load_image_node = LoadImage()
load_image_node_rgb, _ = load_image_node.load_image(image=os.path.basename(image_path)) load_image_node_rgb, _ = load_image_node.load_image(image=os.path.basename(image_path))