cleanup decoders tests (#1122)

* cleanup mostly decoders tests, also a bit encoders

* cleanup mostly decoders tests, also a bit encoders

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:

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:

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:

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:

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:

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: