diff --git a/tests/test_codec_aom_decoder.py b/tests/test_codec_aom_decoder.py index 5d6a902d..b327d30b 100644 --- a/tests/test_codec_aom_decoder.py +++ b/tests/test_codec_aom_decoder.py @@ -7,7 +7,7 @@ from tests.assert_helper import get_test_example_file, get_test_examples_directo from facefusion import state_manager from facefusion.codecs.aom_decoder import create, decode, destroy from facefusion.codecs.aom_encoder import create as create_encoder, encode -from facefusion.common_helper import is_macos +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 aom as aom_module @@ -35,17 +35,14 @@ def test_decode() -> None: video_buffer = cv2.cvtColor(vision_frame, cv2.COLOR_BGR2YUV_I420).tobytes() video_resolution = (vision_frame.shape[1], vision_frame.shape[0]) aom_encoder = create_encoder(video_resolution, 1000, 1, 0) - encoded_buffer = encode(aom_encoder, video_buffer, video_resolution, 0) - aom_pointer = decode(create(1), encoded_buffer) + encode_buffer = encode(aom_encoder, video_buffer, video_resolution, 0) + aom_decoder = create(1) - assert aom_pointer is not None - assert aom_pointer.get('resolution')[0] >= video_resolution[0] - assert aom_pointer.get('resolution')[1] >= video_resolution[1] - assert len(aom_pointer.get('buffer')) == aom_pointer.get('resolution')[0] * aom_pointer.get('resolution')[1] * 3 // 2 - assert decode(create(1), bytes()) is None + if is_linux() or is_windows(): + assert create_hash(decode(aom_decoder, encode_buffer).get('buffer')) == 'e3c0ebd8' if is_macos(): - assert create_hash(bytes(aom_pointer.get('buffer'))) == 'c8c6fdaa' + assert create_hash(decode(aom_decoder, encode_buffer).get('buffer')) == 'c8c6fdaa' def test_destroy() -> None: diff --git a/tests/test_codec_aom_encoder.py b/tests/test_codec_aom_encoder.py index f0d2b1a3..f1e41db1 100644 --- a/tests/test_codec_aom_encoder.py +++ b/tests/test_codec_aom_encoder.py @@ -24,7 +24,9 @@ def before_all() -> None: def test_create() -> None: assert create((320, 240), 1000, 8, 16) - assert create((0, 0), 0, 0, 0) is None + + with patch('facefusion.codecs.aom_encoder.aom_module.create_static_library', return_value = None): + assert create((320, 240), 1000, 8, 16) is None def test_encode() -> None: diff --git a/tests/test_codec_opus_decoder.py b/tests/test_codec_opus_decoder.py index b7329e2a..b809c7f7 100644 --- a/tests/test_codec_opus_decoder.py +++ b/tests/test_codec_opus_decoder.py @@ -7,7 +7,7 @@ from tests.assert_helper import get_test_example_file, get_test_examples_directo from facefusion import state_manager from facefusion.codecs.opus_decoder import create, decode, destroy from facefusion.codecs.opus_encoder import create as create_encoder, encode -from facefusion.common_helper import is_macos +from facefusion.common_helper import is_linux, is_macos, is_windows from facefusion.download import conditional_download from facefusion.ffmpeg import read_audio_buffer from facefusion.hash_helper import create_hash @@ -25,21 +25,23 @@ def before_all() -> None: def test_create() -> None: assert create(48000, 2) - assert create(0, 0) is None + + with patch('facefusion.codecs.opus_decoder.opus_module.create_static_library', return_value = None): + assert create(48000, 2) is None def test_decode() -> None: audio_buffer = read_audio_buffer(get_test_example_file('source.mp3'), 48000, 16, 2) audio_sample = numpy.frombuffer(audio_buffer, dtype = numpy.int16).astype(numpy.float32) / 32768.0 opus_encoder = create_encoder(48000, 2) - encoded_buffer = encode(opus_encoder, audio_sample.tobytes(), 960) + encode_buffer = encode(opus_encoder, audio_sample.tobytes(), 960) opus_decoder = create(48000, 2) - decoded_buffer = decode(opus_decoder, encoded_buffer, 960, 2) - assert len(decoded_buffer) == 960 * 2 * 4 + if is_linux() or is_windows(): + assert create_hash(decode(opus_decoder, encode_buffer, 960, 2)) == 'cadd63d1' if is_macos(): - assert create_hash(decoded_buffer) == '92f7997d' + assert create_hash(decode(opus_decoder, encode_buffer, 960, 2)) == '92f7997d' def test_destroy() -> None: diff --git a/tests/test_codec_opus_encoder.py b/tests/test_codec_opus_encoder.py index 48622bb9..798f11f5 100644 --- a/tests/test_codec_opus_encoder.py +++ b/tests/test_codec_opus_encoder.py @@ -24,7 +24,9 @@ def before_all() -> None: def test_create() -> None: assert create(48000, 2) - assert create(0, 0) is None + + with patch('facefusion.codecs.opus_encoder.opus_module.create_static_library', return_value = None): + assert create(48000, 2) is None def test_encode() -> None: diff --git a/tests/test_codec_vpx_decoder.py b/tests/test_codec_vpx_decoder.py index b64e29e9..f465f1d3 100644 --- a/tests/test_codec_vpx_decoder.py +++ b/tests/test_codec_vpx_decoder.py @@ -7,7 +7,7 @@ from tests.assert_helper import get_test_example_file, get_test_examples_directo from facefusion import state_manager from facefusion.codecs.vpx_decoder import create, decode, destroy from facefusion.codecs.vpx_encoder import create as create_encoder, encode -from facefusion.common_helper import is_macos +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 @@ -35,16 +35,14 @@ def test_decode() -> None: video_buffer = cv2.cvtColor(vision_frame, cv2.COLOR_BGR2YUV_I420).tobytes() video_resolution = (vision_frame.shape[1], vision_frame.shape[0]) vpx_encoder = create_encoder(video_resolution, 1000, 1, 0) - encoded_buffer = encode(vpx_encoder, video_buffer, video_resolution, 0) - vpx_pointer = decode(create(1), encoded_buffer) + encode_buffer = encode(vpx_encoder, video_buffer, video_resolution, 0) + vpx_decoder = create(1) - assert vpx_pointer is not None - assert vpx_pointer.get('resolution') == video_resolution - assert len(vpx_pointer.get('buffer')) == video_resolution[0] * video_resolution[1] * 3 // 2 - assert decode(create(1), bytes()) is None + if is_linux() or is_windows(): + assert create_hash(decode(vpx_decoder, encode_buffer).get('buffer')) == 'dc9c8864' if is_macos(): - assert create_hash(bytes(vpx_pointer.get('buffer'))) == '87450f70' + assert create_hash(decode(vpx_decoder, encode_buffer).get('buffer')) == '87450f70' def test_destroy() -> None: diff --git a/tests/test_codec_vpx_encoder.py b/tests/test_codec_vpx_encoder.py index ec9aacca..1a15a741 100644 --- a/tests/test_codec_vpx_encoder.py +++ b/tests/test_codec_vpx_encoder.py @@ -24,7 +24,9 @@ def before_all() -> None: def test_create() -> None: assert create((320, 240), 1000, 8, 16) - assert create((0, 0), 0, 0, 0) is None + + with patch('facefusion.codecs.vpx_encoder.vpx_module.create_static_library', return_value = None): + assert create((320, 240), 1000, 8, 16) is None def test_encode() -> None: