michael/stable-diffusion-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

test/basic_features/{extras,img2img}_test.py: use absolute path

Browse source
This commit is contained in:
Zhang Hua 2023-03-11 14:42:31 +08:00
parent 9abe2f5e74
commit d25c4b13e4
No known key found for this signature in database
GPG key ID: 69528D6DCD2BE030
2 changed files with 10 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
test/basic_features/extras_test.py
View file

@ -1,7 +1,9 @@
import os
import unittest import unittest
import requests import requests
from gradio.processing_utils import encode_pil_to_base64 from gradio.processing_utils import encode_pil_to_base64
from PIL import Image from PIL import Image
from modules.paths import script_path
class TestExtrasWorking(unittest.TestCase): class TestExtrasWorking(unittest.TestCase):
def setUp(self): def setUp(self):
@ -19,7 +21,7 @@ class TestExtrasWorking(unittest.TestCase):
"upscaler_1": "None", "upscaler_1": "None",
"upscaler_2": "None", "upscaler_2": "None",
"extras_upscaler_2_visibility": 0, "extras_upscaler_2_visibility": 0,
"image": encode_pil_to_base64(Image.open(r"test/test_files/img2img_basic.png")) "image": encode_pil_to_base64(Image.open(os.path.join(script_path, r"test/test_files/img2img_basic.png")))
} }
def test_simple_upscaling_performed(self): def test_simple_upscaling_performed(self):
@ -31,7 +33,7 @@ class TestPngInfoWorking(unittest.TestCase):
def setUp(self): def setUp(self):
self.url_png_info = "http://localhost:7860/sdapi/v1/extra-single-image" self.url_png_info = "http://localhost:7860/sdapi/v1/extra-single-image"
self.png_info = { self.png_info = {
"image": encode_pil_to_base64(Image.open(r"test/test_files/img2img_basic.png")) "image": encode_pil_to_base64(Image.open(os.path.join(script_path, r"test/test_files/img2img_basic.png")))
} }
def test_png_info_performed(self): def test_png_info_performed(self):
@ -42,7 +44,7 @@ class TestInterrogateWorking(unittest.TestCase):
def setUp(self): def setUp(self):
self.url_interrogate = "http://localhost:7860/sdapi/v1/extra-single-image" self.url_interrogate = "http://localhost:7860/sdapi/v1/extra-single-image"
self.interrogate = { self.interrogate = {
"image": encode_pil_to_base64(Image.open(r"test/test_files/img2img_basic.png")), "image": encode_pil_to_base64(Image.open(os.path.join(script_path, r"test/test_files/img2img_basic.png"))),
"model": "clip" "model": "clip"
} }

8
test/basic_features/img2img_test.py
View file

@ -1,14 +1,16 @@
import os
import unittest import unittest
import requests import requests
from gradio.processing_utils import encode_pil_to_base64 from gradio.processing_utils import encode_pil_to_base64
from PIL import Image from PIL import Image
from modules.paths import script_path
class TestImg2ImgWorking(unittest.TestCase): class TestImg2ImgWorking(unittest.TestCase):
def setUp(self): def setUp(self):
self.url_img2img = "http://localhost:7860/sdapi/v1/img2img" self.url_img2img = "http://localhost:7860/sdapi/v1/img2img"
self.simple_img2img = { self.simple_img2img = {
"init_images": [encode_pil_to_base64(Image.open(r"test/test_files/img2img_basic.png"))], "init_images": [encode_pil_to_base64(Image.open(os.path.join(script_path, r"test/test_files/img2img_basic.png")))],
"resize_mode": 0, "resize_mode": 0,
"denoising_strength": 0.75, "denoising_strength": 0.75,
"mask": None, "mask": None,
@ -47,11 +49,11 @@ class TestImg2ImgWorking(unittest.TestCase):
self.assertEqual(requests.post(self.url_img2img, json=self.simple_img2img).status_code, 200) self.assertEqual(requests.post(self.url_img2img, json=self.simple_img2img).status_code, 200)
def test_inpainting_masked_performed(self): def test_inpainting_masked_performed(self):
self.simple_img2img["mask"] = encode_pil_to_base64(Image.open(r"test/test_files/mask_basic.png")) self.simple_img2img["mask"] = encode_pil_to_base64(Image.open(os.path.join(script_path, r"test/test_files/img2img_basic.png")))
self.assertEqual(requests.post(self.url_img2img, json=self.simple_img2img).status_code, 200) self.assertEqual(requests.post(self.url_img2img, json=self.simple_img2img).status_code, 200)
def test_inpainting_with_inverted_masked_performed(self): def test_inpainting_with_inverted_masked_performed(self):
self.simple_img2img["mask"] = encode_pil_to_base64(Image.open(r"test/test_files/mask_basic.png")) self.simple_img2img["mask"] = encode_pil_to_base64(Image.open(os.path.join(script_path, r"test/test_files/img2img_basic.png")))
self.simple_img2img["inpainting_mask_invert"] = True self.simple_img2img["inpainting_mask_invert"] = True
self.assertEqual(requests.post(self.url_img2img, json=self.simple_img2img).status_code, 200) self.assertEqual(requests.post(self.url_img2img, json=self.simple_img2img).status_code, 200)