mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-03-22 05:32:12 -03:00
40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
from py.utils.preview_selection import select_preview_media
|
|
|
|
|
|
def test_select_preview_prefers_safe_media_when_blurred():
|
|
images = [
|
|
{"url": "nsfw", "type": "image", "nsfwLevel": 8},
|
|
{"url": "mid", "type": "image", "nsfwLevel": 4},
|
|
{"url": "safe", "type": "image", "nsfwLevel": 1},
|
|
]
|
|
|
|
selected, level = select_preview_media(images, blur_mature_content=True)
|
|
|
|
assert selected["url"] == "safe"
|
|
assert level == 1
|
|
|
|
|
|
def test_select_preview_returns_lowest_when_no_safe_media():
|
|
images = [
|
|
{"url": "x", "type": "image", "nsfwLevel": 16},
|
|
{"url": "r", "type": "image", "nsfwLevel": 4},
|
|
{"url": "xx", "type": "image", "nsfwLevel": 8},
|
|
]
|
|
|
|
selected, level = select_preview_media(images, blur_mature_content=True)
|
|
|
|
assert selected["url"] == "r"
|
|
assert level == 4
|
|
|
|
|
|
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["url"] == "nsfw"
|
|
assert level == 32
|