import os
import pytest

from test_config import *
from modules.live_portrait.live_portrait_inferencer import LivePortraitInferencer


@pytest.mark.parametrize(
    "input_image",
    [
        TEST_IMAGE_PATH
    ]
)
def test_image_restoration(
    input_image: str,
):
    if not os.path.exists(TEST_IMAGE_PATH):
        download_image(
            TEST_IMAGE_URL,
            TEST_IMAGE_PATH
        )

    inferencer = LivePortraitInferencer()

    restored_output = inferencer.resrgan_inferencer.restore_image(
        input_image,
        overwrite=False
    )

    assert os.path.exists(restored_output)
    assert are_images_different(input_image, restored_output)