diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-03-12 05:55:15 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-03-12 05:55:15 +0000 |
| commit | adf723a9b2a8e83c71219a33a914f5dcd01d88f8 (patch) | |
| tree | 9406471fc6430fbc43d2dfebb3c7ca1ad8206f80 /test/basic_features/img2img_test.py | |
| parent | ddc503d14c23c9e370642448023a0dd565be098c (diff) | |
| parent | d25c4b13e4be0c89401637c769e3634e7ee456a7 (diff) | |
| download | stable-diffusion-webui-gfx803-adf723a9b2a8e83c71219a33a914f5dcd01d88f8.tar.gz stable-diffusion-webui-gfx803-adf723a9b2a8e83c71219a33a914f5dcd01d88f8.tar.bz2 stable-diffusion-webui-gfx803-adf723a9b2a8e83c71219a33a914f5dcd01d88f8.zip | |
Merge pull request #8492 from zhanghua000/absolute-path
fix: gradio's ValueError about fetching extensions files
Diffstat (limited to 'test/basic_features/img2img_test.py')
| -rw-r--r-- | test/basic_features/img2img_test.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/test/basic_features/img2img_test.py b/test/basic_features/img2img_test.py index 08c5c903..5240ec36 100644 --- a/test/basic_features/img2img_test.py +++ b/test/basic_features/img2img_test.py @@ -1,14 +1,16 @@ +import os import unittest import requests from gradio.processing_utils import encode_pil_to_base64 from PIL import Image +from modules.paths import script_path class TestImg2ImgWorking(unittest.TestCase): def setUp(self): self.url_img2img = "http://localhost:7860/sdapi/v1/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, "denoising_strength": 0.75, "mask": None, @@ -47,11 +49,11 @@ class TestImg2ImgWorking(unittest.TestCase): self.assertEqual(requests.post(self.url_img2img, json=self.simple_img2img).status_code, 200) 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) 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.assertEqual(requests.post(self.url_img2img, json=self.simple_img2img).status_code, 200) |