nodes/ComfyAssets-ComfyUI_Selectors
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

test: Add new test case for WidthHeightNode dimensions

Browse Source
This commit is contained in:
Vito Sansevero
2025-06-13 08:32:17 -07:00
parent 96a2c6c85c
commit a608eafe37
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
run_tests.py
+4 -2
View File
@@ -118,10 +118,12 @@ def test_all_nodes():
wh_node = WidthHeightNode()
result = wh_node.get_dimensions(512, 768, "custom", False)
assert result == (512, 768), f"Expected (512, 768), got {result}"
result = wh_node.get_dimensions(512, 768, "1024x768", False)
assert result == (1024, 768), f"Expected (1024, 768), got {result}"
result = wh_node.get_dimensions(512, 768, "1024x1024", False)
assert result == (1024, 1024), f"Expected (1024, 1024), got {result}"
result = wh_node.get_dimensions(512, 768, "custom", True)
assert result == (768, 512), f"Expected (768, 512), got {result}"
result = wh_node.get_dimensions(512, 768, "1152x896", True)
assert result == (896, 1152), f"Expected (896, 1152), got {result}"
print(" ✅ WidthHeightNode works")
print("\n🎉 All tests passed!")