Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
automatic1111
GitHub Repository: automatic1111/stable-diffusion-webui
 Path: blob/master/test/test_img2img.py
3055 views
1


2
import pytest

3
import requests

4


5


6
@pytest.fixture()

7
def url_img2img(base_url):

8
    return f"{base_url}/sdapi/v1/img2img"

9


10


11
@pytest.fixture()

12
def simple_img2img_request(img2img_basic_image_base64):

13
    return {

14
        "batch_size": 1,

15
        "cfg_scale": 7,

16
        "denoising_strength": 0.75,

17
        "eta": 0,

18
        "height": 64,

19
        "include_init_images": False,

20
        "init_images": [img2img_basic_image_base64],

21
        "inpaint_full_res": False,

22
        "inpaint_full_res_padding": 0,

23
        "inpainting_fill": 0,

24
        "inpainting_mask_invert": False,

25
        "mask": None,

26
        "mask_blur": 4,

27
        "n_iter": 1,

28
        "negative_prompt": "",

29
        "override_settings": {},

30
        "prompt": "example prompt",

31
        "resize_mode": 0,

32
        "restore_faces": False,

33
        "s_churn": 0,

34
        "s_noise": 1,

35
        "s_tmax": 0,

36
        "s_tmin": 0,

37
        "sampler_index": "Euler a",

38
        "seed": -1,

39
        "seed_resize_from_h": -1,

40
        "seed_resize_from_w": -1,

41
        "steps": 3,

42
        "styles": [],

43
        "subseed": -1,

44
        "subseed_strength": 0,

45
        "tiling": False,

46
        "width": 64,

47
    }

48


49


50
def test_img2img_simple_performed(url_img2img, simple_img2img_request):

51
    assert requests.post(url_img2img, json=simple_img2img_request).status_code == 200

52


53


54
def test_inpainting_masked_performed(url_img2img, simple_img2img_request, mask_basic_image_base64):

55
    simple_img2img_request["mask"] = mask_basic_image_base64

56
    assert requests.post(url_img2img, json=simple_img2img_request).status_code == 200

57


58


59
def test_inpainting_with_inverted_masked_performed(url_img2img, simple_img2img_request, mask_basic_image_base64):

60
    simple_img2img_request["mask"] = mask_basic_image_base64

61
    simple_img2img_request["inpainting_mask_invert"] = True

62
    assert requests.post(url_img2img, json=simple_img2img_request).status_code == 200

63


64


65
def test_img2img_sd_upscale_performed(url_img2img, simple_img2img_request):

66
    simple_img2img_request["script_name"] = "sd upscale"

67
    simple_img2img_request["script_args"] = ["", 8, "Lanczos", 2.0]

68
    assert requests.post(url_img2img, json=simple_img2img_request).status_code == 200

69


70