OSSForks/faceswap
1
0
Fork 0
mirror of https://github.com/zebrajr/faceswap.git synced 2025-12-06 00:20:09 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

bugfix: preview tool unit test

Browse Source
This commit is contained in:
torzdf 2023-02-20 12:28:02 +00:00
parent 637dc3b55c
commit cd929e0df3
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/tools/alignments/media_test.py
View File

@ -1,6 +1,7 @@
#!/usr/bin python3
""" Pytest unit tests for :mod:`tools.alignments.media` """
import os
from operator import itemgetter
from typing import cast, Dict, Generator, List, Tuple
from unittest.mock import MagicMock
@ -608,8 +609,8 @@ class TestFrames:
dict(frame_fullname="b.png", frame_name="b", frame_extension=".png")]
frames = Frames(folder, None)
returned = list(frames.process_frames())
assert returned == expected
returned = sorted(list(frames.process_frames()), key=itemgetter("frame_fullname"))
assert returned == sorted(expected, key=itemgetter("frame_fullname"))
def test_process_video(self, folder: str) -> None:
""" Test for :class:`~tools.alignments.media.Frames` process_video method