OpenSourceJoker/video2x
3
0
Fork 0
mirror of https://github.com/k4yt3x/video2x.git synced 2024-12-28 23:19:11 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

tests(*): updated test file paths and removed unused dependencies

Browse Source 
Signed-off-by: k4yt3x <i@k4yt3x.com>
This commit is contained in:
k4yt3x 2023-09-24 07:33:07 +00:00
parent 7c70032fe7
commit b4b7cadf0d
No known key found for this signature in database
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
tests/test_upscaler.py
View File

@ -1,9 +1,6 @@
#!/usr/bin/python #!/usr/bin/python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
import ctypes
import multiprocessing
import os
from pathlib import Path from pathlib import Path
import utils import utils
@ -14,9 +11,11 @@ from video2x import Upscaler, Video2X
def test_upscaling(): def test_upscaling():
video2x = Video2X() video2x = Video2X()
output_path = Path("data/test_video_output.mp4")
output_path = Path("tests/data/test_video_output.mp4")
video2x.upscale( video2x.upscale(
Path("data/test_video.mp4"), Path("tests/data/test_video.mp4"),
output_path, output_path,
None, None,
720, 720,
@ -32,9 +31,9 @@ def test_upscale_image():
# initialize upscaler instance # initialize upscaler instance
upscaler = Upscaler() upscaler = Upscaler()
image = Image.open("data/test_image.png") image = Image.open("tests/data/test_image.png")
upscaled_image = upscaler.upscale_image(image, 1680, 960, "waifu2x", 3) upscaled_image = upscaler.upscale_image(image, 1680, 960, "waifu2x", 3)
reference_image = Image.open("data/test_image_ref.png") reference_image = Image.open("tests/data/test_image_ref.png")
assert utils.get_image_diff(upscaled_image, reference_image) < 0.5 assert utils.get_image_diff(upscaled_image, reference_image) < 0.5