42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
import sys
|
|
import os
|
|
|
|
# Add project root to path
|
|
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
|
|
|
|
from utils.presets import resolve_resolution
|
|
|
|
|
|
def test_resolve():
|
|
print("Testing resolve_resolution...")
|
|
|
|
# Case 1: Standard Preset
|
|
w, h = resolve_resolution("Qwen - 1:1 (1328x1328)", 0, "1:1", "qwen-image")
|
|
print(f"Case 1 (Preset 1328x1328): {w}x{h}")
|
|
assert w == 1328 and h == 1328
|
|
|
|
# Case 2: Custom Override (Qwen uses 28)
|
|
w, h = resolve_resolution("Qwen - 1:1 (1328x1328)", 1200, "1:1", "qwen-image")
|
|
# 1200 / 28 = 42.85 -> 43 * 28 = 1204
|
|
print(f"Case 2 (Custom 1200, 1:1): {w}x{h}")
|
|
assert w == 1204 and h == 1204
|
|
|
|
# Case 3: Custom Ratio
|
|
w, h = resolve_resolution("Qwen - 1:1 (1328x1328)", 1000, "16:9", "qwen-image")
|
|
# 16:9 -> w=1000, h=562.5
|
|
# 1000/28 = 35.7 -> 36 * 28 = 1008
|
|
# 562.5/28 = 20.08 -> 20 * 28 = 560
|
|
print(f"Case 3 (Custom 1000, 16:9): {w}x{h}")
|
|
assert w == 1008 and h == 560
|
|
|
|
# Case 4: Zero Custom Size (Should use Preset)
|
|
w, h = resolve_resolution("Qwen - 1:1 (1328x1328)", 0, "16:9", "qwen-image")
|
|
print(f"Case 4 (Custom 0 -> Preset 1328x1328): {w}x{h}")
|
|
assert w == 1328 and h == 1328
|
|
|
|
print("All tests passed!")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
test_resolve()
|