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

[refactor] Clean up ImageStitch tests

Browse Source 
- Remove unnecessary sys.path manipulation (pythonpath set in pytest.ini)
- Remove metadata tests that test framework internals rather than functionality
- Rename complex scenario test to be more descriptive of what it tests
This commit is contained in:
bymyself 2025-06-01 00:46:12 -07:00
parent b8712f2494
commit fcbb1a7ff7
1 changed files with 2 additions and 38 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
tests-unit/comfy_extras_test/image_stitch_test.py
View File

@ -1,11 +1,6 @@
import torch
import sys
import os
from unittest.mock import patch, MagicMock
# Add the project root to Python path
sys.path.insert(0, os.path.join(os.path.dirname(__file__), '../..'))
# Mock nodes module to prevent CUDA initialization during import
mock_nodes = MagicMock()
mock_nodes.MAX_RESOLUTION = 16384
@ -226,8 +221,8 @@ class TestImageStitch:
result = node.stitch(image1, direction, False, 0, "white", image2)
assert result[0].shape == (1, 32, 64, 3) if direction in ["right", "left"] else (1, 64, 32, 3)
def test_complex_scenario(self):
"""Test complex scenario with all features enabled"""
def test_batch_size_channel_border_integration(self):
"""Test integration of batch matching, channel matching, size matching, and borders"""
node = ImageStitch()
image1 = self.create_test_image(batch_size=2, height=64, width=48, channels=3)
image2 = self.create_test_image(batch_size=1, height=32, width=32, channels=4)
@ -243,34 +238,3 @@ class TestImageStitch:
expected_total_width = 48 + 8 + expected_image2_width
assert result[0].shape[2] == expected_total_width
def test_input_types_structure(self):
"""Test that INPUT_TYPES returns correct structure"""
input_types = ImageStitch.INPUT_TYPES()
assert "required" in input_types
assert "optional" in input_types
required = input_types["required"]
assert "image1" in required
assert "direction" in required
assert "match_image_size" in required
assert "border_width" in required
assert "border_color" in required
optional = input_types["optional"]
assert "image2" in optional
# Check direction options
directions = required["direction"][0]
assert set(directions) == {"right", "down", "left", "up"}
# Check color options
colors = required["border_color"][0]
assert set(colors) == {"white", "black", "red", "green", "blue"}
def test_node_metadata(self):
"""Test node metadata and properties"""
assert ImageStitch.RETURN_TYPES == ("IMAGE",)
assert ImageStitch.FUNCTION == "stitch"
assert ImageStitch.CATEGORY == "image/transform"
assert hasattr(ImageStitch, "DESCRIPTION")