From 02d11435a68baed1fadbe51c0cfa8f429029438a Mon Sep 17 00:00:00 2001 From: henryruhs Date: Sun, 26 Jan 2025 04:16:28 +0100 Subject: [PATCH] Simplify set_audio_sample_size() --- facefusion/ffmpeg_builder.py | 4 ++-- tests/test_ffmpeg_builder.py | 8 ++++---- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/facefusion/ffmpeg_builder.py b/facefusion/ffmpeg_builder.py index 0b6d01a..58a95f6 100644 --- a/facefusion/ffmpeg_builder.py +++ b/facefusion/ffmpeg_builder.py @@ -114,9 +114,9 @@ def set_audio_sample_rate(audio_sample_rate : int) -> Commands: def set_audio_sample_size(audio_sample_size : int) -> Commands: if audio_sample_size == 16: - return [ '-f', 's16le', '-acodec', 'pcm_s16le' ] + return [ '-f', 's16le' ] if audio_sample_size == 32: - return [ '-f', 's32le', '-acodec', 'pcm_s32le' ] + return [ '-f', 's32le' ] return [] diff --git a/tests/test_ffmpeg_builder.py b/tests/test_ffmpeg_builder.py index 445b9e8..66b1aab 100644 --- a/tests/test_ffmpeg_builder.py +++ b/tests/test_ffmpeg_builder.py @@ -12,7 +12,7 @@ def test_chain() -> None: assert chain(ffmpeg_builder.set_progress()) == [ '-progress' ] -def test_stream_mode() -> None: +def test_set_stream_mode() -> None: assert set_stream_mode('udp') == [ '-f', 'mpegts' ] assert set_stream_mode('v4l2') == [ '-f', 'v4l2' ] @@ -24,9 +24,9 @@ def test_select_frame_range() -> None: assert select_frame_range(None, None, 30) == [ '-vf', 'fps=30' ] -def test_audio_sample_size() -> None: - assert set_audio_sample_size(16) == [ '-f', 's16le', '-acodec', 'pcm_s16le' ] - assert set_audio_sample_size(32) == [ '-f', 's32le', '-acodec', 'pcm_s32le' ] +def test_set_audio_sample_size() -> None: + assert set_audio_sample_size(16) == [ '-f', 's16le' ] + assert set_audio_sample_size(32) == [ '-f', 's32le' ] def test_set_audio_quality() -> None: