test_extras.py 1.1 KB

1234567891011121314151617181920212223242526272829303132333435
  1. import requests
  2. def test_simple_upscaling_performed(base_url, img2img_basic_image_base64):
  3. payload = {
  4. "resize_mode": 0,
  5. "show_extras_results": True,
  6. "gfpgan_visibility": 0,
  7. "codeformer_visibility": 0,
  8. "codeformer_weight": 0,
  9. "upscaling_resize": 2,
  10. "upscaling_resize_w": 128,
  11. "upscaling_resize_h": 128,
  12. "upscaling_crop": True,
  13. "upscaler_1": "Lanczos",
  14. "upscaler_2": "None",
  15. "extras_upscaler_2_visibility": 0,
  16. "image": img2img_basic_image_base64,
  17. }
  18. assert requests.post(f"{base_url}/sdapi/v1/extra-single-image", json=payload).status_code == 200
  19. def test_png_info_performed(base_url, img2img_basic_image_base64):
  20. payload = {
  21. "image": img2img_basic_image_base64,
  22. }
  23. assert requests.post(f"{base_url}/sdapi/v1/extra-single-image", json=payload).status_code == 200
  24. def test_interrogate_performed(base_url, img2img_basic_image_base64):
  25. payload = {
  26. "image": img2img_basic_image_base64,
  27. "model": "clip",
  28. }
  29. assert requests.post(f"{base_url}/sdapi/v1/extra-single-image", json=payload).status_code == 200