mirror of
https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
synced 2026-03-23 06:40:23 -07:00
51 lines
1.6 KiB
Python
51 lines
1.6 KiB
Python
"""Extended API tests for quality tier. Increases coverage of API endpoints."""
|
|
import pytest
|
|
import requests
|
|
|
|
|
|
@pytest.mark.parametrize("url", [
|
|
"sdapi/v1/progress",
|
|
"sdapi/v1/sd-vae",
|
|
"sdapi/v1/latent-upscale-modes",
|
|
"sdapi/v1/memory",
|
|
"sdapi/v1/scripts",
|
|
"sdapi/v1/script-info",
|
|
"sdapi/v1/extensions",
|
|
"sdapi/v1/samplers",
|
|
"sdapi/v1/schedulers",
|
|
"sdapi/v1/upscalers",
|
|
"sdapi/v1/sd-models",
|
|
"sdapi/v1/options",
|
|
"sdapi/v1/cmd-flags",
|
|
"sdapi/v1/face-restorers",
|
|
"sdapi/v1/prompt-styles",
|
|
])
|
|
def test_get_api_endpoint(base_url, url):
|
|
"""Verify extended API endpoints return 200."""
|
|
assert requests.get(f"{base_url}/{url}").status_code == 200
|
|
|
|
|
|
def test_png_info(base_url, img2img_basic_image_base64):
|
|
"""PNG info endpoint."""
|
|
payload = {"image": img2img_basic_image_base64}
|
|
assert requests.post(f"{base_url}/sdapi/v1/png-info", json=payload).status_code == 200
|
|
|
|
|
|
def test_extra_batch_images(base_url, img2img_basic_image_base64):
|
|
"""Extra batch images API."""
|
|
batch_payload = {
|
|
"resize_mode": 0,
|
|
"show_extras_results": True,
|
|
"gfpgan_visibility": 0,
|
|
"codeformer_visibility": 0,
|
|
"codeformer_weight": 0,
|
|
"upscaling_resize": 2,
|
|
"upscaling_resize_w": 128,
|
|
"upscaling_resize_h": 128,
|
|
"upscaling_crop": True,
|
|
"upscaler_1": "Lanczos",
|
|
"upscaler_2": "None",
|
|
"extras_upscaler_2_visibility": 0,
|
|
"imageList": [{"data": img2img_basic_image_base64, "name": "test.png"}],
|
|
}
|
|
assert requests.post(f"{base_url}/sdapi/v1/extra-batch-images", json=batch_payload).status_code == 200
|