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

Revisit and cleanup ffprobe integration (#1027)

Browse Source 
* Revisit and cleanup ffprobe integration

* Revisit and cleanup ffprobe integration
This commit is contained in:
Henry Ruhs
2026-03-17 14:01:49 +01:00
committed by henryruhs
parent 1b8e3b00b0
commit ba8093e844
6 changed files with 47 additions and 91 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_ffprobe_builder.py
+5 -14
View File
@@ -1,7 +1,7 @@
from shutil import which
from facefusion import ffprobe_builder
from facefusion.ffprobe_builder import chain, format_to_key_value, format_to_value, run, select_audio_stream, set_input, show_stream_entries
from facefusion.ffprobe_builder import chain, format_to_key_value, format_to_value, run, set_input, show_entries
def test_run() -> None:
@@ -10,24 +10,15 @@ def test_run() -> None:
def test_chain() -> None:
assert chain(
ffprobe_builder.select_audio_stream(0)
) == [ '-select_streams', 'a:0' ]
assert chain(
ffprobe_builder.select_audio_stream(0),
ffprobe_builder.show_stream_entries([ 'sample_rate' ]),
ffprobe_builder.show_entries([ 'sample_rate' ]),
ffprobe_builder.format_to_value(),
ffprobe_builder.set_input('audio.mp3')
) == [ '-select_streams', 'a:0', '-show_entries', 'stream=sample_rate', '-of', 'default=noprint_wrappers=1:nokey=1', 'audio.mp3' ]
def test_select_audio_stream() -> None:
assert select_audio_stream(0) == [ '-select_streams', 'a:0' ]
) == [ '-show_entries', 'stream=sample_rate', '-of', 'default=noprint_wrappers=1:nokey=1', 'audio.mp3' ]
def test_show_entries() -> None:
assert show_stream_entries([ 'sample_rate' ]) == [ '-show_entries', 'stream=sample_rate' ]
assert show_stream_entries([ 'channels' ]) == [ '-show_entries', 'stream=channels' ]
assert show_stream_entries([ 'duration', 'sample_rate' ]) == [ '-show_entries', 'stream=duration,sample_rate' ]
assert show_entries([ 'duration' ]) == [ '-show_entries', 'stream=duration' ]
assert show_entries([ 'duration', 'sample_rate']) == [ '-show_entries', 'stream=duration,sample_rate' ]
def test_format_to_value() -> None: