m4
94 lines
2.9 KiB
Python
94 lines
2.9 KiB
Python
"""Tests for the image generation agent tools."""
|
|
|
|
import json
|
|
|
|
import pytest
|
|
|
|
from EvoScientist.tools import image as image_tools
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def fake_service(monkeypatch, tmp_path):
|
|
"""Patch the service entry points the tools call."""
|
|
async def fake_generate(workspace, **kwargs):
|
|
return {
|
|
"ok": True,
|
|
"path": "artifacts/generated_1.png",
|
|
"paths": ["artifacts/generated_1.png"],
|
|
"mime_type": "image/png",
|
|
"model": "gpt-image-2",
|
|
"size": kwargs.get("size", "1024x1024"),
|
|
}
|
|
|
|
async def fake_edit(workspace, **kwargs):
|
|
return {
|
|
"ok": True,
|
|
"path": "artifacts/edited_1.png",
|
|
"paths": ["artifacts/edited_1.png"],
|
|
"mime_type": "image/png",
|
|
"model": "gpt-image-2",
|
|
"size": kwargs.get("size", "1024x1024"),
|
|
}
|
|
|
|
monkeypatch.setattr(image_tools, "_generate_for_workspace", fake_generate)
|
|
monkeypatch.setattr(image_tools, "_edit_for_workspace", fake_edit)
|
|
monkeypatch.setattr(image_tools, "_workspace", lambda: tmp_path)
|
|
|
|
|
|
async def test_generate_image_success():
|
|
result = json.loads(
|
|
await image_tools.generate_image.ainvoke({"prompt": "a cat"})
|
|
)
|
|
assert result["ok"] is True
|
|
assert result["paths"] == ["artifacts/generated_1.png"]
|
|
|
|
|
|
async def test_edit_image_success():
|
|
result = json.loads(
|
|
await image_tools.edit_image.ainvoke(
|
|
{"image_path": "artifacts/src.png", "prompt": "make blue"}
|
|
)
|
|
)
|
|
assert result["ok"] is True
|
|
assert result["paths"] == ["artifacts/edited_1.png"]
|
|
|
|
|
|
async def test_error_becomes_json(monkeypatch):
|
|
from EvoScientist.image_gen.adapters.base import ImageGenError
|
|
|
|
async def boom(workspace, **kwargs):
|
|
raise ImageGenError("no image models configured")
|
|
|
|
monkeypatch.setattr(image_tools, "_generate_for_workspace", boom)
|
|
result = json.loads(await image_tools.generate_image.ainvoke({"prompt": "x"}))
|
|
assert result["ok"] is False
|
|
assert "no image models" in result["error"]
|
|
|
|
|
|
def test_refresh_descriptions_includes_models(monkeypatch, tmp_path):
|
|
config = tmp_path / "config.yaml"
|
|
config.write_text(
|
|
"""
|
|
image_generation:
|
|
default_model: gpt-image-2
|
|
models:
|
|
- id: gpt-image-2
|
|
name: GPT Image 2
|
|
provider: openai
|
|
api_key: sk-x
|
|
""",
|
|
encoding="utf-8",
|
|
)
|
|
monkeypatch.setattr(image_tools, "_settings_path", lambda: config)
|
|
image_tools.refresh_image_tool_descriptions()
|
|
assert "GPT Image 2" in image_tools.generate_image.description
|
|
assert "Default: gpt-image-2" in image_tools.generate_image.description
|
|
assert "sk-x" not in image_tools.generate_image.description
|
|
|
|
|
|
def test_tools_package_exports():
|
|
from EvoScientist import tools
|
|
|
|
assert tools.generate_image is image_tools.generate_image
|
|
assert tools.edit_image is image_tools.edit_image
|