CalvinBackup/facefusion
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion.git synced 2026-06-02 19:01:35 +02:00
Code Issues Packages Projects Releases Wiki Activity

Improve encoder tests with hash assertion (#1103)

Browse Source 
* improve test_encode_opus_buffer

* try different hash per os

* fix lint

* add windows check

* update windows hash

* fix test and lint

* update windows hash

* update CI for test_video_encoder.py

* update hash for macos

* update method to use single cpu

* update mac hash

* update windows hash

* cleanup

* restore ci.yml

* remove argument defaults

* selected CI tests

* selected CI tests

* restore ci.yml

---------

Co-authored-by: henryruhs <info@henryruhs.com>
This commit is contained in:
Harisreedhar
2026-05-12 16:36:06 +05:30
committed by GitHub
parent 8690ccf49e
commit 5e39c60b5c
4 changed files with 24 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_video_encoder.py
+12 -5
View File
@@ -5,7 +5,9 @@ import pytest
from tests.assert_helper import get_test_example_file, get_test_examples_directory
from facefusion import state_manager
from facefusion.common_helper import is_linux, is_macos, is_windows
from facefusion.download import conditional_download
from facefusion.hash_helper import create_hash
from facefusion.libraries import vpx as vpx_module
from facefusion.video_encoder import create_vpx_encoder, destroy_vpx_encoder, encode_vpx_buffer
from facefusion.vision import read_video_frame
@@ -21,24 +23,29 @@ def before_all() -> None:
def test_create_vpx_encoder() -> None:
assert create_vpx_encoder(320, 240, 1000)
assert create_vpx_encoder(0, 0, 0) is None
assert create_vpx_encoder(320, 240, 1000, 8, 16)
assert create_vpx_encoder(0, 0, 0, 0, 0) is None
def test_encode_vpx_buffer() -> None:
vision_frame = read_video_frame(get_test_example_file('target-240p.mp4'))
height, width = vision_frame.shape[:2]
vpx_encoder = create_vpx_encoder(width, height, 1000)
vpx_encoder = create_vpx_encoder(width, height, 1000, 1, 0)
buffer_valid = cv2.cvtColor(vision_frame, cv2.COLOR_BGR2YUV_I420).tobytes()
buffer_invalid = bytes(0)
assert encode_vpx_buffer(vpx_encoder, buffer_valid, width, height, 3, 1)
if is_linux() or is_windows():
assert create_hash(encode_vpx_buffer(vpx_encoder, buffer_valid, width, height, 3, 1)) == 'ce133a1f'
if is_macos():
assert create_hash(encode_vpx_buffer(vpx_encoder, buffer_valid, width, height, 3, 1)) == '21c36925'
assert encode_vpx_buffer(vpx_encoder, buffer_invalid, width, height, 0, 0) == b''
def test_destroy_vpx_encoder() -> None:
vpx_encoder = create_vpx_encoder(320, 240, 1000)
vpx_encoder = create_vpx_encoder(320, 240, 1000, 8, 16)
with patch.object(vpx_module.create_static_library(), 'vpx_codec_destroy') as mock:
destroy_vpx_encoder(vpx_encoder)