26 lines
835 B
Python
26 lines
835 B
Python
  
|
from nanobot.utils.image_generation_intent import image_generation_prompt
|
||
|
|
|
||
|
|
|
||
|
|
def test_image_generation_prompt_ignores_plain_messages() -> None:
|
||
|
|
assert image_generation_prompt("hello", {}) == "hello"
|
||
|
|
|
||
|
|
|
||
|
|
def test_image_generation_prompt_uses_auto_aspect_instruction() -> None:
|
||
|
|
prompt = image_generation_prompt(
|
||
|
|
"Draw a poster",
|
||
|
|
{"image_generation": {"enabled": True, "aspect_ratio": None}},
|
||
|
|
)
|
||
|
|
|
||
|
|
assert "Draw a poster" in prompt
|
||
|
|
assert "Use the generate_image tool" in prompt
|
||
|
|
assert "Choose the most suitable aspect_ratio yourself" in prompt
|
||
|
|
|
||
|
|
|
||
|
|
def test_image_generation_prompt_uses_selected_aspect_ratio() -> None:
|
||
|
|
prompt = image_generation_prompt(
|
||
|
|
"Draw a banner",
|
||
|
|
{"image_generation": {"enabled": True, "aspect_ratio": "16:9"}},
|
||
|
|
)
|
||
|
|
|
||
|
|
assert "aspect_ratio='16:9'" in prompt
|