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

2


3


4
def test_simple_upscaling_performed(base_url, img2img_basic_image_base64):

5
    payload = {

6
        "resize_mode": 0,

7
        "show_extras_results": True,

8
        "gfpgan_visibility": 0,

9
        "codeformer_visibility": 0,

10
        "codeformer_weight": 0,

11
        "upscaling_resize": 2,

12
        "upscaling_resize_w": 128,

13
        "upscaling_resize_h": 128,

14
        "upscaling_crop": True,

15
        "upscaler_1": "Lanczos",

16
        "upscaler_2": "None",

17
        "extras_upscaler_2_visibility": 0,

18
        "image": img2img_basic_image_base64,

19
    }

20
    assert requests.post(f"{base_url}/sdapi/v1/extra-single-image", json=payload).status_code == 200

21


22


23
def test_png_info_performed(base_url, img2img_basic_image_base64):

24
    payload = {

25
        "image": img2img_basic_image_base64,

26
    }

27
    assert requests.post(f"{base_url}/sdapi/v1/extra-single-image", json=payload).status_code == 200

28


29


30
def test_interrogate_performed(base_url, img2img_basic_image_base64):

31
    payload = {

32
        "image": img2img_basic_image_base64,

33
        "model": "clip",

34
    }

35
    assert requests.post(f"{base_url}/sdapi/v1/extra-single-image", json=payload).status_code == 200

36


37