Upload 23 files

.install/LICENSE.md

@@ -0,0 +1,3 @@
+CC BY-NC license
+
+Copyright (c) 2024 Henry Ruhs

.install/facefusion.nsi

@@ -0,0 +1,183 @@
+!include MUI2.nsh
+!include nsDialogs.nsh
+!include LogicLib.nsh
+
+RequestExecutionLevel admin
+ManifestDPIAware true
+
+Name 'FaceFusion 2.6.1'
+OutFile 'FaceFusion_2.6.1.exe'
+
+!define MUI_ICON 'facefusion.ico'
+
+!insertmacro MUI_PAGE_DIRECTORY
+Page custom InstallPage PostInstallPage
+!insertmacro MUI_PAGE_INSTFILES
+!insertmacro MUI_LANGUAGE English
+
+Var UseDefault
+Var UseCuda
+Var UseDirectMl
+Var UseOpenVino
+
+Function .onInit
+	StrCpy $INSTDIR 'C:\FaceFusion'
+FunctionEnd
+
+Function InstallPage
+	nsDialogs::Create 1018
+	!insertmacro MUI_HEADER_TEXT 'Choose Your Accelerator' 'Choose your accelerator based on the graphics card.'
+
+	${NSD_CreateRadioButton} 0 40u 100% 10u 'Default'
+	Pop $UseDefault
+
+	${NSD_CreateRadioButton} 0 55u 100% 10u 'CUDA (NVIDIA)'
+	Pop $UseCuda
+
+	${NSD_CreateRadioButton} 0 70u 100% 10u 'DirectML (AMD, Intel, NVIDIA)'
+	Pop $UseDirectMl
+
+	${NSD_CreateRadioButton} 0 85u 100% 10u 'OpenVINO (Intel)'
+	Pop $UseOpenVino
+
+	${NSD_Check} $UseDefault
+
+	nsDialogs::Show
+FunctionEnd
+
+Function PostInstallPage
+	${NSD_GetState} $UseDefault $UseDefault
+	${NSD_GetState} $UseCuda $UseCuda
+	${NSD_GetState} $UseDirectMl $UseDirectMl
+	${NSD_GetState} $UseOpenVino $UseOpenVino
+FunctionEnd
+
+Function Destroy
+	${If} ${Silent}
+		Quit
+	${Else}
+		Abort
+	${EndIf}
+FunctionEnd
+
+Section 'Prepare Your Platform'
+	DetailPrint 'Install GIT'
+	inetc::get 'https://github.com/git-for-windows/git/releases/download/v2.45.2.windows.1/Git-2.45.2-64-bit.exe' '$TEMP\Git.exe'
+	ExecWait '$TEMP\Git.exe /CURRENTUSER /VERYSILENT /DIR=$LOCALAPPDATA\Programs\Git' $0
+	Delete '$TEMP\Git.exe'
+
+	${If} $0 > 0
+		DetailPrint 'Git installation aborted with error code $0'
+		Call Destroy
+	${EndIf}
+
+	DetailPrint 'Uninstall Conda'
+	ExecWait '$LOCALAPPDATA\Programs\Miniconda3\Uninstall-Miniconda3.exe /S _?=$LOCALAPPDATA\Programs\Miniconda3'
+	RMDir /r '$LOCALAPPDATA\Programs\Miniconda3'
+
+	DetailPrint 'Install Conda'
+	inetc::get 'https://repo.anaconda.com/miniconda/Miniconda3-py310_24.3.0-0-Windows-x86_64.exe' '$TEMP\Miniconda3.exe'
+	ExecWait '$TEMP\Miniconda3.exe /InstallationType=JustMe /AddToPath=1 /S /D=$LOCALAPPDATA\Programs\Miniconda3' $1
+	Delete '$TEMP\Miniconda3.exe'
+
+	${If} $1 > 0
+		DetailPrint 'Conda installation aborted with error code $1'
+		Call Destroy
+	${EndIf}
+SectionEnd
+
+Section 'Download Your Copy'
+	SetOutPath $INSTDIR
+
+	DetailPrint 'Download Your Copy'
+	RMDir /r $INSTDIR
+	nsExec::Exec '$LOCALAPPDATA\Programs\Git\cmd\git.exe clone https://github.com/facefusion/facefusion --branch 2.6.1 .'
+SectionEnd
+
+Section 'Setup Your Environment'
+	DetailPrint 'Setup Your Environment'
+	nsExec::Exec '$LOCALAPPDATA\Programs\Miniconda3\Scripts\conda.exe init --all'
+	nsExec::Exec '$LOCALAPPDATA\Programs\Miniconda3\Scripts\conda.exe create --name facefusion python=3.10 --yes'
+SectionEnd
+
+Section 'Create Install Batch'
+	SetOutPath $INSTDIR
+
+	FileOpen $0 install-ffmpeg.bat w
+	FileOpen $1 install-accelerator.bat w
+	FileOpen $2 install-application.bat w
+
+	FileWrite $0 '@echo off && conda activate facefusion && conda install conda-forge::ffmpeg=7.0.1 --yes'
+	${If} $UseCuda == 1
+		FileWrite $1 '@echo off && conda activate facefusion && conda install cudatoolkit=11.8 cudnn=8.9.2.26 conda-forge::gputil=1.4.0 conda-forge::zlib-wapi --yes'
+		FileWrite $2 '@echo off && conda activate facefusion && python install.py --onnxruntime cuda-11.8'
+	${ElseIf} $UseDirectMl == 1
+		FileWrite $2 '@echo off && conda activate facefusion && python install.py --onnxruntime directml'
+	${ElseIf} $UseOpenVino == 1
+		FileWrite $1 '@echo off && conda activate facefusion && conda install conda-forge::openvino=2023.1.0 --yes'
+		FileWrite $2 '@echo off && conda activate facefusion && python install.py --onnxruntime openvino'
+	${Else}
+		FileWrite $2 '@echo off && conda activate facefusion && python install.py --onnxruntime default'
+	${EndIf}
+
+	FileClose $0
+	FileClose $1
+	FileClose $2
+SectionEnd
+
+Section 'Install Your FFmpeg'
+	SetOutPath $INSTDIR
+
+	DetailPrint 'Install Your FFmpeg'
+	nsExec::ExecToLog 'install-ffmpeg.bat'
+SectionEnd
+
+Section 'Install Your Accelerator'
+	SetOutPath $INSTDIR
+
+	DetailPrint 'Install Your Accelerator'
+	nsExec::ExecToLog 'install-accelerator.bat'
+SectionEnd
+
+Section 'Install The Application'
+	SetOutPath $INSTDIR
+
+	DetailPrint 'Install The Application'
+	nsExec::ExecToLog 'install-application.bat'
+SectionEnd
+
+Section 'Create Run Batch'
+	SetOutPath $INSTDIR
+	FileOpen $0 run.bat w
+	FileWrite $0 '@echo off && conda activate facefusion && python run.py %*'
+	FileClose $0
+SectionEnd
+
+Section 'Register The Application'
+	DetailPrint 'Register The Application'
+
+	CreateDirectory $SMPROGRAMS\FaceFusion
+	CreateShortcut '$SMPROGRAMS\FaceFusion\FaceFusion.lnk' $INSTDIR\run.bat '--open-browser' $INSTDIR\.install\facefusion.ico
+	CreateShortcut '$SMPROGRAMS\FaceFusion\FaceFusion Benchmark.lnk' $INSTDIR\run.bat '--ui-layouts benchmark --open-browser' $INSTDIR\.install\facefusion.ico
+	CreateShortcut '$SMPROGRAMS\FaceFusion\FaceFusion Webcam.lnk' $INSTDIR\run.bat '--ui-layouts webcam --open-browser' $INSTDIR\.install\facefusion.ico
+
+	CreateShortcut $DESKTOP\FaceFusion.lnk $INSTDIR\run.bat '--open-browser' $INSTDIR\.install\facefusion.ico
+
+	WriteUninstaller $INSTDIR\Uninstall.exe
+
+	WriteRegStr HKLM SOFTWARE\Microsoft\Windows\CurrentVersion\Uninstall\FaceFusion DisplayName 'FaceFusion'
+	WriteRegStr HKLM SOFTWARE\Microsoft\Windows\CurrentVersion\Uninstall\FaceFusion DisplayVersion '2.6.0'
+	WriteRegStr HKLM SOFTWARE\Microsoft\Windows\CurrentVersion\Uninstall\FaceFusion Publisher 'Henry Ruhs'
+	WriteRegStr HKLM SOFTWARE\Microsoft\Windows\CurrentVersion\Uninstall\FaceFusion InstallLocation $INSTDIR
+	WriteRegStr HKLM SOFTWARE\Microsoft\Windows\CurrentVersion\Uninstall\FaceFusion UninstallString $INSTDIR\uninstall.exe
+SectionEnd
+
+Section 'Uninstall'
+	nsExec::Exec '$LOCALAPPDATA\Programs\Miniconda3\Scripts\conda.exe env remove --name facefusion --yes'
+
+	Delete $DESKTOP\FaceFusion.lnk
+	RMDir /r $SMPROGRAMS\FaceFusion
+	RMDir /r $INSTDIR
+
+	DeleteRegKey HKLM SOFTWARE\Microsoft\Windows\CurrentVersion\Uninstall\FaceFusion
+SectionEnd

tests/test_audio.py

@@ -0,0 +1,26 @@
+import subprocess
+import pytest
+
+from facefusion.audio import get_audio_frame, read_static_audio
+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.mp3'
+	])
+	subprocess.run([ 'ffmpeg', '-i', '.assets/examples/source.mp3', '.assets/examples/source.wav' ])
+
+
+def test_get_audio_frame() -> 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)) == 280
+	assert len(read_static_audio('.assets/examples/source.wav', 25)) == 280
+	assert read_static_audio('invalid', 25) is None

tests/test_cli_face_debugger.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-240p.mp4'
+	])
+	subprocess.run([ 'ffmpeg', '-i', '.assets/examples/target-240p.mp4', '-vframes', '1', '.assets/examples/target-240p.jpg' ])
+
+
+def test_debug_face_to_image() -> None:
+	commands = [ sys.executable, 'run.py', '--frame-processors', 'face_debugger', '-t', '.assets/examples/target-240p.jpg', '-o', '.assets/examples/test_debug_face_to_image.jpg', '--headless' ]
+	run = subprocess.run(commands, stdout = subprocess.PIPE, stderr = subprocess.STDOUT)
+
+	assert run.returncode == 0
+	assert 'image succeed' in run.stdout.decode()
+
+
+def test_debug_face_to_video() -> None:
+	commands = [ sys.executable, 'run.py', '--frame-processors', 'face_debugger', '-t', '.assets/examples/target-240p.mp4', '-o', '.assets/examples/test_debug_face_to_video.mp4', '--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_cli_face_enhancer.py

@@ -0,0 +1,32 @@
+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-240p.mp4'
+	])
+	subprocess.run([ 'ffmpeg', '-i', '.assets/examples/target-240p.mp4', '-vframes', '1', '.assets/examples/target-240p.jpg' ])
+
+
+def test_enhance_face_to_image() -> None:
+	commands = [ sys.executable, 'run.py', '--frame-processors', 'face_enhancer', '-t', '.assets/examples/target-240p.jpg', '-o', '.assets/examples/test_enhance_face_to_image.jpg', '--headless' ]
+	run = subprocess.run(commands, stdout = subprocess.PIPE, stderr = subprocess.STDOUT)
+
+	assert run.returncode == 0
+	assert 'image succeed' in run.stdout.decode()
+
+
+def test_enhance_face_to_video() -> None:
+	commands = [ sys.executable, 'run.py', '--frame-processors', 'face_enhancer', '-t', '.assets/examples/target-240p.mp4', '-o', '.assets/examples/test_enhance_face_to_video.mp4', '--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_cli_face_swapper.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-240p.mp4'
+	])
+	subprocess.run([ 'ffmpeg', '-i', '.assets/examples/target-240p.mp4', '-vframes', '1', '.assets/examples/target-240p.jpg' ])
+
+
+def test_swap_face_to_image() -> None:
+	commands = [ sys.executable, 'run.py', '--frame-processors', 'face_swapper', '-s', '.assets/examples/source.jpg', '-t', '.assets/examples/target-240p.jpg', '-o', '.assets/examples/test_swap_face_to_image.jpg', '--headless' ]
+	run = subprocess.run(commands, stdout = subprocess.PIPE, stderr = subprocess.STDOUT)
+
+	assert run.returncode == 0
+	assert 'image succeed' in run.stdout.decode()
+
+
+def test_swap_face_to_video() -> None:
+	commands = [ sys.executable, 'run.py', '--frame-processors', 'face_swapper', '-s', '.assets/examples/source.jpg', '-t', '.assets/examples/target-240p.mp4', '-o', '.assets/examples/test_swap_face_to_video.mp4', '--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_cli_frame_colorizer.py

@@ -0,0 +1,32 @@
+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-240p.mp4'
+	])
+	subprocess.run([ 'ffmpeg', '-i', '.assets/examples/target-240p.mp4', '-vframes', '1', '-vf', 'hue=s=0', '.assets/examples/target-240p-0sat.jpg' ])
+	subprocess.run([ 'ffmpeg', '-i', '.assets/examples/target-240p.mp4', '-vf', 'hue=s=0', '.assets/examples/target-240p-0sat.mp4' ])
+
+
+def test_colorize_frame_to_image() -> None:
+	commands = [ sys.executable, 'run.py', '--frame-processors', 'frame_colorizer', '-t', '.assets/examples/target-240p-0sat.jpg', '-o', '.assets/examples/test_colorize_frame_to_image.jpg', '--headless' ]
+	run = subprocess.run(commands, stdout = subprocess.PIPE, stderr = subprocess.STDOUT)
+
+	assert run.returncode == 0
+	assert 'image succeed' in run.stdout.decode()
+
+
+def test_colorize_frame_to_video() -> None:
+	commands = [ sys.executable, 'run.py', '--frame-processors', 'frame_colorizer', '-t', '.assets/examples/target-240p-0sat.mp4', '-o', '.assets/examples/test_colorize_frame_to_video.mp4', '--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_cli_frame_enhancer.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-240p.mp4'
+	])
+	subprocess.run([ 'ffmpeg', '-i', '.assets/examples/target-240p.mp4', '-vframes', '1', '.assets/examples/target-240p.jpg' ])
+
+
+def test_enhance_frame_to_image() -> None:
+	commands = [ sys.executable, 'run.py', '--frame-processors', 'frame_enhancer', '-t', '.assets/examples/target-240p.jpg', '-o', '.assets/examples/test_enhance_frame_to_image.jpg', '--headless' ]
+	run = subprocess.run(commands, stdout = subprocess.PIPE, stderr = subprocess.STDOUT)
+
+	assert run.returncode == 0
+	assert 'image succeed' in run.stdout.decode()
+
+
+def test_enhance_frame_to_video() -> None:
+	commands = [ sys.executable, 'run.py', '--frame-processors', 'frame_enhancer', '-t', '.assets/examples/target-240p.mp4', '-o', '.assets/examples/test_enhance_frame_to_video.mp4', '--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_cli_lip_syncer.py

@@ -0,0 +1,32 @@
+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/source.mp3',
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples/target-240p.mp4'
+	])
+	subprocess.run([ 'ffmpeg', '-i', '.assets/examples/target-240p.mp4', '-vframes', '1', '.assets/examples/target-240p.jpg' ])
+
+
+def test_sync_lip_to_image() -> None:
+	commands = [ sys.executable, 'run.py', '--frame-processors', 'lip_syncer', '-s', '.assets/examples/source.mp3', '-t', '.assets/examples/target-240p.jpg', '-o', '.assets/examples/test_sync_lip_to_image.jpg', '--headless' ]
+	run = subprocess.run(commands, stdout = subprocess.PIPE, stderr = subprocess.STDOUT)
+
+	assert run.returncode == 0
+	assert 'image succeed' in run.stdout.decode()
+
+
+def test_sync_lip_to_video() -> None:
+	commands = [ sys.executable, 'run.py', '--frame-processors', 'lip_syncer', '-s', '.assets/examples/source.mp3', '-t', '.assets/examples/target-240p.mp4', '-o', '.assets/examples/test_sync_lip_to_video.mp4', '--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, 1.0, 0.05) == [ 0.0, 0.05, 0.10, 0.15, 0.20, 0.25, 0.30, 0.35, 0.40, 0.45, 0.50, 0.55, 0.60, 0.65, 0.70, 0.75, 0.80, 0.85, 0.90, 0.95, 1.0 ]

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.py

@@ -0,0 +1,27 @@
+from facefusion.execution import encode_execution_providers, decode_execution_providers, has_execution_provider, apply_execution_provider_options
+
+
+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_has_execution_provider() -> None:
+	assert has_execution_provider('CPUExecutionProvider') is True
+	assert has_execution_provider('InvalidExecutionProvider') is False
+
+
+def test_multiple_execution_providers() -> None:
+	execution_provider_with_options =\
+	[
+		'CPUExecutionProvider',
+		('CUDAExecutionProvider',
+		{
+			'device_id': '1',
+			'cudnn_conv_algo_search': 'DEFAULT'
+		})
+	]
+	assert apply_execution_provider_options('1', [ 'CPUExecutionProvider', 'CUDAExecutionProvider' ]) == execution_provider_with_options

tests/test_face_analyser.py

@@ -0,0 +1,103 @@
+import subprocess
+import pytest
+
+import facefusion.globals
+from facefusion.download import conditional_download
+from facefusion.face_analyser import pre_check, clear_face_analyser, get_one_face
+from facefusion.typing import Face
+from facefusion.vision import read_static_image
+
+
+@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'
+	])
+	subprocess.run([ 'ffmpeg', '-i', '.assets/examples/source.jpg', '-vf', 'crop=iw*0.8:ih*0.8', '.assets/examples/source-80crop.jpg' ])
+	subprocess.run([ 'ffmpeg', '-i', '.assets/examples/source.jpg', '-vf', 'crop=iw*0.7:ih*0.7', '.assets/examples/source-70crop.jpg' ])
+	subprocess.run([ 'ffmpeg', '-i', '.assets/examples/source.jpg', '-vf', 'crop=iw*0.6:ih*0.6', '.assets/examples/source-60crop.jpg' ])
+
+
+@pytest.fixture(autouse = True)
+def before_each() -> None:
+	facefusion.globals.face_detector_score = 0.5
+	facefusion.globals.face_landmarker_score = 0.5
+	facefusion.globals.face_recognizer_model = 'arcface_inswapper'
+	clear_face_analyser()
+
+
+def test_get_one_face_with_retinaface() -> None:
+	facefusion.globals.face_detector_model = 'retinaface'
+	facefusion.globals.face_detector_size = '320x320'
+
+	pre_check()
+	source_paths =\
+	[
+		'.assets/examples/source.jpg',
+		'.assets/examples/source-80crop.jpg',
+		'.assets/examples/source-70crop.jpg',
+		'.assets/examples/source-60crop.jpg'
+	]
+	for source_path in source_paths:
+		source_frame = read_static_image(source_path)
+		face = get_one_face(source_frame)
+
+		assert isinstance(face, Face)
+
+
+def test_get_one_face_with_scrfd() -> None:
+	facefusion.globals.face_detector_model = 'scrfd'
+	facefusion.globals.face_detector_size = '640x640'
+
+	pre_check()
+	source_paths =\
+	[
+		'.assets/examples/source.jpg',
+		'.assets/examples/source-80crop.jpg',
+		'.assets/examples/source-70crop.jpg',
+		'.assets/examples/source-60crop.jpg'
+	]
+	for source_path in source_paths:
+		source_frame = read_static_image(source_path)
+		face = get_one_face(source_frame)
+
+		assert isinstance(face, Face)
+
+
+def test_get_one_face_with_yoloface() -> None:
+	facefusion.globals.face_detector_model = 'yoloface'
+	facefusion.globals.face_detector_size = '640x640'
+
+	pre_check()
+	source_paths =\
+	[
+		'.assets/examples/source.jpg',
+		'.assets/examples/source-80crop.jpg',
+		'.assets/examples/source-70crop.jpg',
+		'.assets/examples/source-60crop.jpg'
+	]
+	for source_path in source_paths:
+		source_frame = read_static_image(source_path)
+		face = get_one_face(source_frame)
+
+		assert isinstance(face, Face)
+
+
+def test_get_one_face_with_yunet() -> None:
+	facefusion.globals.face_detector_model = 'yunet'
+	facefusion.globals.face_detector_size = '640x640'
+
+	pre_check()
+	source_paths =\
+	[
+		'.assets/examples/source.jpg',
+		'.assets/examples/source-80crop.jpg',
+		'.assets/examples/source-70crop.jpg',
+		'.assets/examples/source-60crop.jpg'
+	]
+	for source_path in source_paths:
+		source_frame = read_static_image(source_path)
+		face = get_one_face(source_frame)
+
+		assert isinstance(face, Face)

tests/test_ffmpeg.py

@@ -0,0 +1,113 @@
+import glob
+import subprocess
+import pytest
+
+import facefusion.globals
+from facefusion import process_manager
+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
+
+
+@pytest.fixture(scope = 'module', autouse = True)
+def before_all() -> None:
+	process_manager.start()
+	conditional_download('.assets/examples',
+	[
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples/source.jpg',
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples/source.mp3',
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples/target-240p.mp4'
+	])
+	subprocess.run([ 'ffmpeg', '-i', '.assets/examples/source.mp3', '.assets/examples/source.wav' ])
+	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_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)
+
+
+def test_read_audio_buffer() -> 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

tests/test_filesystem.py

@@ -0,0 +1,91 @@
+import shutil
+import pytest
+
+from facefusion.common_helper import is_windows
+from facefusion.download import conditional_download
+from facefusion.filesystem import get_file_size, is_file, is_directory, is_audio, has_audio, is_image, has_image, is_video, filter_audio_paths, filter_image_paths, list_directory, sanitize_path_for_windows
+
+
+@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/source.mp3',
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples/target-240p.mp4'
+	])
+	shutil.copyfile('.assets/examples/source.jpg', '.assets/examples/söurce.jpg')
+
+
+def test_get_file_size() -> None:
+	assert get_file_size('.assets/examples/source.jpg') > 0
+	assert get_file_size('invalid') == 0
+
+
+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_audio() -> None:
+	assert is_audio('.assets/examples/source.mp3') is True
+	assert is_audio('.assets/examples/source.jpg') is False
+	assert is_audio('invalid') is False
+
+
+def test_has_audio() -> None:
+	assert has_audio([ '.assets/examples/source.mp3' ]) is True
+	assert has_audio([ '.assets/examples/source.mp3', '.assets/examples/source.jpg' ]) is True
+	assert has_audio([ '.assets/examples/source.jpg', '.assets/examples/source.jpg' ]) is False
+	assert has_audio([ '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_has_image() -> None:
+	assert has_image([ '.assets/examples/source.jpg' ]) is True
+	assert has_image([ '.assets/examples/source.jpg', '.assets/examples/source.mp3' ]) is True
+	assert has_image([ '.assets/examples/source.mp3', '.assets/examples/source.mp3' ]) is False
+	assert has_image([ '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_filter_audio_paths() -> None:
+	assert filter_audio_paths([ '.assets/examples/source.jpg', '.assets/examples/source.mp3' ]) == [ '.assets/examples/source.mp3' ]
+	assert filter_audio_paths([ '.assets/examples/source.jpg', '.assets/examples/source.jpg' ]) == []
+	assert filter_audio_paths([ 'invalid' ]) == []
+
+
+def test_filter_image_paths() -> None:
+	assert filter_image_paths([ '.assets/examples/source.jpg', '.assets/examples/source.mp3' ]) == [ '.assets/examples/source.jpg' ]
+	assert filter_image_paths([ '.assets/examples/source.mp3', '.assets/examples/source.mp3' ]) == []
+	assert filter_audio_paths([ 'invalid' ]) == []
+
+
+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
+
+
+@pytest.mark.skip()
+def test_sanitize_path_for_windows() -> None:
+	if is_windows():
+		assert sanitize_path_for_windows('.assets/examples/söurce.jpg') == 'ASSETS~1/examples/SURCE~1.JPG'
+		assert sanitize_path_for_windows('invalid') is None

tests/test_memory.py

@@ -0,0 +1,8 @@
+from facefusion.common_helper import is_linux, is_macos
+from facefusion.memory import limit_system_memory
+
+
+def test_limit_system_memory() -> None:
+	assert limit_system_memory(4) is True
+	if is_linux() or is_macos():
+		assert limit_system_memory(1024) is False

tests/test_normalizer.py

@@ -0,0 +1,30 @@
+from facefusion.common_helper import is_linux, is_macos
+from facefusion.normalizer import normalize_output_path, normalize_padding, normalize_fps
+
+
+def test_normalize_output_path() -> None:
+	if is_linux() or is_macos():
+		assert normalize_output_path('.assets/examples/target-240p.mp4', '.assets/examples/target-240p.mp4') == '.assets/examples/target-240p.mp4'
+		assert normalize_output_path('.assets/examples/target-240p.mp4', '.assets/examples').startswith('.assets/examples/target-240p')
+		assert normalize_output_path('.assets/examples/target-240p.mp4', '.assets/examples').endswith('.mp4')
+		assert normalize_output_path('.assets/examples/target-240p.mp4', '.assets/examples/output.mp4') == '.assets/examples/output.mp4'
+	assert normalize_output_path('.assets/examples/target-240p.mp4', '.assets/examples/invalid') is None
+	assert normalize_output_path('.assets/examples/target-240p.mp4', '.assets/invalid/output.mp4') is None
+	assert normalize_output_path('.assets/examples/target-240p.mp4', 'invalid') is None
+	assert normalize_output_path('.assets/examples/target-240p.mp4', None) is None
+	assert normalize_output_path(None, '.assets/examples/output.mp4') 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_process_manager.py

@@ -0,0 +1,22 @@
+from facefusion.process_manager import set_process_state, is_processing, is_stopping, is_pending, start, stop, end
+
+
+def test_start() -> None:
+	set_process_state('pending')
+	start()
+
+	assert is_processing()
+
+
+def test_stop() -> None:
+	set_process_state('processing')
+	stop()
+
+	assert is_stopping()
+
+
+def test_end() -> None:
+	set_process_state('processing')
+	end()
+
+	assert is_pending()

tests/test_vision.py

@@ -0,0 +1,109 @@
+import subprocess
+import pytest
+
+from facefusion.download import conditional_download
+from facefusion.vision import detect_image_resolution, restrict_image_resolution, create_image_resolutions, get_video_frame, count_video_frame_total, detect_video_fps, restrict_video_fps, detect_video_resolution, restrict_video_resolution, create_video_resolutions, normalize_resolution, pack_resolution, unpack_resolution
+
+
+@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', '-vframes', '1', '.assets/examples/target-240p.jpg' ])
+	subprocess.run([ 'ffmpeg', '-i', '.assets/examples/target-1080p.mp4', '-vframes', '1', '.assets/examples/target-1080p.jpg' ])
+	subprocess.run([ 'ffmpeg', '-i', '.assets/examples/target-240p.mp4', '-vframes', '1', '-vf', 'transpose=0', '.assets/examples/target-240p-90deg.jpg' ])
+	subprocess.run([ 'ffmpeg', '-i', '.assets/examples/target-1080p.mp4', '-vframes', '1', '-vf', 'transpose=0', '.assets/examples/target-1080p-90deg.jpg' ])
+	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_detect_image_resolution() -> None:
+	assert detect_image_resolution('.assets/examples/target-240p.jpg') == (426, 226)
+	assert detect_image_resolution('.assets/examples/target-240p-90deg.jpg') == (226, 426)
+	assert detect_image_resolution('.assets/examples/target-1080p.jpg') == (2048, 1080)
+	assert detect_image_resolution('.assets/examples/target-1080p-90deg.jpg') == (1080, 2048)
+	assert detect_image_resolution('invalid') is None
+
+
+def test_restrict_image_resolution() -> None:
+	assert restrict_image_resolution('.assets/examples/target-1080p.jpg', (426, 226)) == (426, 226)
+	assert restrict_image_resolution('.assets/examples/target-1080p.jpg', (2048, 1080)) == (2048, 1080)
+	assert restrict_image_resolution('.assets/examples/target-1080p.jpg', (4096, 2160)) == (2048, 1080)
+
+
+def test_create_image_resolutions() -> None:
+	assert create_image_resolutions((426, 226)) == [ '106x56', '212x112', '320x170', '426x226', '640x340', '852x452', '1064x564', '1278x678', '1492x792', '1704x904' ]
+	assert create_image_resolutions((226, 426)) == [ '56x106', '112x212', '170x320', '226x426', '340x640', '452x852', '564x1064', '678x1278', '792x1492', '904x1704' ]
+	assert create_image_resolutions((2048, 1080)) == [ '512x270', '1024x540', '1536x810', '2048x1080', '3072x1620', '4096x2160', '5120x2700', '6144x3240', '7168x3780', '8192x4320' ]
+	assert create_image_resolutions((1080, 2048)) == [ '270x512', '540x1024', '810x1536', '1080x2048', '1620x3072', '2160x4096', '2700x5120', '3240x6144', '3780x7168', '4320x8192' ]
+	assert create_image_resolutions(None) == []
+
+
+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_restrict_video_fps() -> None:
+	assert restrict_video_fps('.assets/examples/target-1080p.mp4', 20.0) == 20.0
+	assert restrict_video_fps('.assets/examples/target-1080p.mp4', 25.0) == 25.0
+	assert restrict_video_fps('.assets/examples/target-1080p.mp4', 60.0) == 25.0
+
+
+def test_detect_video_resolution() -> None:
+	assert detect_video_resolution('.assets/examples/target-240p.mp4') == (426, 226)
+	assert detect_video_resolution('.assets/examples/target-240p-90deg.mp4') == (226, 426)
+	assert detect_video_resolution('.assets/examples/target-1080p.mp4') == (2048, 1080)
+	assert detect_video_resolution('.assets/examples/target-1080p-90deg.mp4') == (1080, 2048)
+	assert detect_video_resolution('invalid') is None
+
+
+def test_restrict_video_resolution() -> None:
+	assert restrict_video_resolution('.assets/examples/target-1080p.mp4', (426, 226)) == (426, 226)
+	assert restrict_video_resolution('.assets/examples/target-1080p.mp4', (2048, 1080)) == (2048, 1080)
+	assert restrict_video_resolution('.assets/examples/target-1080p.mp4', (4096, 2160)) == (2048, 1080)
+
+
+def test_create_video_resolutions() -> None:
+	assert create_video_resolutions((426, 226)) == [ '426x226', '452x240', '678x360', '904x480', '1018x540', '1358x720', '2036x1080', '2714x1440', '4072x2160', '8144x4320' ]
+	assert create_video_resolutions((226, 426)) == [ '226x426', '240x452', '360x678', '480x904', '540x1018', '720x1358', '1080x2036', '1440x2714', '2160x4072', '4320x8144' ]
+	assert create_video_resolutions((2048, 1080)) == [ '456x240', '682x360', '910x480', '1024x540', '1366x720', '2048x1080', '2730x1440', '4096x2160', '8192x4320' ]
+	assert create_video_resolutions((1080, 2048)) == [ '240x456', '360x682', '480x910', '540x1024', '720x1366', '1080x2048', '1440x2730', '2160x4096', '4320x8192' ]
+	assert create_video_resolutions(None) == []
+
+
+def test_normalize_resolution() -> None:
+	assert normalize_resolution((2.5, 2.5)) == (2, 2)
+	assert normalize_resolution((3.0, 3.0)) == (4, 4)
+	assert normalize_resolution((6.5, 6.5)) == (6, 6)
+
+
+def test_pack_resolution() -> None:
+	assert pack_resolution((1, 1)) == '0x0'
+	assert pack_resolution((2, 2)) == '2x2'
+
+
+def test_unpack_resolution() -> None:
+	assert unpack_resolution('0x0') == (0, 0)
+	assert unpack_resolution('2x2') == (2, 2)

tests/test_wording.py

@@ -0,0 +1,7 @@
+from facefusion import wording
+
+
+def test_get() -> None:
+	assert wording.get('python_not_supported')
+	assert wording.get('help.source')
+	assert wording.get('invalid') is None