37a3fb0f86
* nano-banana-pro: respect explicit --resolution when editing images * Changelog: note nano banana resolution fix * Update CHANGELOG.md --------- Co-authored-by: Vincent Koc <vincentkoc@ieee.org>
37 lines
1.1 KiB
Python
37 lines
1.1 KiB
Python
import importlib.util
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
MODULE_PATH = Path(__file__).with_name("generate_image.py")
|
|
SPEC = importlib.util.spec_from_file_location("generate_image", MODULE_PATH)
|
|
assert SPEC and SPEC.loader
|
|
MODULE = importlib.util.module_from_spec(SPEC)
|
|
SPEC.loader.exec_module(MODULE)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("max_input_dim", "expected"),
|
|
[
|
|
(0, "1K"),
|
|
(1499, "1K"),
|
|
(1500, "2K"),
|
|
(2999, "2K"),
|
|
(3000, "4K"),
|
|
],
|
|
)
|
|
def test_auto_detect_resolution_thresholds(max_input_dim, expected):
|
|
assert MODULE.auto_detect_resolution(max_input_dim) == expected
|
|
|
|
|
|
def test_choose_output_resolution_auto_detects_when_resolution_omitted():
|
|
assert MODULE.choose_output_resolution(None, 2200, True) == ("2K", True)
|
|
|
|
|
|
def test_choose_output_resolution_defaults_to_1k_without_inputs():
|
|
assert MODULE.choose_output_resolution(None, 0, False) == ("1K", False)
|
|
|
|
|
|
def test_choose_output_resolution_respects_explicit_1k_with_large_input():
|
|
assert MODULE.choose_output_resolution("1K", 3500, True) == ("1K", False)
|