From d66b7fa144b8f3e73f89bd956f2f9ee0c6f67926 Mon Sep 17 00:00:00 2001 From: k4yt3x Date: Wed, 31 May 2023 15:23:53 +0000 Subject: [PATCH] fixed upscaler test --- tests/test_upscaler.py | 7 +------ 1 file changed, 1 insertion(+), 6 deletions(-) diff --git a/tests/test_upscaler.py b/tests/test_upscaler.py index 699acd4..ea2ec3e 100644 --- a/tests/test_upscaler.py +++ b/tests/test_upscaler.py @@ -29,16 +29,11 @@ def test_upscaling(): def test_upscale_image(): - # initialize upscaler instance - processing_queue = multiprocessing.Queue(maxsize=30) - processed_frames = multiprocessing.Manager().list([None]) - pause = multiprocessing.Value(ctypes.c_bool, False) - upscaler = Upscaler(processing_queue, processed_frames, pause) + upscaler = Upscaler() image = Image.open("data/test_image.png") upscaled_image = upscaler.upscale_image(image, 1680, 960, "waifu2x", 3) - reference_image = Image.open("data/test_image_ref.png") assert utils.get_image_diff(upscaled_image, reference_image) < 0.5