jhj0517
commited on
Commit
·
a79cb95
1
Parent(s):
63f92e8
Fix test
Browse files
tests/test_image_restoration.py
CHANGED
|
@@ -14,7 +14,7 @@ from modules.live_portrait.live_portrait_inferencer import LivePortraitInference
|
|
| 14 |
def test_image_restoration(
|
| 15 |
input_image: str,
|
| 16 |
):
|
| 17 |
-
if not os.path.exists(
|
| 18 |
download_image(
|
| 19 |
TEST_IMAGE_URL,
|
| 20 |
TEST_IMAGE_PATH
|
|
@@ -23,7 +23,8 @@ def test_image_restoration(
|
|
| 23 |
inferencer = LivePortraitInferencer()
|
| 24 |
|
| 25 |
restored_output = inferencer.resrgan_inferencer.restore_image(
|
| 26 |
-
input_image
|
|
|
|
| 27 |
)
|
| 28 |
|
| 29 |
assert os.path.exists(restored_output)
|
|
|
|
| 14 |
def test_image_restoration(
|
| 15 |
input_image: str,
|
| 16 |
):
|
| 17 |
+
if not os.path.exists(input_image):
|
| 18 |
download_image(
|
| 19 |
TEST_IMAGE_URL,
|
| 20 |
TEST_IMAGE_PATH
|
|
|
|
| 23 |
inferencer = LivePortraitInferencer()
|
| 24 |
|
| 25 |
restored_output = inferencer.resrgan_inferencer.restore_image(
|
| 26 |
+
input_image,
|
| 27 |
+
overwrite=False
|
| 28 |
)
|
| 29 |
|
| 30 |
assert os.path.exists(restored_output)
|