include audio file functions

tests/test_audio.py

@@ -1,7 +1,7 @@
 import subprocess
 import pytest
 
-#from facefusion.audio import get_audio_frame
+from facefusion.audio import get_audio_frame, read_static_audio
 from facefusion.download import conditional_download
 
 
@@ -15,8 +15,12 @@ def before_all() -> None:
 
 
 def test_get_audio_frame() -> None:
-	pass
-	# todo: testing
-	#assert get_audio_frame('.assets/examples/source.mp3', 25) is not None
-	# assert get_audio_frame('.assets/examples/source.wav', 25) is not None
-	#assert get_audio_frame('invalid', 25) is None
+	assert get_audio_frame('.assets/examples/source.mp3', 25) is not None
+	assert get_audio_frame('.assets/examples/source.wav', 25) is not None
+	assert get_audio_frame('invalid', 25) is None
+
+
+def test_read_static_audio() -> None:
+	assert len(read_static_audio('.assets/examples/source.mp3', 25)) == 91
+	assert len(read_static_audio('.assets/examples/source.wav', 25)) == 91
+	assert read_static_audio('invalid', 25) is None

tests/test_ffmpeg.py

@@ -5,7 +5,7 @@ import pytest
 import facefusion.globals
 from facefusion.filesystem import get_temp_directory_path, create_temp, clear_temp
 from facefusion.download import conditional_download
-from facefusion.ffmpeg import extract_frames# #read_audio_buffer, replace_audio
+from facefusion.ffmpeg import extract_frames, read_audio_buffer
 
 
 @pytest.fixture(scope = 'module', autouse = True)
@@ -103,12 +103,9 @@ def test_extract_frames_with_trim_end() -> None:
 
 
 def test_read_audio_buffer() -> None:
-	# todo: testing
-	pass
-	# todo: if you can use different values, then test not just against None or not None
-	#assert read_audio_buffer('.assets/examples/source.mp3', 1, 1) is not None
-	#assert read_audio_buffer('.assets/examples/source.wav', 1, 1) is not None
-	#assert read_audio_buffer('.assets/examples/invalid.mp3', 1, 1) is None
+	assert isinstance(read_audio_buffer('.assets/examples/source.mp3', 1, 1), bytes)
+	assert isinstance(read_audio_buffer('.assets/examples/source.wav', 1, 1), bytes)
+	assert read_audio_buffer('.assets/examples/invalid.mp3', 1, 1) is None
 
 
 def test_replace_audio() -> None: