From 5611932fda31a80852492bb3fafe1535e6dd9ead Mon Sep 17 00:00:00 2001 From: henryruhs Date: Wed, 5 Aug 2026 22:45:17 +0200 Subject: [PATCH] add master fix_audio/video_encoder tests (#1212) Add master's test_fix_audio_encoder and test_fix_video_encoder; v4 has both functions and matches master's expected mappings. Kept v4's sanitize_* and spawn_frames tests and path-isolation signatures. Co-Authored-By: Claude Opus 4.8 Claude-Session: https://claude.ai/code/session_01V7MdZEmd1GE8uSDTMyq34r --- tests/test_ffmpeg.py | 28 +++++++++++++++++++++++++++- 1 file changed, 27 insertions(+), 1 deletion(-) diff --git a/tests/test_ffmpeg.py b/tests/test_ffmpeg.py index c114ae45..9389f14f 100644 --- a/tests/test_ffmpeg.py +++ b/tests/test_ffmpeg.py @@ -6,7 +6,7 @@ import pytest import facefusion.ffmpeg from facefusion import ffmpeg, ffmpeg_builder, process_manager, state_manager from facefusion.download import conditional_download -from facefusion.ffmpeg import concat_video, extract_frames, merge_video, read_audio_buffer, replace_audio, restore_audio, sanitize_audio, sanitize_image, sanitize_video, spawn_frames +from facefusion.ffmpeg import concat_video, extract_frames, fix_audio_encoder, fix_video_encoder, merge_video, read_audio_buffer, replace_audio, restore_audio, sanitize_audio, sanitize_image, sanitize_video, spawn_frames from facefusion.ffprobe import probe_audio_entries, probe_video_entries from facefusion.filesystem import copy_file, is_image from facefusion.temp_helper import clear_temp_directory, create_temp_directory, get_temp_file_path, resolve_temp_frame_paths @@ -284,3 +284,29 @@ def test_sanitize_video() -> None: assert probe_video_entries(output_paths[1], [ 'codec_name' ]).get('codec_name') == 'hevc' +def test_fix_audio_encoder() -> None: + assert fix_audio_encoder('avi', 'libopus') == 'aac' + assert fix_audio_encoder('m4v', 'libopus') == 'aac' + assert fix_audio_encoder('mpeg', 'libopus') == 'aac' + assert fix_audio_encoder('wmv', 'libopus') == 'aac' + assert fix_audio_encoder('mov', 'flac') == 'aac' + assert fix_audio_encoder('mov', 'libopus') == 'aac' + assert fix_audio_encoder('mxf', 'libopus') == 'pcm_s16le' + assert fix_audio_encoder('webm', 'aac') == 'libopus' + assert fix_audio_encoder('mp4', 'aac') == 'aac' + assert fix_audio_encoder('avi', 'aac') == 'aac' + + +def test_fix_video_encoder() -> None: + assert fix_video_encoder('m4v', 'libx265') == 'libx264' + assert fix_video_encoder('mpeg', 'libx265') == 'libx264' + assert fix_video_encoder('mxf', 'libx265') == 'libx264' + assert fix_video_encoder('wmv', 'libx265') == 'libx264' + assert fix_video_encoder('mkv', 'rawvideo') == 'libx264' + assert fix_video_encoder('mp4', 'rawvideo') == 'libx264' + assert fix_video_encoder('mov', 'libvpx-vp9') == 'libx264' + assert fix_video_encoder('webm', 'libx264') == 'libvpx-vp9' + assert fix_video_encoder('mp4', 'libx265') == 'libx265' + assert fix_video_encoder('avi', 'rawvideo') == 'rawvideo' + +