mirror of
https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
synced 2026-03-22 06:10:51 -07:00
9aa32e8917
* test(M04): add quality tests for coverage gate 40% Made-with: Cursor * fix: remove test_interrogate_deepbooru (404 in CI stub) Made-with: Cursor
52 lines
1.7 KiB
Python
52 lines
1.7 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",
|
|
])
|
|
def test_get_api_endpoint(base_url, url):
|
|
"""Verify extended API endpoints return 200."""
|
|
assert requests.get(f"{base_url}/{url}").status_code == 200
|
|
|
|
|
|
<<<<<<< HEAD
|
|
=======
|
|
def test_interrogate_deepbooru(base_url, img2img_basic_image_base64):
|
|
"""Interrogate with deepbooru model."""
|
|
payload = {"image": img2img_basic_image_base64, "model": "deepbooru"}
|
|
assert requests.post(f"{base_url}/sdapi/v1/interrogate", json=payload).status_code == 200
|
|
|
|
|
|
>>>>>>> origin/main
|
|
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
|