summaryrefslogtreecommitdiff
path: root/tests/conftest.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/conftest.py')
-rw-r--r--tests/conftest.py38
1 files changed, 38 insertions, 0 deletions
diff --git a/tests/conftest.py b/tests/conftest.py
new file mode 100644
index 0000000..d364dbd
--- /dev/null
+++ b/tests/conftest.py
@@ -0,0 +1,38 @@
+from __future__ import annotations
+
+import base64
+import io
+from pathlib import Path
+
+import pytest
+from PIL import Image
+
+from mcp_sd15.config import (
+ AppConfig,
+ ControlModelConfig,
+ ControlNetConfig,
+ GenerationConfig,
+ ModelConfig,
+ OutputConfig,
+)
+
+
+@pytest.fixture
+def app_config(tmp_path: Path) -> AppConfig:
+ return AppConfig(
+ model=ModelConfig(tmp_path, "base.safetensors", None, ()),
+ generation=GenerationConfig(),
+ controlnet=ControlNetConfig(
+ depth=ControlModelConfig(True, "depth.safetensors", "midas"),
+ openpose=ControlModelConfig(True, "pose.safetensors"),
+ inpaint=ControlModelConfig(True, "inpaint.safetensors"),
+ ),
+ output=OutputConfig(tmp_path / "outputs", False),
+ )
+
+
+@pytest.fixture
+def image_base64() -> str:
+ output = io.BytesIO()
+ Image.new("RGB", (8, 8), "blue").save(output, format="PNG")
+ return base64.b64encode(output.getvalue()).decode("ascii")