mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-08-09 15:30:16 -03:00
d2f955266d
Fix ~790 basedpyright errors across the test suite: - Type stub subclasses of real production classes with super().__init__() - Add missing generic type arguments and Dict[str, Any] annotations - Add None guards before subscript/member access - Adapt tests to production API changes (removed dead handlers, PersistentModelCache.get_default, _i18n_filter_added location)
52 lines
1.7 KiB
Python
52 lines
1.7 KiB
Python
import pytest
|
|
|
|
from py.utils.constants import NSFW_LEVELS
|
|
from py.utils.preview_selection import resolve_mature_threshold, select_preview_media
|
|
|
|
|
|
def test_select_preview_returns_first_when_blur_disabled():
|
|
images = [
|
|
{"url": "nsfw", "type": "image", "nsfwLevel": 32},
|
|
{"url": "safe", "type": "image", "nsfwLevel": 1},
|
|
]
|
|
|
|
selected, level = select_preview_media(images, blur_mature_content=False)
|
|
|
|
assert selected is not None
|
|
assert selected["url"] == "nsfw"
|
|
assert level == 32
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("threshold_name", "expected_url"),
|
|
[
|
|
("PG13", "pg"),
|
|
("R", "pg13"),
|
|
("X", "r"),
|
|
("XXX", "x"),
|
|
],
|
|
)
|
|
def test_select_preview_respects_configurable_threshold(threshold_name, expected_url):
|
|
images = [
|
|
{"url": "xxx", "type": "image", "nsfwLevel": NSFW_LEVELS["XXX"]},
|
|
{"url": "x", "type": "image", "nsfwLevel": NSFW_LEVELS["X"]},
|
|
{"url": "r", "type": "image", "nsfwLevel": NSFW_LEVELS["R"]},
|
|
{"url": "pg13", "type": "image", "nsfwLevel": NSFW_LEVELS["PG13"]},
|
|
{"url": "pg", "type": "image", "nsfwLevel": NSFW_LEVELS["PG"]},
|
|
]
|
|
|
|
selected, level = select_preview_media(
|
|
images,
|
|
blur_mature_content=True,
|
|
mature_threshold=NSFW_LEVELS[threshold_name],
|
|
)
|
|
|
|
assert selected is not None
|
|
assert selected["url"] == expected_url
|
|
assert level == next(item["nsfwLevel"] for item in images if item["url"] == expected_url)
|
|
|
|
|
|
def test_resolve_mature_threshold_falls_back_to_r_for_invalid_value():
|
|
assert resolve_mature_threshold({"mature_blur_level": "invalid"}) == NSFW_LEVELS["R"]
|
|
assert resolve_mature_threshold({}) == NSFW_LEVELS["R"]
|