Upload 11 files

tests/test_cli.py

@@ -0,0 +1,31 @@
+import subprocess
+import sys
+import pytest
+
+from facefusion.download import conditional_download
+
+
+@pytest.fixture(scope = 'module', autouse = True)
+def before_all() -> None:
+	conditional_download('.assets/examples',
+	[
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples/source.jpg',
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples/target-1080p.mp4'
+	])
+	subprocess.run([ 'ffmpeg', '-i', '.assets/examples/target-1080p.mp4', '-vframes', '1', '.assets/examples/target-1080p.jpg' ])
+
+
+def test_image_to_image() -> None:
+	commands = [ sys.executable, 'run.py', '-s', '.assets/examples/source.jpg', '-t', '.assets/examples/target-1080p.jpg', '-o', '.assets/examples', '--headless' ]
+	run = subprocess.run(commands, stdout = subprocess.PIPE, stderr = subprocess.STDOUT)
+
+	assert run.returncode == 0
+	assert 'image succeed' in run.stdout.decode()
+
+
+def test_image_to_video() -> None:
+	commands = [ sys.executable, 'run.py', '-s', '.assets/examples/source.jpg', '-t', '.assets/examples/target-1080p.mp4', '-o', '.assets/examples', '--trim-frame-end', '10', '--headless' ]
+	run = subprocess.run(commands, stdout = subprocess.PIPE, stderr = subprocess.STDOUT)
+
+	assert run.returncode == 0
+	assert 'video succeed' in run.stdout.decode()

tests/test_common_helper.py

@@ -0,0 +1,15 @@
+from facefusion.common_helper import create_metavar, create_int_range, create_float_range
+
+
+def test_create_metavar() -> None:
+	assert create_metavar([ 1, 2, 3, 4, 5 ]) == '[1-5]'
+
+
+def test_create_int_range() -> None:
+	assert create_int_range(0, 2, 1) == [ 0, 1, 2 ]
+	assert create_float_range(0, 1, 1) == [ 0, 1 ]
+
+
+def test_create_float_range() -> None:
+	assert create_float_range(0.0, 1.0, 0.5) == [ 0.0, 0.5, 1.0 ]
+	assert create_float_range(0.0, 0.2, 0.05) == [ 0.0, 0.05, 0.10, 0.15, 0.20 ]

tests/test_config.py

@@ -0,0 +1,96 @@
+from configparser import ConfigParser
+import pytest
+
+from facefusion import config
+
+
+@pytest.fixture(scope = 'module', autouse = True)
+def before_all() -> None:
+	config.CONFIG = ConfigParser()
+	config.CONFIG.read_dict(
+	{
+		'str':
+		{
+			'valid': 'a',
+			'unset': ''
+		},
+		'int':
+		{
+			'valid': '1',
+			'unset': ''
+		},
+		'float':
+		{
+			'valid': '1.0',
+			'unset': ''
+		},
+		'bool':
+		{
+			'valid': 'True',
+			'unset': ''
+		},
+		'str_list':
+		{
+			'valid': 'a b c',
+			'unset': ''
+		},
+		'int_list':
+		{
+			'valid': '1 2 3',
+			'unset': ''
+		},
+		'float_list':
+		{
+			'valid': '1.0 2.0 3.0',
+			'unset': ''
+		}
+	})
+
+
+def test_get_str_value() -> None:
+	assert config.get_str_value('str.valid') == 'a'
+	assert config.get_str_value('str.unset', 'b') == 'b'
+	assert config.get_str_value('str.unset') is None
+	assert config.get_str_value('str.invalid') is None
+
+
+def test_get_int_value() -> None:
+	assert config.get_int_value('int.valid') == 1
+	assert config.get_int_value('int.unset', '1') == 1
+	assert config.get_int_value('int.unset') is None
+	assert config.get_int_value('int.invalid') is None
+
+
+def test_get_float_value() -> None:
+	assert config.get_float_value('float.valid') == 1.0
+	assert config.get_float_value('float.unset', '1.0') == 1.0
+	assert config.get_float_value('float.unset') is None
+	assert config.get_float_value('float.invalid') is None
+
+
+def test_get_bool_value() -> None:
+	assert config.get_bool_value('bool.valid') is True
+	assert config.get_bool_value('bool.unset', 'False') is False
+	assert config.get_bool_value('bool.unset') is None
+	assert config.get_bool_value('bool.invalid') is None
+
+
+def test_get_str_list() -> None:
+	assert config.get_str_list('str_list.valid') == [ 'a', 'b', 'c' ]
+	assert config.get_str_list('str_list.unset', 'c b a') == [ 'c', 'b', 'a' ]
+	assert config.get_str_list('str_list.unset') is None
+	assert config.get_str_list('str_list.invalid') is None
+
+
+def test_get_int_list() -> None:
+	assert config.get_int_list('int_list.valid') == [ 1, 2, 3 ]
+	assert config.get_int_list('int_list.unset', '3 2 1') == [ 3, 2, 1 ]
+	assert config.get_int_list('int_list.unset') is None
+	assert config.get_int_list('int_list.invalid') is None
+
+
+def test_get_float_list() -> None:
+	assert config.get_float_list('float_list.valid') == [ 1.0, 2.0, 3.0 ]
+	assert config.get_float_list('float_list.unset', '3.0 2.0 1.0') == [ 3.0, 2.0, 1.0 ]
+	assert config.get_float_list('float_list.unset') is None
+	assert config.get_float_list('float_list.invalid') is None

tests/test_download.py

@@ -0,0 +1,23 @@
+import pytest
+
+from facefusion.download import conditional_download, get_download_size, is_download_done
+
+
+@pytest.fixture(scope = 'module', autouse = True)
+def before_all() -> None:
+	conditional_download('.assets/examples',
+	[
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples/target-240p.mp4'
+	])
+
+
+def test_get_download_size() -> None:
+	assert get_download_size('https://github.com/facefusion/facefusion-assets/releases/download/examples/target-240p.mp4') == 191675
+	assert get_download_size('https://github.com/facefusion/facefusion-assets/releases/download/examples/target-360p.mp4') == 370732
+	assert get_download_size('invalid') == 0
+
+
+def test_is_download_done() -> None:
+	assert is_download_done('https://github.com/facefusion/facefusion-assets/releases/download/examples/target-240p.mp4', '.assets/examples/target-240p.mp4') is True
+	assert is_download_done('https://github.com/facefusion/facefusion-assets/releases/download/examples/target-240p.mp4','invalid') is False
+	assert is_download_done('invalid', 'invalid') is False

tests/test_execution_helper.py

@@ -0,0 +1,26 @@
+from facefusion.execution_helper import encode_execution_providers, decode_execution_providers, apply_execution_provider_options, map_torch_backend
+
+
+def test_encode_execution_providers() -> None:
+	assert encode_execution_providers([ 'CPUExecutionProvider' ]) == [ 'cpu' ]
+
+
+def test_decode_execution_providers() -> None:
+	assert decode_execution_providers([ 'cpu' ]) == [ 'CPUExecutionProvider' ]
+
+
+def test_multiple_execution_providers() -> None:
+	execution_provider_with_options =\
+	[
+		'CPUExecutionProvider',
+		('CUDAExecutionProvider',
+		{
+			'cudnn_conv_algo_search': 'DEFAULT'
+		})
+	]
+	assert apply_execution_provider_options([ 'CPUExecutionProvider', 'CUDAExecutionProvider' ]) == execution_provider_with_options
+
+
+def test_map_device() -> None:
+	assert map_torch_backend([ 'CPUExecutionProvider' ]) == 'cpu'
+	assert map_torch_backend([ 'CPUExecutionProvider', 'CUDAExecutionProvider' ]) == 'cuda'

tests/test_ffmpeg.py

@@ -0,0 +1,100 @@
+import glob
+import subprocess
+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
+
+
+@pytest.fixture(scope = 'module', autouse = True)
+def before_all() -> None:
+	conditional_download('.assets/examples',
+	[
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples/source.jpg',
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples/target-240p.mp4'
+	])
+	subprocess.run([ 'ffmpeg', '-i', '.assets/examples/target-240p.mp4', '-vf', 'fps=25', '.assets/examples/target-240p-25fps.mp4' ])
+	subprocess.run([ 'ffmpeg', '-i', '.assets/examples/target-240p.mp4', '-vf', 'fps=30', '.assets/examples/target-240p-30fps.mp4' ])
+	subprocess.run([ 'ffmpeg', '-i', '.assets/examples/target-240p.mp4', '-vf', 'fps=60', '.assets/examples/target-240p-60fps.mp4' ])
+
+
+@pytest.fixture(scope = 'function', autouse = True)
+def before_each() -> None:
+	facefusion.globals.trim_frame_start = None
+	facefusion.globals.trim_frame_end = None
+	facefusion.globals.temp_frame_quality = 80
+	facefusion.globals.temp_frame_format = 'jpg'
+
+
+def test_extract_frames() -> None:
+	target_paths =\
+	[
+		'.assets/examples/target-240p-25fps.mp4',
+		'.assets/examples/target-240p-30fps.mp4',
+		'.assets/examples/target-240p-60fps.mp4'
+	]
+	for target_path in target_paths:
+		temp_directory_path = get_temp_directory_path(target_path)
+		create_temp(target_path)
+
+		assert extract_frames(target_path, '452x240', 30.0) is True
+		assert len(glob.glob1(temp_directory_path, '*.jpg')) == 324
+
+		clear_temp(target_path)
+
+
+def test_extract_frames_with_trim_start() -> None:
+	facefusion.globals.trim_frame_start = 224
+	data_provider =\
+	[
+		('.assets/examples/target-240p-25fps.mp4', 55),
+		('.assets/examples/target-240p-30fps.mp4', 100),
+		('.assets/examples/target-240p-60fps.mp4', 212)
+	]
+	for target_path, frame_total in data_provider:
+		temp_directory_path = get_temp_directory_path(target_path)
+		create_temp(target_path)
+
+		assert extract_frames(target_path, '452x240', 30.0) is True
+		assert len(glob.glob1(temp_directory_path, '*.jpg')) == frame_total
+
+		clear_temp(target_path)
+
+
+def test_extract_frames_with_trim_start_and_trim_end() -> None:
+	facefusion.globals.trim_frame_start = 124
+	facefusion.globals.trim_frame_end = 224
+	data_provider =\
+	[
+		('.assets/examples/target-240p-25fps.mp4', 120),
+		('.assets/examples/target-240p-30fps.mp4', 100),
+		('.assets/examples/target-240p-60fps.mp4', 50)
+	]
+	for target_path, frame_total in data_provider:
+		temp_directory_path = get_temp_directory_path(target_path)
+		create_temp(target_path)
+
+		assert extract_frames(target_path, '452x240', 30.0) is True
+		assert len(glob.glob1(temp_directory_path, '*.jpg')) == frame_total
+
+		clear_temp(target_path)
+
+
+def test_extract_frames_with_trim_end() -> None:
+	facefusion.globals.trim_frame_end = 100
+	data_provider =\
+	[
+		('.assets/examples/target-240p-25fps.mp4', 120),
+		('.assets/examples/target-240p-30fps.mp4', 100),
+		('.assets/examples/target-240p-60fps.mp4', 50)
+	]
+	for target_path, frame_total in data_provider:
+		temp_directory_path = get_temp_directory_path(target_path)
+		create_temp(target_path)
+
+		assert extract_frames(target_path, '426x240', 30.0) is True
+		assert len(glob.glob1(temp_directory_path, '*.jpg')) == frame_total
+
+		clear_temp(target_path)

tests/test_filesystem.py

@@ -0,0 +1,48 @@
+import pytest
+
+from facefusion.download import conditional_download
+from facefusion.filesystem import is_file, is_directory, is_image, are_images, is_video, list_directory
+
+
+@pytest.fixture(scope = 'module', autouse = True)
+def before_all() -> None:
+	conditional_download('.assets/examples',
+	[
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples/source.jpg'
+	])
+
+
+def test_is_file() -> None:
+	assert is_file('.assets/examples/source.jpg') is True
+	assert is_file('.assets/examples') is False
+	assert is_file('invalid') is False
+
+
+def test_is_directory() -> None:
+	assert is_directory('.assets/examples') is True
+	assert is_directory('.assets/examples/source.jpg') is False
+	assert is_directory('invalid') is False
+
+
+def test_is_image() -> None:
+	assert is_image('.assets/examples/source.jpg') is True
+	assert is_image('.assets/examples/target-240p.mp4') is False
+	assert is_image('invalid') is False
+
+
+def test_are_images() -> None:
+	assert are_images([ '.assets/examples/source.jpg' ]) is True
+	assert are_images([ '.assets/examples/source.jpg', '.assets/examples/target-240p.mp4' ]) is False
+	assert are_images([ 'invalid' ]) is False
+
+
+def test_is_video() -> None:
+	assert is_video('.assets/examples/target-240p.mp4') is True
+	assert is_video('.assets/examples/source.jpg') is False
+	assert is_video('invalid') is False
+
+
+def test_list_directory() -> None:
+	assert list_directory('.assets/examples')
+	assert list_directory('.assets/examples/source.jpg') is None
+	assert list_directory('invalid') is None

tests/test_memory.py

@@ -0,0 +1,9 @@
+import platform
+
+from facefusion.memory import limit_system_memory
+
+
+def test_limit_system_memory() -> None:
+	assert limit_system_memory(4) is True
+	if platform.system().lower() == 'darwin' or platform.system().lower() == 'linux':
+		assert limit_system_memory(1024) is False

tests/test_normalizer.py

@@ -0,0 +1,32 @@
+import platform
+
+from facefusion.normalizer import normalize_output_path, normalize_padding, normalize_fps
+
+
+def test_normalize_output_path() -> None:
+	if platform.system().lower() != 'windows':
+		assert normalize_output_path([ '.assets/examples/source.jpg' ], None, '.assets/examples/target-240p.mp4') == '.assets/examples/target-240p.mp4'
+		assert normalize_output_path(None, '.assets/examples/target-240p.mp4', '.assets/examples/target-240p.mp4') == '.assets/examples/target-240p.mp4'
+		assert normalize_output_path(None, '.assets/examples/target-240p.mp4', '.assets/examples') == '.assets/examples/target-240p.mp4'
+		assert normalize_output_path([ '.assets/examples/source.jpg' ], '.assets/examples/target-240p.mp4', '.assets/examples') == '.assets/examples/source-target-240p.mp4'
+		assert normalize_output_path(None, '.assets/examples/target-240p.mp4', '.assets/examples/output.mp4') == '.assets/examples/output.mp4'
+		assert normalize_output_path(None, '.assets/examples/target-240p.mp4', '.assets/output.mov') == '.assets/output.mp4'
+	assert normalize_output_path(None, '.assets/examples/target-240p.mp4', '.assets/examples/invalid') is None
+	assert normalize_output_path(None, '.assets/examples/target-240p.mp4', '.assets/invalid/output.mp4') is None
+	assert normalize_output_path(None, '.assets/examples/target-240p.mp4', 'invalid') is None
+	assert normalize_output_path([ '.assets/examples/source.jpg' ], '.assets/examples/target-240p.mp4', None) is None
+
+
+def test_normalize_padding() -> None:
+	assert normalize_padding([ 0, 0, 0, 0 ]) == (0, 0, 0, 0)
+	assert normalize_padding([ 1 ]) == (1, 1, 1, 1)
+	assert normalize_padding([ 1, 2 ]) == (1, 2, 1, 2)
+	assert normalize_padding([ 1, 2, 3 ]) == (1, 2, 3, 2)
+	assert normalize_padding(None) is None
+
+
+def test_normalize_fps() -> None:
+	assert normalize_fps(0.0) == 1.0
+	assert normalize_fps(25.0) == 25.0
+	assert normalize_fps(61.0) == 60.0
+	assert normalize_fps(None) is None

tests/test_vision.py

@@ -0,0 +1,63 @@
+import subprocess
+import pytest
+
+from facefusion.download import conditional_download
+from facefusion.vision import get_video_frame, count_video_frame_total, detect_video_fps, detect_video_resolution, pack_resolution, unpack_resolution, create_video_resolutions
+
+
+@pytest.fixture(scope = 'module', autouse = True)
+def before_all() -> None:
+	conditional_download('.assets/examples',
+	[
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples/source.jpg',
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples/target-240p.mp4',
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples/target-1080p.mp4'
+	])
+	subprocess.run([ 'ffmpeg', '-i', '.assets/examples/target-240p.mp4', '-vf', 'fps=25', '.assets/examples/target-240p-25fps.mp4' ])
+	subprocess.run([ 'ffmpeg', '-i', '.assets/examples/target-240p.mp4', '-vf', 'fps=30', '.assets/examples/target-240p-30fps.mp4' ])
+	subprocess.run([ 'ffmpeg', '-i', '.assets/examples/target-240p.mp4', '-vf', 'fps=60', '.assets/examples/target-240p-60fps.mp4' ])
+	subprocess.run([ 'ffmpeg', '-i', '.assets/examples/target-240p.mp4', '-vf', 'transpose=0', '.assets/examples/target-240p-90deg.mp4' ])
+	subprocess.run([ 'ffmpeg', '-i', '.assets/examples/target-1080p.mp4', '-vf', 'transpose=0', '.assets/examples/target-1080p-90deg.mp4' ])
+
+
+def test_get_video_frame() -> None:
+	assert get_video_frame('.assets/examples/target-240p-25fps.mp4') is not None
+	assert get_video_frame('invalid') is None
+
+
+def test_count_video_frame_total() -> None:
+	assert count_video_frame_total('.assets/examples/target-240p-25fps.mp4') == 270
+	assert count_video_frame_total('.assets/examples/target-240p-30fps.mp4') == 324
+	assert count_video_frame_total('.assets/examples/target-240p-60fps.mp4') == 648
+	assert count_video_frame_total('invalid') == 0
+
+
+def test_detect_video_fps() -> None:
+	assert detect_video_fps('.assets/examples/target-240p-25fps.mp4') == 25.0
+	assert detect_video_fps('.assets/examples/target-240p-30fps.mp4') == 30.0
+	assert detect_video_fps('.assets/examples/target-240p-60fps.mp4') == 60.0
+	assert detect_video_fps('invalid') is None
+
+
+def test_detect_video_resolution() -> None:
+	assert detect_video_resolution('.assets/examples/target-240p.mp4') == (426.0, 226.0)
+	assert detect_video_resolution('.assets/examples/target-1080p.mp4') == (2048.0, 1080.0)
+	assert detect_video_resolution('invalid') is None
+
+
+def test_pack_resolution() -> None:
+	assert pack_resolution((1.0, 1.0)) == '0x0'
+	assert pack_resolution((2.0, 2.0)) == '2x2'
+
+
+def test_unpack_resolution() -> None:
+	assert unpack_resolution('0x0') == (0, 0)
+	assert unpack_resolution('2x2') == (2, 2)
+
+
+def test_create_video_resolutions() -> None:
+	assert create_video_resolutions('.assets/examples/target-240p.mp4') == [ '426x226', '452x240', '678x360', '904x480', '1018x540', '1358x720', '2036x1080', '2714x1440', '4072x2160' ]
+	assert create_video_resolutions('.assets/examples/target-240p-90deg.mp4') == [ '226x426', '240x452', '360x678', '480x904', '540x1018', '720x1358', '1080x2036', '1440x2714', '2160x4072' ]
+	assert create_video_resolutions('.assets/examples/target-1080p.mp4') == [ '456x240', '682x360', '910x480', '1024x540', '1366x720', '2048x1080', '2730x1440', '4096x2160' ]
+	assert create_video_resolutions('.assets/examples/target-1080p-90deg.mp4') == [ '240x456', '360x682', '480x910', '540x1024', '720x1366', '1080x2048', '1440x2730', '2160x4096' ]
+	assert create_video_resolutions('invalid') is None