26 lines
892 B
Python
26 lines
892 B
Python
from app.homelab.anima_prompt import (
|
|
build_draw_self_prompt,
|
|
build_scene_tags_prompt,
|
|
looks_like_booru_tags,
|
|
)
|
|
|
|
|
|
def test_build_draw_self_prompt_includes_appearance():
|
|
bundle = build_draw_self_prompt("silver_hair, wolf_ears, blue_eyes")
|
|
assert "silver_hair" in bundle.positive
|
|
assert "wolf_ears" in bundle.positive
|
|
assert "looking_at_viewer" in bundle.positive
|
|
assert "POV:" not in bundle.positive
|
|
assert ". " not in bundle.positive
|
|
assert "worst quality" in bundle.negative
|
|
|
|
|
|
def test_build_draw_self_prompt_lora():
|
|
bundle = build_draw_self_prompt("1girl", lora_name="rin_lora", lora_weight=0.75)
|
|
assert "<lora:rin_lora:0.75>" in bundle.positive
|
|
|
|
|
|
def test_looks_like_booru_tags():
|
|
assert looks_like_booru_tags("1girl, smile, indoors")
|
|
assert not looks_like_booru_tags("draw a picture of a cat on the moon")
|