aboutsummaryrefslogtreecommitdiffstats
path: root/test/basic_features/extras_test.py
diff options
context:
space:
mode:
authorZhang Hua <zhanghua.00@qq.com>2023-03-11 06:42:31 +0000
committerZhang Hua <zhanghua.00@qq.com>2023-03-11 12:00:12 +0000
commitd25c4b13e4be0c89401637c769e3634e7ee456a7 (patch)
tree3d89f2be7849baa7c310581bca4966420cb4b68a /test/basic_features/extras_test.py
parent9abe2f5e74f02d4c8e47d2d4e03464179e9f0aa2 (diff)
downloadstable-diffusion-webui-gfx803-d25c4b13e4be0c89401637c769e3634e7ee456a7.tar.gz
stable-diffusion-webui-gfx803-d25c4b13e4be0c89401637c769e3634e7ee456a7.tar.bz2
stable-diffusion-webui-gfx803-d25c4b13e4be0c89401637c769e3634e7ee456a7.zip
test/basic_features/{extras,img2img}_test.py: use absolute path
Diffstat (limited to 'test/basic_features/extras_test.py')
-rw-r--r--test/basic_features/extras_test.py8
1 files changed, 5 insertions, 3 deletions
diff --git a/test/basic_features/extras_test.py b/test/basic_features/extras_test.py
index 0170c511..8ed98747 100644
--- a/test/basic_features/extras_test.py
+++ b/test/basic_features/extras_test.py
@@ -1,7 +1,9 @@
+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 TestExtrasWorking(unittest.TestCase):
def setUp(self):
@@ -19,7 +21,7 @@ class TestExtrasWorking(unittest.TestCase):
"upscaler_1": "None",
"upscaler_2": "None",
"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):
@@ -31,7 +33,7 @@ class TestPngInfoWorking(unittest.TestCase):
def setUp(self):
self.url_png_info = "http://localhost:7860/sdapi/v1/extra-single-image"
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):
@@ -42,7 +44,7 @@ class TestInterrogateWorking(unittest.TestCase):
def setUp(self):
self.url_interrogate = "http://localhost:7860/sdapi/v1/extra-single-image"
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"
}