|
|
|
import pytest |
|
import requests |
|
|
|
|
|
@pytest.fixture() |
|
def url_img2img(base_url): |
|
return f"{base_url}/sdapi/v1/img2img" |
|
|
|
|
|
@pytest.fixture() |
|
def simple_img2img_request(img2img_basic_image_base64): |
|
return { |
|
"batch_size": 1, |
|
"cfg_scale": 7, |
|
"denoising_strength": 0.75, |
|
"eta": 0, |
|
"height": 64, |
|
"include_init_images": False, |
|
"init_images": [img2img_basic_image_base64], |
|
"inpaint_full_res": False, |
|
"inpaint_full_res_padding": 0, |
|
"inpainting_fill": 0, |
|
"inpainting_mask_invert": False, |
|
"mask": None, |
|
"mask_blur": 4, |
|
"n_iter": 1, |
|
"negative_prompt": "", |
|
"override_settings": {}, |
|
"prompt": "example prompt", |
|
"resize_mode": 0, |
|
"restore_faces": False, |
|
"s_churn": 0, |
|
"s_noise": 1, |
|
"s_tmax": 0, |
|
"s_tmin": 0, |
|
"sampler_index": "Euler a", |
|
"seed": -1, |
|
"seed_resize_from_h": -1, |
|
"seed_resize_from_w": -1, |
|
"steps": 3, |
|
"styles": [], |
|
"subseed": -1, |
|
"subseed_strength": 0, |
|
"tiling": False, |
|
"width": 64, |
|
} |
|
|
|
|
|
def test_img2img_simple_performed(url_img2img, simple_img2img_request): |
|
assert requests.post(url_img2img, json=simple_img2img_request).status_code == 200 |
|
|
|
|
|
def test_inpainting_masked_performed(url_img2img, simple_img2img_request, mask_basic_image_base64): |
|
simple_img2img_request["mask"] = mask_basic_image_base64 |
|
assert requests.post(url_img2img, json=simple_img2img_request).status_code == 200 |
|
|
|
|
|
def test_inpainting_with_inverted_masked_performed(url_img2img, simple_img2img_request, mask_basic_image_base64): |
|
simple_img2img_request["mask"] = mask_basic_image_base64 |
|
simple_img2img_request["inpainting_mask_invert"] = True |
|
assert requests.post(url_img2img, json=simple_img2img_request).status_code == 200 |
|
|
|
|
|
def test_img2img_sd_upscale_performed(url_img2img, simple_img2img_request): |
|
simple_img2img_request["script_name"] = "sd upscale" |
|
simple_img2img_request["script_args"] = ["", 8, "Lanczos", 2.0] |
|
assert requests.post(url_img2img, json=simple_img2img_request).status_code == 200 |
|
|
