Upload 48 files

LICENSE.md

@@ -0,0 +1,3 @@
+OpenRAIL-AS license
+
+Copyright (c) 2025 Henry Ruhs

README.md

@@ -1,12 +1,60 @@
----
-title: Raphtalia
-emoji: 🌍
-colorFrom: green
-colorTo: gray
-sdk: gradio
-sdk_version: 5.22.0
-app_file: app.py
-pinned: false
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+FaceFusion
+==========
+
+> Industry leading face manipulation platform.
+
+[![Build Status](https://img.shields.io/github/actions/workflow/status/facefusion/facefusion/ci.yml.svg?branch=master)](https://github.com/facefusion/facefusion/actions?query=workflow:ci)
+[![Coverage Status](https://img.shields.io/coveralls/facefusion/facefusion.svg)](https://coveralls.io/r/facefusion/facefusion)
+![License](https://img.shields.io/badge/license-OpenRAIL--AS-green)
+
+
+Preview
+-------
+
+![Preview](https://raw.githubusercontent.com/facefusion/facefusion/master/.github/preview.png?sanitize=true)
+
+
+Installation
+------------
+
+Be aware, the [installation](https://docs.facefusion.io/installation) needs technical skills and is not recommended for beginners. In case you are not comfortable using a terminal, our [Windows Installer](http://windows-installer.facefusion.io) and [macOS Installer](http://macos-installer.facefusion.io) get you started.
+
+
+Usage
+-----
+
+Run the command:
+
+```
+python facefusion.py [commands] [options]
+
+options:
+  -h, --help                                      show this help message and exit
+  -v, --version                                   show program's version number and exit
+
+commands:
+    run                                           run the program
+    headless-run                                  run the program in headless mode
+    batch-run                                     run the program in batch mode
+    force-download                                force automate downloads and exit
+    job-list                                      list jobs by status
+    job-create                                    create a drafted job
+    job-submit                                    submit a drafted job to become a queued job
+    job-submit-all                                submit all drafted jobs to become a queued jobs
+    job-delete                                    delete a drafted, queued, failed or completed job
+    job-delete-all                                delete all drafted, queued, failed and completed jobs
+    job-add-step                                  add a step to a drafted job
+    job-remix-step                                remix a previous step from a drafted job
+    job-insert-step                               insert a step to a drafted job
+    job-remove-step                               remove a step from a drafted job
+    job-run                                       run a queued job
+    job-run-all                                   run all queued jobs
+    job-retry                                     retry a failed job
+    job-retry-all                                 retry all failed jobs
+```
+
+
+Documentation
+-------------
+
+Read the [documentation](https://docs.facefusion.io) for a deep dive.

facefusion.ini

@@ -0,0 +1,118 @@
+[paths]
+temp_path =
+jobs_path =
+source_paths =
+target_path =
+output_path =
+
+[patterns]
+source_pattern =
+target_pattern =
+output_pattern =
+
+[face_detector]
+face_detector_model =
+face_detector_size =
+face_detector_angles =
+face_detector_score =
+
+[face_landmarker]
+face_landmarker_model =
+face_landmarker_score =
+
+[face_selector]
+face_selector_mode =
+face_selector_order =
+face_selector_age_start =
+face_selector_age_end =
+face_selector_gender =
+face_selector_race =
+reference_face_position =
+reference_face_distance =
+reference_frame_number =
+
+[face_masker]
+face_occluder_model =
+face_parser_model =
+face_mask_types =
+face_mask_blur =
+face_mask_padding =
+face_mask_regions =
+
+[frame_extraction]
+trim_frame_start =
+trim_frame_end =
+temp_frame_format =
+keep_temp =
+
+[output_creation]
+output_image_quality =
+output_image_resolution =
+output_audio_encoder =
+output_audio_quality =
+output_audio_volume =
+output_video_encoder =
+output_video_preset =
+output_video_quality =
+output_video_resolution =
+output_video_fps =
+skip_audio =
+
+[processors]
+processors =
+age_modifier_model =
+age_modifier_direction =
+deep_swapper_model =
+deep_swapper_morph =
+expression_restorer_model =
+expression_restorer_factor =
+face_debugger_items =
+face_editor_model =
+face_editor_eyebrow_direction =
+face_editor_eye_gaze_horizontal =
+face_editor_eye_gaze_vertical =
+face_editor_eye_open_ratio =
+face_editor_lip_open_ratio =
+face_editor_mouth_grim =
+face_editor_mouth_pout =
+face_editor_mouth_purse =
+face_editor_mouth_smile =
+face_editor_mouth_position_horizontal =
+face_editor_mouth_position_vertical =
+face_editor_head_pitch =
+face_editor_head_yaw =
+face_editor_head_roll =
+face_enhancer_model =
+face_enhancer_blend =
+face_enhancer_weight =
+face_swapper_model =
+face_swapper_pixel_boost =
+frame_colorizer_model =
+frame_colorizer_size =
+frame_colorizer_blend =
+frame_enhancer_model =
+frame_enhancer_blend =
+lip_syncer_model =
+
+[uis]
+open_browser =
+ui_layouts =
+ui_workflow =
+
+[execution]
+execution_device_id =
+execution_providers =
+execution_thread_count =
+execution_queue_count =
+
+[download]
+download_providers =
+download_scope =
+
+[memory]
+video_memory_strategy =
+system_memory_limit =
+
+[misc]
+log_level =
+halt_on_error =

facefusion.py

@@ -0,0 +1,10 @@
+#!/usr/bin/env python3
+
+import os
+
+os.environ['OMP_NUM_THREADS'] = '1'
+
+from facefusion import core
+
+if __name__ == '__main__':
+	core.cli()

install.py

@@ -0,0 +1,10 @@
+#!/usr/bin/env python3
+
+import os
+
+os.environ['SYSTEM_VERSION_COMPAT'] = '0'
+
+from facefusion import installer
+
+if __name__ == '__main__':
+	installer.cli()

mypy.ini

@@ -0,0 +1,7 @@
+[mypy]
+check_untyped_defs = True
+disallow_any_generics = True
+disallow_untyped_calls = True
+disallow_untyped_defs = True
+ignore_missing_imports = True
+strict_optional = False

requirements.txt

@@ -0,0 +1,21 @@
+gradio
+gradio-rangeslider
+numpy
+onnx
+onnxruntime
+opencv-python
+psutil
+tqdm
+scipy
+
+
+
+#gradio==5.20.0
+#gradio-rangeslider==0.0.8
+#numpy==2.2.3
+#onnx==1.17.0
+#onnxruntime==1.21.0
+#opencv-python==4.11.0.86
+#psutil==7.0.0
+#tqdm==4.67.1
+#scipy==1.15.2

run.py

@@ -0,0 +1,8 @@
+import sys
+import subprocess
+
+python_executable = sys.executable  # Jo bhi active Python hai usko detect karega
+command = [python_executable, "facefusion.py", "run"]
+
+print(f"🚀 Launching FaceFusion with {python_executable}...")
+subprocess.run(command)

tests/helper.py

@@ -0,0 +1,44 @@
+import os
+import tempfile
+
+from facefusion.filesystem import create_directory, is_directory, is_file, remove_directory
+from facefusion.types import JobStatus
+
+
+def is_test_job_file(file_path : str, job_status : JobStatus) -> bool:
+	return is_file(get_test_job_file(file_path, job_status))
+
+
+def get_test_job_file(file_path : str, job_status : JobStatus) -> str:
+	return os.path.join(get_test_jobs_directory(), job_status, file_path)
+
+
+def get_test_jobs_directory() -> str:
+	return os.path.join(tempfile.gettempdir(), 'facefusion-test-jobs')
+
+
+def get_test_example_file(file_path : str) -> str:
+	return os.path.join(get_test_examples_directory(), file_path)
+
+
+def get_test_examples_directory() -> str:
+	return os.path.join(tempfile.gettempdir(), 'facefusion-test-examples')
+
+
+def is_test_output_file(file_path : str) -> bool:
+	return is_file(get_test_output_file(file_path))
+
+
+def get_test_output_file(file_path : str) -> str:
+	return os.path.join(get_test_outputs_directory(), file_path)
+
+
+def get_test_outputs_directory() -> str:
+	return os.path.join(tempfile.gettempdir(), 'facefusion-test-outputs')
+
+
+def prepare_test_output_directory() -> bool:
+	test_outputs_directory = get_test_outputs_directory()
+	remove_directory(test_outputs_directory)
+	create_directory(test_outputs_directory)
+	return is_directory(test_outputs_directory)

tests/test_audio.py

@@ -0,0 +1,28 @@
+import subprocess
+
+import pytest
+
+from facefusion.audio import get_audio_frame, read_static_audio
+from facefusion.download import conditional_download
+from .helper import get_test_example_file, get_test_examples_directory
+
+
+@pytest.fixture(scope = 'module', autouse = True)
+def before_all() -> None:
+	conditional_download(get_test_examples_directory(),
+	[
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/source.mp3'
+	])
+	subprocess.run([ 'ffmpeg', '-i', get_test_example_file('source.mp3'), get_test_example_file('source.wav') ])
+
+
+def test_get_audio_frame() -> None:
+	assert hasattr(get_audio_frame(get_test_example_file('source.mp3'), 25), '__array_interface__')
+	assert hasattr(get_audio_frame(get_test_example_file('source.wav'), 25), '__array_interface__')
+	assert get_audio_frame('invalid', 25) is None
+
+
+def test_read_static_audio() -> None:
+	assert len(read_static_audio(get_test_example_file('source.mp3'), 25)) == 280
+	assert len(read_static_audio(get_test_example_file('source.wav'), 25)) == 280
+	assert read_static_audio('invalid', 25) is None

tests/test_cli_age_modifier.py

@@ -0,0 +1,38 @@
+import subprocess
+import sys
+
+import pytest
+
+from facefusion.download import conditional_download
+from facefusion.jobs.job_manager import clear_jobs, init_jobs
+from .helper import get_test_example_file, get_test_examples_directory, get_test_jobs_directory, get_test_output_file, is_test_output_file, prepare_test_output_directory
+
+
+@pytest.fixture(scope = 'module', autouse = True)
+def before_all() -> None:
+	conditional_download(get_test_examples_directory(),
+	[
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/target-240p.mp4'
+	])
+	subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), '-vframes', '1', get_test_example_file('target-240p.jpg') ])
+
+
+@pytest.fixture(scope = 'function', autouse = True)
+def before_each() -> None:
+	clear_jobs(get_test_jobs_directory())
+	init_jobs(get_test_jobs_directory())
+	prepare_test_output_directory()
+
+
+def test_modify_age_to_image() -> None:
+	commands = [ sys.executable, 'facefusion.py', 'headless-run', '--jobs-path', get_test_jobs_directory(), '--processors', 'age_modifier', '--age-modifier-direction', '100', '-t', get_test_example_file('target-240p.jpg'), '-o', get_test_output_file('test-age-face-to-image.jpg') ]
+
+	assert subprocess.run(commands).returncode == 0
+	assert is_test_output_file('test-age-face-to-image.jpg') is True
+
+
+def test_modify_age_to_video() -> None:
+	commands = [ sys.executable, 'facefusion.py', 'headless-run', '--jobs-path', get_test_jobs_directory(), '--processors', 'age_modifier', '--age-modifier-direction', '100', '-t', get_test_example_file('target-240p.mp4'), '-o', get_test_output_file('test-age-face-to-video.mp4'), '--trim-frame-end', '1' ]
+
+	assert subprocess.run(commands).returncode == 0
+	assert is_test_output_file('test-age-face-to-video.mp4') is True

tests/test_cli_batch_runner.py

@@ -0,0 +1,45 @@
+import subprocess
+import sys
+
+import pytest
+
+from facefusion.download import conditional_download
+from facefusion.jobs.job_manager import clear_jobs, init_jobs
+from .helper import get_test_example_file, get_test_examples_directory, get_test_jobs_directory, get_test_output_file, is_test_output_file, prepare_test_output_directory
+
+
+@pytest.fixture(scope = 'module', autouse = True)
+def before_all() -> None:
+	conditional_download(get_test_examples_directory(),
+	[
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/target-240p.mp4'
+	])
+	subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), '-vframes', '1', get_test_example_file('target-240p-batch-1.jpg') ])
+	subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), '-vframes', '2', get_test_example_file('target-240p-batch-2.jpg') ])
+
+
+@pytest.fixture(scope = 'function', autouse = True)
+def before_each() -> None:
+	clear_jobs(get_test_jobs_directory())
+	init_jobs(get_test_jobs_directory())
+	prepare_test_output_directory()
+
+
+def test_batch_run_targets() -> None:
+	commands = [ sys.executable, 'facefusion.py', 'batch-run', '--jobs-path', get_test_jobs_directory(), '--processors', 'face_debugger', '-t', get_test_example_file('target-240p-batch-*.jpg'), '-o', get_test_output_file('test-batch-run-targets-{index}.jpg') ]
+
+	assert subprocess.run(commands).returncode == 0
+	assert is_test_output_file('test-batch-run-targets-0.jpg') is True
+	assert is_test_output_file('test-batch-run-targets-1.jpg') is True
+	assert is_test_output_file('test-batch-run-targets-2.jpg') is False
+
+
+def test_batch_run_sources_to_targets() -> None:
+	commands = [ sys.executable, 'facefusion.py', 'batch-run', '--jobs-path', get_test_jobs_directory(), '-s', get_test_example_file('target-240p-batch-*.jpg'), '-t', get_test_example_file('target-240p-batch-*.jpg'), '-o', get_test_output_file('test-batch-run-sources-to-targets-{index}.jpg') ]
+
+	assert subprocess.run(commands).returncode == 0
+	assert is_test_output_file('test-batch-run-sources-to-targets-0.jpg') is True
+	assert is_test_output_file('test-batch-run-sources-to-targets-1.jpg') is True
+	assert is_test_output_file('test-batch-run-sources-to-targets-2.jpg') is True
+	assert is_test_output_file('test-batch-run-sources-to-targets-3.jpg') is True
+	assert is_test_output_file('test-batch-run-sources-to-targets-4.jpg') is False

tests/test_cli_expression_restorer.py

@@ -0,0 +1,38 @@
+import subprocess
+import sys
+
+import pytest
+
+from facefusion.download import conditional_download
+from facefusion.jobs.job_manager import clear_jobs, init_jobs
+from .helper import get_test_example_file, get_test_examples_directory, get_test_jobs_directory, get_test_output_file, is_test_output_file, prepare_test_output_directory
+
+
+@pytest.fixture(scope = 'module', autouse = True)
+def before_all() -> None:
+	conditional_download(get_test_examples_directory(),
+	[
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/target-240p.mp4'
+	])
+	subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), '-vframes', '1', get_test_example_file('target-240p.jpg') ])
+
+
+@pytest.fixture(scope = 'function', autouse = True)
+def before_each() -> None:
+	clear_jobs(get_test_jobs_directory())
+	init_jobs(get_test_jobs_directory())
+	prepare_test_output_directory()
+
+
+def test_restore_expression_to_image() -> None:
+	commands = [ sys.executable, 'facefusion.py', 'headless-run', '--jobs-path', get_test_jobs_directory(), '--processors', 'expression_restorer', '-t', get_test_example_file('target-240p.jpg'), '-o', get_test_output_file('test-restore-expression-to-image.jpg') ]
+
+	assert subprocess.run(commands).returncode == 0
+	assert is_test_output_file('test-restore-expression-to-image.jpg') is True
+
+
+def test_restore_expression_to_video() -> None:
+	commands = [ sys.executable, 'facefusion.py', 'headless-run', '--jobs-path', get_test_jobs_directory(), '--processors', 'expression_restorer', '-t', get_test_example_file('target-240p.mp4'), '-o', get_test_output_file('test-restore-expression-to-video.mp4'), '--trim-frame-end', '1' ]
+
+	assert subprocess.run(commands).returncode == 0
+	assert is_test_output_file('test-restore-expression-to-video.mp4') is True

tests/test_cli_face_debugger.py

@@ -0,0 +1,39 @@
+import subprocess
+import sys
+
+import pytest
+
+from facefusion.download import conditional_download
+from facefusion.jobs.job_manager import clear_jobs, init_jobs
+from .helper import get_test_example_file, get_test_examples_directory, get_test_jobs_directory, get_test_output_file, is_test_output_file, prepare_test_output_directory
+
+
+@pytest.fixture(scope = 'module', autouse = True)
+def before_all() -> None:
+	conditional_download(get_test_examples_directory(),
+	[
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/source.jpg',
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/target-240p.mp4'
+	])
+	subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), '-vframes', '1', get_test_example_file('target-240p.jpg') ])
+
+
+@pytest.fixture(scope = 'function', autouse = True)
+def before_each() -> None:
+	clear_jobs(get_test_jobs_directory())
+	init_jobs(get_test_jobs_directory())
+	prepare_test_output_directory()
+
+
+def test_debug_face_to_image() -> None:
+	commands = [ sys.executable, 'facefusion.py', 'headless-run', '--jobs-path', get_test_jobs_directory(), '--processors', 'face_debugger', '-t', get_test_example_file('target-240p.jpg'), '-o', get_test_output_file('test-debug-face-to-image.jpg') ]
+
+	assert subprocess.run(commands).returncode == 0
+	assert is_test_output_file('test-debug-face-to-image.jpg') is True
+
+
+def test_debug_face_to_video() -> None:
+	commands = [ sys.executable, 'facefusion.py', 'headless-run', '--jobs-path', get_test_jobs_directory(), '--processors', 'face_debugger', '-t', get_test_example_file('target-240p.mp4'), '-o', get_test_output_file('test-debug-face-to-video.mp4'), '--trim-frame-end', '1' ]
+
+	assert subprocess.run(commands).returncode == 0
+	assert is_test_output_file('test-debug-face-to-video.mp4') is True

tests/test_cli_face_editor.py

@@ -0,0 +1,39 @@
+import subprocess
+import sys
+
+import pytest
+
+from facefusion.download import conditional_download
+from facefusion.jobs.job_manager import clear_jobs, init_jobs
+from .helper import get_test_example_file, get_test_examples_directory, get_test_jobs_directory, get_test_output_file, is_test_output_file, prepare_test_output_directory
+
+
+@pytest.fixture(scope = 'module', autouse = True)
+def before_all() -> None:
+	conditional_download(get_test_examples_directory(),
+	[
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/source.jpg',
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/target-240p.mp4'
+	])
+	subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), '-vframes', '1', get_test_example_file('target-240p.jpg') ])
+
+
+@pytest.fixture(scope = 'function', autouse = True)
+def before_each() -> None:
+	clear_jobs(get_test_jobs_directory())
+	init_jobs(get_test_jobs_directory())
+	prepare_test_output_directory()
+
+
+def test_edit_face_to_image() -> None:
+	commands = [ sys.executable, 'facefusion.py', 'headless-run', '--jobs-path', get_test_jobs_directory(), '--processors', 'face_editor', '-t', get_test_example_file('target-240p.jpg'), '-o', get_test_output_file('test-edit-face-to-image.jpg') ]
+
+	assert subprocess.run(commands).returncode == 0
+	assert is_test_output_file('test-edit-face-to-image.jpg') is True
+
+
+def test_edit_face_to_video() -> None:
+	commands = [ sys.executable, 'facefusion.py', 'headless-run', '--jobs-path', get_test_jobs_directory(), '--processors', 'face_editor', '-t', get_test_example_file('target-240p.mp4'), '-o', get_test_output_file('test-edit-face-to-video.mp4'), '--trim-frame-end', '1' ]
+
+	assert subprocess.run(commands).returncode == 0
+	assert is_test_output_file('test-edit-face-to-video.mp4') is True

tests/test_cli_face_enhancer.py

@@ -0,0 +1,39 @@
+import subprocess
+import sys
+
+import pytest
+
+from facefusion.download import conditional_download
+from facefusion.jobs.job_manager import clear_jobs, init_jobs
+from .helper import get_test_example_file, get_test_examples_directory, get_test_jobs_directory, get_test_output_file, is_test_output_file, prepare_test_output_directory
+
+
+@pytest.fixture(scope = 'module', autouse = True)
+def before_all() -> None:
+	conditional_download(get_test_examples_directory(),
+	[
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/source.jpg',
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/target-240p.mp4'
+	])
+	subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), '-vframes', '1', get_test_example_file('target-240p.jpg') ])
+
+
+@pytest.fixture(scope = 'function', autouse = True)
+def before_each() -> None:
+	clear_jobs(get_test_jobs_directory())
+	init_jobs(get_test_jobs_directory())
+	prepare_test_output_directory()
+
+
+def test_enhance_face_to_image() -> None:
+	commands = [ sys.executable, 'facefusion.py', 'headless-run', '--jobs-path', get_test_jobs_directory(), '--processors', 'face_enhancer', '-t', get_test_example_file('target-240p.jpg'), '-o', get_test_output_file('test-enhance-face-to-image.jpg') ]
+
+	assert subprocess.run(commands).returncode == 0
+	assert is_test_output_file('test-enhance-face-to-image.jpg') is True
+
+
+def test_enhance_face_to_video() -> None:
+	commands = [ sys.executable, 'facefusion.py', 'headless-run', '--jobs-path', get_test_jobs_directory(), '--processors', 'face_enhancer', '-t', get_test_example_file('target-240p.mp4'), '-o', get_test_output_file('test-enhance-face-to-video.mp4'), '--trim-frame-end', '1' ]
+
+	assert subprocess.run(commands).returncode == 0
+	assert is_test_output_file('test-enhance-face-to-video.mp4') is True

tests/test_cli_face_swapper.py

@@ -0,0 +1,39 @@
+import subprocess
+import sys
+
+import pytest
+
+from facefusion.download import conditional_download
+from facefusion.jobs.job_manager import clear_jobs, init_jobs
+from .helper import get_test_example_file, get_test_examples_directory, get_test_jobs_directory, get_test_output_file, is_test_output_file, prepare_test_output_directory
+
+
+@pytest.fixture(scope = 'module', autouse = True)
+def before_all() -> None:
+	conditional_download(get_test_examples_directory(),
+	[
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/source.jpg',
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/target-240p.mp4'
+	])
+	subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), '-vframes', '1', get_test_example_file('target-240p.jpg') ])
+
+
+@pytest.fixture(scope = 'function', autouse = True)
+def before_each() -> None:
+	clear_jobs(get_test_jobs_directory())
+	init_jobs(get_test_jobs_directory())
+	prepare_test_output_directory()
+
+
+def test_swap_face_to_image() -> None:
+	commands = [ sys.executable, 'facefusion.py', 'headless-run', '--jobs-path', get_test_jobs_directory(), '--processors', 'face_swapper', '-s', get_test_example_file('source.jpg'), '-t', get_test_example_file('target-240p.jpg'), '-o', get_test_output_file('test-swap-face-to-image.jpg') ]
+
+	assert subprocess.run(commands).returncode == 0
+	assert is_test_output_file('test-swap-face-to-image.jpg') is True
+
+
+def test_swap_face_to_video() -> None:
+	commands = [ sys.executable, 'facefusion.py', 'headless-run', '--jobs-path', get_test_jobs_directory(), '--processors', 'face_swapper', '-s', get_test_example_file('source.jpg'), '-t', get_test_example_file('target-240p.mp4'), '-o', get_test_output_file('test-swap-face-to-video.mp4'), '--trim-frame-end', '1' ]
+
+	assert subprocess.run(commands).returncode == 0
+	assert is_test_output_file('test-swap-face-to-video.mp4') is True

tests/test_cli_frame_colorizer.py

@@ -0,0 +1,40 @@
+import subprocess
+import sys
+
+import pytest
+
+from facefusion.download import conditional_download
+from facefusion.jobs.job_manager import clear_jobs, init_jobs
+from .helper import get_test_example_file, get_test_examples_directory, get_test_jobs_directory, get_test_output_file, is_test_output_file, prepare_test_output_directory
+
+
+@pytest.fixture(scope = 'module', autouse = True)
+def before_all() -> None:
+	conditional_download(get_test_examples_directory(),
+	[
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/source.jpg',
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/target-240p.mp4'
+	])
+	subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), '-vframes', '1', '-vf', 'hue=s=0', get_test_example_file('target-240p-0sat.jpg') ])
+	subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), '-vf', 'hue=s=0', get_test_example_file('target-240p-0sat.mp4') ])
+
+
+@pytest.fixture(scope = 'function', autouse = True)
+def before_each() -> None:
+	clear_jobs(get_test_jobs_directory())
+	init_jobs(get_test_jobs_directory())
+	prepare_test_output_directory()
+
+
+def test_colorize_frame_to_image() -> None:
+	commands = [ sys.executable, 'facefusion.py', 'headless-run', '--jobs-path', get_test_jobs_directory(), '--processors', 'frame_colorizer', '-t', get_test_example_file('target-240p-0sat.jpg'), '-o', get_test_output_file('test_colorize-frame-to-image.jpg') ]
+
+	assert subprocess.run(commands).returncode == 0
+	assert is_test_output_file('test_colorize-frame-to-image.jpg') is True
+
+
+def test_colorize_frame_to_video() -> None:
+	commands = [ sys.executable, 'facefusion.py', 'headless-run', '--jobs-path', get_test_jobs_directory(), '--processors', 'frame_colorizer', '-t', get_test_example_file('target-240p-0sat.mp4'), '-o', get_test_output_file('test-colorize-frame-to-video.mp4'), '--trim-frame-end', '1' ]
+
+	assert subprocess.run(commands).returncode == 0
+	assert is_test_output_file('test-colorize-frame-to-video.mp4') is True

tests/test_cli_frame_enhancer.py

@@ -0,0 +1,39 @@
+import subprocess
+import sys
+
+import pytest
+
+from facefusion.download import conditional_download
+from facefusion.jobs.job_manager import clear_jobs, init_jobs
+from .helper import get_test_example_file, get_test_examples_directory, get_test_jobs_directory, get_test_output_file, is_test_output_file, prepare_test_output_directory
+
+
+@pytest.fixture(scope = 'module', autouse = True)
+def before_all() -> None:
+	conditional_download(get_test_examples_directory(),
+	[
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/source.jpg',
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/target-240p.mp4'
+	])
+	subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), '-vframes', '1', get_test_example_file('target-240p.jpg') ])
+
+
+@pytest.fixture(scope = 'function', autouse = True)
+def before_each() -> None:
+	clear_jobs(get_test_jobs_directory())
+	init_jobs(get_test_jobs_directory())
+	prepare_test_output_directory()
+
+
+def test_enhance_frame_to_image() -> None:
+	commands = [ sys.executable, 'facefusion.py', 'headless-run', '--jobs-path', get_test_jobs_directory(), '--processors', 'frame_enhancer', '-t', get_test_example_file('target-240p.jpg'), '-o', get_test_output_file('test-enhance-frame-to-image.jpg') ]
+
+	assert subprocess.run(commands).returncode == 0
+	assert is_test_output_file('test-enhance-frame-to-image.jpg') is True
+
+
+def test_enhance_frame_to_video() -> None:
+	commands = [ sys.executable, 'facefusion.py', 'headless-run', '--jobs-path', get_test_jobs_directory(), '--processors', 'frame_enhancer', '-t', get_test_example_file('target-240p.mp4'), '-o', get_test_output_file('test-enhance-frame-to-video.mp4'), '--trim-frame-end', '1' ]
+
+	assert subprocess.run(commands).returncode == 0
+	assert is_test_output_file('test-enhance-frame-to-video.mp4') is True

tests/test_cli_job_manager.py

@@ -0,0 +1,209 @@
+import subprocess
+import sys
+
+import pytest
+
+from facefusion.download import conditional_download
+from facefusion.jobs.job_manager import clear_jobs, count_step_total, init_jobs
+from .helper import get_test_example_file, get_test_examples_directory, get_test_jobs_directory, get_test_output_file, is_test_job_file
+
+
+@pytest.fixture(scope = 'module', autouse = True)
+def before_all() -> None:
+	conditional_download(get_test_examples_directory(),
+	[
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/source.jpg',
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/target-240p.mp4'
+	])
+	subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), '-vframes', '1', get_test_example_file('target-240p.jpg') ])
+
+
+@pytest.fixture(scope = 'function', autouse = True)
+def before_each() -> None:
+	clear_jobs(get_test_jobs_directory())
+	init_jobs(get_test_jobs_directory())
+
+
+@pytest.mark.skip()
+def test_job_list() -> None:
+	pass
+
+
+def test_job_create() -> None:
+	commands = [ sys.executable, 'facefusion.py', 'job-create', 'test-job-create', '--jobs-path', get_test_jobs_directory() ]
+
+	assert subprocess.run(commands).returncode == 0
+	assert is_test_job_file('test-job-create.json', 'drafted') is True
+
+	commands = [ sys.executable, 'facefusion.py', 'job-create', 'test-job-create', '--jobs-path', get_test_jobs_directory() ]
+
+	assert subprocess.run(commands).returncode == 1
+
+
+def test_job_submit() -> None:
+	commands = [ sys.executable, 'facefusion.py', 'job-submit', 'test-job-submit', '--jobs-path', get_test_jobs_directory() ]
+
+	assert subprocess.run(commands).returncode == 1
+
+	commands = [ sys.executable, 'facefusion.py', 'job-create', 'test-job-submit', '--jobs-path', get_test_jobs_directory() ]
+	subprocess.run(commands)
+
+	assert subprocess.run(commands).returncode == 1
+
+	commands = [ sys.executable, 'facefusion.py', 'job-add-step', 'test-job-submit', '--jobs-path', get_test_jobs_directory(), '-s', get_test_example_file('source.jpg'), '-t', get_test_example_file('target-240p.jpg'), '-o', get_test_output_file('test-job-remix-step.jpg') ]
+	subprocess.run(commands)
+
+	commands = [ sys.executable, 'facefusion.py', 'job-submit', 'test-job-submit', '--jobs-path', get_test_jobs_directory() ]
+
+	assert subprocess.run(commands).returncode == 0
+	assert is_test_job_file('test-job-submit.json', 'queued') is True
+	assert subprocess.run(commands).returncode == 1
+
+
+def test_submit_all() -> None:
+	commands = [ sys.executable, 'facefusion.py', 'job-submit-all', '--jobs-path', get_test_jobs_directory(), '--halt-on-error' ]
+
+	assert subprocess.run(commands).returncode == 1
+
+	commands = [ sys.executable, 'facefusion.py', 'job-create', 'test-job-submit-all-1', '--jobs-path', get_test_jobs_directory() ]
+	subprocess.run(commands)
+
+	commands = [ sys.executable, 'facefusion.py', 'job-create', 'test-job-submit-all-2', '--jobs-path', get_test_jobs_directory() ]
+	subprocess.run(commands)
+
+	assert subprocess.run(commands).returncode == 1
+
+	commands = [ sys.executable, 'facefusion.py', 'job-add-step', 'test-job-submit-all-1', '--jobs-path', get_test_jobs_directory(), '-s', get_test_example_file('source.jpg'), '-t', get_test_example_file('target-240p.jpg'), '-o', get_test_output_file('test-job-remix-step.jpg') ]
+	subprocess.run(commands)
+
+	commands = [ sys.executable, 'facefusion.py', 'job-add-step', 'test-job-submit-all-2', '--jobs-path', get_test_jobs_directory(), '-s', get_test_example_file('source.jpg'), '-t', get_test_example_file('target-240p.jpg'), '-o', get_test_output_file('test-job-remix-step.jpg') ]
+	subprocess.run(commands)
+
+	commands = [ sys.executable, 'facefusion.py', 'job-submit-all', '--jobs-path', get_test_jobs_directory(), '--halt-on-error' ]
+
+	assert subprocess.run(commands).returncode == 0
+	assert is_test_job_file('test-job-submit-all-1.json', 'queued') is True
+	assert is_test_job_file('test-job-submit-all-2.json', 'queued') is True
+	assert subprocess.run(commands).returncode == 1
+
+
+def test_job_delete() -> None:
+	commands = [ sys.executable, 'facefusion.py', 'job-delete', 'test-job-delete', '--jobs-path', get_test_jobs_directory() ]
+
+	assert subprocess.run(commands).returncode == 1
+
+	commands = [ sys.executable, 'facefusion.py', 'job-create', 'test-job-delete', '--jobs-path', get_test_jobs_directory() ]
+	subprocess.run(commands)
+
+	commands = [ sys.executable, 'facefusion.py', 'job-delete', 'test-job-delete', '--jobs-path', get_test_jobs_directory() ]
+
+	assert subprocess.run(commands).returncode == 0
+	assert is_test_job_file('test-job-delete.json', 'drafted') is False
+	assert subprocess.run(commands).returncode == 1
+
+
+def test_job_delete_all() -> None:
+	commands = [ sys.executable, 'facefusion.py', 'job-delete-all', '--jobs-path', get_test_jobs_directory(), '--halt-on-error' ]
+
+	assert subprocess.run(commands).returncode == 1
+
+	commands = [ sys.executable, 'facefusion.py', 'job-create', 'test-job-delete-all-1', '--jobs-path', get_test_jobs_directory() ]
+	subprocess.run(commands)
+
+	commands = [ sys.executable, 'facefusion.py', 'job-create', 'test-job-delete-all-2', '--jobs-path', get_test_jobs_directory() ]
+	subprocess.run(commands)
+
+	commands = [ sys.executable, 'facefusion.py', 'job-delete-all', '--jobs-path', get_test_jobs_directory(), '--halt-on-error' ]
+
+	assert subprocess.run(commands).returncode == 0
+	assert is_test_job_file('test-job-delete-all-1.json', 'drafted') is False
+	assert is_test_job_file('test-job-delete-all-2.json', 'drafted') is False
+	assert subprocess.run(commands).returncode == 1
+
+
+def test_job_add_step() -> None:
+	commands = [ sys.executable, 'facefusion.py', 'job-add-step', 'test-job-add-step', '--jobs-path', get_test_jobs_directory(), '-s', get_test_example_file('source.jpg'), '-t', get_test_example_file('target-240p.jpg'), '-o', get_test_output_file('test-job-remix-step.jpg') ]
+
+	assert subprocess.run(commands).returncode == 1
+	assert count_step_total('test-job-add-step') == 0
+
+	commands = [ sys.executable, 'facefusion.py', 'job-create', 'test-job-add-step', '--jobs-path', get_test_jobs_directory() ]
+	subprocess.run(commands)
+
+	commands = [ sys.executable, 'facefusion.py', 'job-add-step', 'test-job-add-step', '--jobs-path', get_test_jobs_directory(), '-s', get_test_example_file('source.jpg'), '-t', get_test_example_file('target-240p.jpg'), '-o', get_test_output_file('test-job-remix-step.jpg') ]
+
+	assert subprocess.run(commands).returncode == 0
+	assert count_step_total('test-job-add-step') == 1
+
+
+def test_job_remix() -> None:
+	commands = [ sys.executable, 'facefusion.py', 'job-remix-step', 'test-job-remix-step', '0', '--jobs-path', get_test_jobs_directory(), '-s', get_test_example_file('source.jpg'), '-o', get_test_output_file('test-job-remix-step.jpg') ]
+
+	assert subprocess.run(commands).returncode == 1
+	assert count_step_total('test-job-remix-step') == 0
+
+	commands = [ sys.executable, 'facefusion.py', 'job-create', 'test-job-remix-step', '--jobs-path', get_test_jobs_directory() ]
+	subprocess.run(commands)
+
+	commands = [ sys.executable, 'facefusion.py', 'job-add-step', 'test-job-remix-step', '--jobs-path', get_test_jobs_directory(), '-s', get_test_example_file('source.jpg'), '-t', get_test_example_file('target-240p.jpg'), '-o', get_test_output_file('test-job-remix-step.jpg') ]
+	subprocess.run(commands)
+
+	commands = [ sys.executable, 'facefusion.py', 'job-remix-step', 'test-job-remix-step', '0', '--jobs-path', get_test_jobs_directory(), '-s', get_test_example_file('source.jpg'), '-o', get_test_output_file('test-job-remix-step.jpg') ]
+
+	assert count_step_total('test-job-remix-step') == 1
+	assert subprocess.run(commands).returncode == 0
+	assert count_step_total('test-job-remix-step') == 2
+
+	commands = [ sys.executable, 'facefusion.py', 'job-remix-step', 'test-job-remix-step', '-1', '--jobs-path', get_test_jobs_directory(), '-s', get_test_example_file('source.jpg'), '-o', get_test_output_file('test-job-remix-step.jpg') ]
+
+	assert subprocess.run(commands).returncode == 0
+	assert count_step_total('test-job-remix-step') == 3
+
+
+def test_job_insert_step() -> None:
+	commands = [ sys.executable, 'facefusion.py', 'job-insert-step', 'test-job-insert-step', '0', '--jobs-path', get_test_jobs_directory(), '-s', get_test_example_file('source.jpg'), '-t', get_test_example_file('target-240p.jpg'), '-o', get_test_output_file('test-job-remix-step.jpg') ]
+
+	assert subprocess.run(commands).returncode == 1
+	assert count_step_total('test-job-insert-step') == 0
+
+	commands = [ sys.executable, 'facefusion.py', 'job-create', 'test-job-insert-step', '--jobs-path', get_test_jobs_directory() ]
+	subprocess.run(commands)
+
+	commands = [ sys.executable, 'facefusion.py', 'job-add-step', 'test-job-insert-step', '--jobs-path', get_test_jobs_directory(), '-s', get_test_example_file('source.jpg'), '-t', get_test_example_file('target-240p.jpg'), '-o', get_test_output_file('test-job-remix-step.jpg') ]
+	subprocess.run(commands)
+
+	commands = [ sys.executable, 'facefusion.py', 'job-insert-step', 'test-job-insert-step', '0', '--jobs-path', get_test_jobs_directory(), '-s', get_test_example_file('source.jpg'), '-t', get_test_example_file('target-240p.jpg'), '-o', get_test_output_file('test-job-remix-step.jpg') ]
+
+	assert count_step_total('test-job-insert-step') == 1
+	assert subprocess.run(commands).returncode == 0
+	assert count_step_total('test-job-insert-step') == 2
+
+	commands = [ sys.executable, 'facefusion.py', 'job-insert-step', 'test-job-insert-step', '-1', '--jobs-path', get_test_jobs_directory(), '-s', get_test_example_file('source.jpg'), '-t', get_test_example_file('target-240p.jpg'), '-o', get_test_output_file('test-job-remix-step.jpg') ]
+
+	assert subprocess.run(commands).returncode == 0
+	assert count_step_total('test-job-insert-step') == 3
+
+
+def test_job_remove_step() -> None:
+	commands = [ sys.executable, 'facefusion.py', 'job-remove-step', 'test-job-remove-step', '0', '--jobs-path', get_test_jobs_directory() ]
+
+	assert subprocess.run(commands).returncode == 1
+
+	commands = [ sys.executable, 'facefusion.py', 'job-create', 'test-job-remove-step', '--jobs-path', get_test_jobs_directory() ]
+	subprocess.run(commands)
+
+	commands = [ sys.executable, 'facefusion.py', 'job-add-step', 'test-job-remove-step', '--jobs-path', get_test_jobs_directory(), '-s', get_test_example_file('source.jpg'), '-t', get_test_example_file('target-240p.jpg'), '-o', get_test_output_file('test-job-remix-step.jpg') ]
+	subprocess.run(commands)
+	subprocess.run(commands)
+
+	commands = [ sys.executable, 'facefusion.py', 'job-remove-step', 'test-job-remove-step', '0', '--jobs-path', get_test_jobs_directory() ]
+
+	assert count_step_total('test-job-remove-step') == 2
+	assert subprocess.run(commands).returncode == 0
+	assert count_step_total('test-job-remove-step') == 1
+
+	commands = [ sys.executable, 'facefusion.py', 'job-remove-step', 'test-job-remove-step', '-1', '--jobs-path', get_test_jobs_directory() ]
+
+	assert subprocess.run(commands).returncode == 0
+	assert subprocess.run(commands).returncode == 1
+	assert count_step_total('test-job-remove-step') == 0

tests/test_cli_job_runner.py

@@ -0,0 +1,147 @@
+import subprocess
+import sys
+
+import pytest
+
+from facefusion.download import conditional_download
+from facefusion.jobs.job_manager import clear_jobs, init_jobs, move_job_file, set_steps_status
+from .helper import get_test_example_file, get_test_examples_directory, get_test_jobs_directory, get_test_output_file, is_test_output_file, prepare_test_output_directory
+
+
+@pytest.fixture(scope = 'module', autouse = True)
+def before_all() -> None:
+	conditional_download(get_test_examples_directory(),
+	[
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/source.jpg',
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/target-240p.mp4'
+	])
+	subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), '-vframes', '1', get_test_example_file('target-240p.jpg') ])
+
+
+@pytest.fixture(scope = 'function', autouse = True)
+def before_each() -> None:
+	clear_jobs(get_test_jobs_directory())
+	init_jobs(get_test_jobs_directory())
+	prepare_test_output_directory()
+
+
+def test_job_run() -> None:
+	commands = [ sys.executable, 'facefusion.py', 'job-run', 'test-job-run', '--jobs-path', get_test_jobs_directory() ]
+
+	assert subprocess.run(commands).returncode == 1
+
+	commands = [ sys.executable, 'facefusion.py', 'job-create', 'test-job-run', '--jobs-path', get_test_jobs_directory() ]
+	subprocess.run(commands)
+
+	commands = [ sys.executable, 'facefusion.py', 'job-add-step', 'test-job-run', '--jobs-path', get_test_jobs_directory(), '--processors', 'face_debugger', '-t', get_test_example_file('target-240p.jpg'), '-o', get_test_output_file('test-job-run.jpg') ]
+	subprocess.run(commands)
+
+	commands = [ sys.executable, 'facefusion.py', 'job-run', 'test-job-run', '--jobs-path', get_test_jobs_directory() ]
+
+	assert subprocess.run(commands).returncode == 1
+
+	commands = [ sys.executable, 'facefusion.py', 'job-submit', 'test-job-run', '--jobs-path', get_test_jobs_directory() ]
+	subprocess.run(commands)
+
+	commands = [ sys.executable, 'facefusion.py', 'job-run', 'test-job-run', '--jobs-path', get_test_jobs_directory() ]
+
+	assert subprocess.run(commands).returncode == 0
+	assert subprocess.run(commands).returncode == 1
+	assert is_test_output_file('test-job-run.jpg') is True
+
+
+def test_job_run_all() -> None:
+	commands = [ sys.executable, 'facefusion.py', 'job-run-all', '--jobs-path', get_test_jobs_directory(), '--halt-on-error' ]
+
+	assert subprocess.run(commands).returncode == 1
+
+	commands = [ sys.executable, 'facefusion.py', 'job-create', 'test-job-run-all-1', '--jobs-path', get_test_jobs_directory() ]
+	subprocess.run(commands)
+
+	commands = [ sys.executable, 'facefusion.py', 'job-create', 'test-job-run-all-2', '--jobs-path', get_test_jobs_directory() ]
+	subprocess.run(commands)
+
+	commands = [ sys.executable, 'facefusion.py', 'job-add-step', 'test-job-run-all-1', '--jobs-path', get_test_jobs_directory(), '--processors', 'face_debugger', '-t', get_test_example_file('target-240p.jpg'), '-o', get_test_output_file('test-job-run-all-1.jpg') ]
+	subprocess.run(commands)
+
+	commands = [ sys.executable, 'facefusion.py', 'job-add-step', 'test-job-run-all-2', '--jobs-path', get_test_jobs_directory(), '--processors', 'face_debugger', '-t', get_test_example_file('target-240p.mp4'), '-o', get_test_output_file('test-job-run-all-2.mp4'), '--trim-frame-end', '1' ]
+	subprocess.run(commands)
+
+	commands = [ sys.executable, 'facefusion.py', 'job-add-step', 'test-job-run-all-2', '--jobs-path', get_test_jobs_directory(), '--processors', 'face_debugger', '-t', get_test_example_file('target-240p.mp4'), '-o', get_test_output_file('test-job-run-all-2.mp4'), '--trim-frame-start', '0', '--trim-frame-end', '1' ]
+	subprocess.run(commands)
+
+	commands = [ sys.executable, 'facefusion.py', 'job-run-all', '--jobs-path', get_test_jobs_directory(), '--halt-on-error' ]
+
+	assert subprocess.run(commands).returncode == 1
+
+	commands = [ sys.executable, 'facefusion.py', 'job-submit-all', '--jobs-path', get_test_jobs_directory(), '--halt-on-error' ]
+	subprocess.run(commands)
+
+	commands = [ sys.executable, 'facefusion.py', 'job-run-all', '--jobs-path', get_test_jobs_directory(), '--halt-on-error' ]
+
+	assert subprocess.run(commands).returncode == 0
+	assert subprocess.run(commands).returncode == 1
+	assert is_test_output_file('test-job-run-all-1.jpg') is True
+	assert is_test_output_file('test-job-run-all-2.mp4') is True
+
+
+def test_job_retry() -> None:
+	commands = [ sys.executable, 'facefusion.py', 'job-retry', 'test-job-retry', '--jobs-path', get_test_jobs_directory() ]
+
+	assert subprocess.run(commands).returncode == 1
+
+	commands = [ sys.executable, 'facefusion.py', 'job-create', 'test-job-retry', '--jobs-path', get_test_jobs_directory() ]
+	subprocess.run(commands)
+
+	commands = [ sys.executable, 'facefusion.py', 'job-add-step', 'test-job-retry', '--jobs-path', get_test_jobs_directory(), '--processors', 'face_debugger', '-t', get_test_example_file('target-240p.jpg'), '-o', get_test_output_file('test-job-retry.jpg') ]
+	subprocess.run(commands)
+
+	commands = [ sys.executable, 'facefusion.py', 'job-retry', 'test-job-retry', '--jobs-path', get_test_jobs_directory() ]
+
+	assert subprocess.run(commands).returncode == 1
+
+	set_steps_status('test-job-retry', 'failed')
+	move_job_file('test-job-retry', 'failed')
+
+	commands = [ sys.executable, 'facefusion.py', 'job-retry', 'test-job-retry', '--jobs-path', get_test_jobs_directory() ]
+
+	assert subprocess.run(commands).returncode == 0
+	assert subprocess.run(commands).returncode == 1
+	assert is_test_output_file('test-job-retry.jpg') is True
+
+
+def test_job_retry_all() -> None:
+	commands = [ sys.executable, 'facefusion.py', 'job-retry-all', '--jobs-path', get_test_jobs_directory(), '--halt-on-error' ]
+
+	assert subprocess.run(commands).returncode == 1
+
+	commands = [ sys.executable, 'facefusion.py', 'job-create', 'test-job-retry-all-1', '--jobs-path', get_test_jobs_directory() ]
+	subprocess.run(commands)
+
+	commands = [ sys.executable, 'facefusion.py', 'job-create', 'test-job-retry-all-2', '--jobs-path', get_test_jobs_directory() ]
+	subprocess.run(commands)
+
+	commands = [ sys.executable, 'facefusion.py', 'job-add-step', 'test-job-retry-all-1', '--jobs-path', get_test_jobs_directory(), '--processors', 'face_debugger', '-t', get_test_example_file('target-240p.jpg'), '-o', get_test_output_file('test-job-retry-all-1.jpg') ]
+	subprocess.run(commands)
+
+	commands = [ sys.executable, 'facefusion.py', 'job-add-step', 'test-job-retry-all-2', '--jobs-path', get_test_jobs_directory(), '--processors', 'face_debugger', '-t', get_test_example_file('target-240p.mp4'), '-o', get_test_output_file('test-job-retry-all-2.mp4'), '--trim-frame-end', '1' ]
+	subprocess.run(commands)
+
+	commands = [ sys.executable, 'facefusion.py', 'job-add-step', 'test-job-retry-all-2', '--jobs-path', get_test_jobs_directory(), '--processors', 'face_debugger', '-t', get_test_example_file('target-240p.mp4'), '-o', get_test_output_file('test-job-retry-all-2.mp4'), '--trim-frame-start', '0', '--trim-frame-end', '1' ]
+	subprocess.run(commands)
+
+	commands = [ sys.executable, 'facefusion.py', 'job-retry-all', '--jobs-path', get_test_jobs_directory(), '--halt-on-error' ]
+
+	assert subprocess.run(commands).returncode == 1
+
+	set_steps_status('test-job-retry-all-1', 'failed')
+	set_steps_status('test-job-retry-all-2', 'failed')
+	move_job_file('test-job-retry-all-1', 'failed')
+	move_job_file('test-job-retry-all-2', 'failed')
+
+	commands = [ sys.executable, 'facefusion.py', 'job-retry-all', '--jobs-path', get_test_jobs_directory(), '--halt-on-error' ]
+
+	assert subprocess.run(commands).returncode == 0
+	assert subprocess.run(commands).returncode == 1
+	assert is_test_output_file('test-job-retry-all-1.jpg') is True
+	assert is_test_output_file('test-job-retry-all-2.mp4') is True

tests/test_cli_lip_syncer.py

@@ -0,0 +1,40 @@
+import subprocess
+import sys
+
+import pytest
+
+from facefusion.download import conditional_download
+from facefusion.jobs.job_manager import clear_jobs, init_jobs
+from .helper import get_test_example_file, get_test_examples_directory, get_test_jobs_directory, get_test_output_file, is_test_output_file, prepare_test_output_directory
+
+
+@pytest.fixture(scope = 'module', autouse = True)
+def before_all() -> None:
+	conditional_download(get_test_examples_directory(),
+	[
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/source.jpg',
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/source.mp3',
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/target-240p.mp4'
+	])
+	subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), '-vframes', '1', get_test_example_file('target-240p.jpg') ])
+
+
+@pytest.fixture(scope = 'function', autouse = True)
+def before_each() -> None:
+	clear_jobs(get_test_jobs_directory())
+	init_jobs(get_test_jobs_directory())
+	prepare_test_output_directory()
+
+
+def test_sync_lip_to_image() -> None:
+	commands = [ sys.executable, 'facefusion.py', 'headless-run', '--jobs-path', get_test_jobs_directory(), '--processors', 'lip_syncer', '-s', get_test_example_file('source.mp3'), '-t', get_test_example_file('target-240p.jpg'), '-o', get_test_output_file('test_sync_lip_to_image.jpg') ]
+
+	assert subprocess.run(commands).returncode == 0
+	assert is_test_output_file('test_sync_lip_to_image.jpg') is True
+
+
+def test_sync_lip_to_video() -> None:
+	commands = [ sys.executable, 'facefusion.py', 'headless-run', '--jobs-path', get_test_jobs_directory(), '--processors', 'lip_syncer', '-s', get_test_example_file('source.mp3'), '-t', get_test_example_file('target-240p.mp4'), '-o', get_test_output_file('test_sync_lip_to_video.mp4'), '--trim-frame-end', '1' ]
+
+	assert subprocess.run(commands).returncode == 0
+	assert is_test_output_file('test_sync_lip_to_video.mp4') is True

tests/test_common_helper.py

@@ -0,0 +1,27 @@
+from facefusion.common_helper import calc_float_step, calc_int_step, create_float_metavar, create_float_range, create_int_metavar, create_int_range
+
+
+def test_create_int_metavar() -> None:
+	assert create_int_metavar([ 1, 2, 3, 4, 5 ]) == '[1..5:1]'
+
+
+def test_create_float_metavar() -> None:
+	assert create_float_metavar([ 0.1, 0.2, 0.3, 0.4, 0.5 ]) == '[0.1..0.5:0.1]'
+
+
+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 ]
+
+
+def test_calc_int_step() -> None:
+	assert calc_int_step([ 0, 1 ]) == 1
+
+
+def test_calc_float_step() -> None:
+	assert calc_float_step([ 0.1, 0.2 ]) == 0.1

tests/test_config.py

@@ -0,0 +1,85 @@
+from configparser import ConfigParser
+
+import pytest
+
+from facefusion import config
+
+
+@pytest.fixture(scope = 'module', autouse = True)
+def before_all() -> None:
+	config.CONFIG_PARSER = ConfigParser()
+	config.CONFIG_PARSER.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': ''
+		}
+	})
+
+
+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

tests/test_curl_builder.py

@@ -0,0 +1,14 @@
+from shutil import which
+
+from facefusion import metadata
+from facefusion.curl_builder import chain, head, run
+
+
+def test_run() -> None:
+	user_agent = metadata.get('name') + '/' + metadata.get('version')
+
+	assert run([]) == [ which('curl'), '--user-agent', user_agent, '--insecure', '--location', '--silent' ]
+
+
+def test_chain() -> None:
+	assert chain(head(metadata.get('url'))) == [ '-I', metadata.get('url') ]

tests/test_date_helper.py

@@ -0,0 +1,15 @@
+from datetime import datetime, timedelta
+
+from facefusion.date_helper import describe_time_ago
+
+
+def get_time_ago(days : int, hours : int, minutes : int) -> datetime:
+	previous_time = datetime.now() - timedelta(days = days, hours = hours, minutes = minutes)
+	return previous_time.astimezone()
+
+
+def test_describe_time_ago() -> None:
+	assert describe_time_ago(get_time_ago(0, 0, 0)) == 'just now'
+	assert describe_time_ago(get_time_ago(0, 0, 10)) == '10 minutes ago'
+	assert describe_time_ago(get_time_ago(0, 5, 10)) == '5 hours and 10 minutes ago'
+	assert describe_time_ago(get_time_ago(1, 5, 10)) == '1 days, 5 hours and 10 minutes ago'

tests/test_download.py

@@ -0,0 +1,18 @@
+from facefusion.download import get_static_download_size, ping_static_url, resolve_download_url_by_provider
+
+
+def test_get_static_download_size() -> None:
+	assert get_static_download_size('https://github.com/facefusion/facefusion-assets/releases/download/models-3.0.0/fairface.onnx') == 85170772
+	assert get_static_download_size('https://huggingface.co/facefusion/models-3.0.0/resolve/main/fairface.onnx') == 85170772
+	assert get_static_download_size('invalid') == 0
+
+
+def test_static_ping_url() -> None:
+	assert ping_static_url('https://github.com') is True
+	assert ping_static_url('https://huggingface.co') is True
+	assert ping_static_url('invalid') is False
+
+
+def test_resolve_download_url_by_provider() -> None:
+	assert resolve_download_url_by_provider('github', 'models-3.0.0', 'fairface.onnx') == 'https://github.com/facefusion/facefusion-assets/releases/download/models-3.0.0/fairface.onnx'
+	assert resolve_download_url_by_provider('huggingface', 'models-3.0.0', 'fairface.onnx') == 'https://huggingface.co/facefusion/models-3.0.0/resolve/main/fairface.onnx'

tests/test_execution.py

@@ -0,0 +1,24 @@
+from facefusion.execution import create_inference_session_providers, get_available_execution_providers, has_execution_provider
+
+
+def test_has_execution_provider() -> None:
+	assert has_execution_provider('cpu') is True
+	assert has_execution_provider('openvino') is False
+
+
+def test_get_available_execution_providers() -> None:
+	assert 'cpu' in get_available_execution_providers()
+
+
+def test_create_inference_session_providers() -> None:
+	inference_session_providers =\
+	[
+		('CUDAExecutionProvider',
+		{
+			'device_id': '1',
+			'cudnn_conv_algo_search': 'EXHAUSTIVE'
+		}),
+		'CPUExecutionProvider'
+	]
+
+	assert create_inference_session_providers('1', [ 'cpu', 'cuda' ]) == inference_session_providers

tests/test_face_analyser.py

@@ -0,0 +1,113 @@
+import subprocess
+
+import pytest
+
+from facefusion import face_classifier, face_detector, face_landmarker, face_recognizer, state_manager
+from facefusion.download import conditional_download
+from facefusion.face_analyser import get_many_faces, get_one_face
+from facefusion.types import Face
+from facefusion.vision import read_static_image
+from .helper import get_test_example_file, get_test_examples_directory
+
+
+@pytest.fixture(scope = 'module', autouse = True)
+def before_all() -> None:
+	conditional_download(get_test_examples_directory(),
+	[
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/source.jpg'
+	])
+	subprocess.run([ 'ffmpeg', '-i', get_test_example_file('source.jpg'), '-vf', 'crop=iw*0.8:ih*0.8', get_test_example_file('source-80crop.jpg') ])
+	subprocess.run([ 'ffmpeg', '-i', get_test_example_file('source.jpg'), '-vf', 'crop=iw*0.7:ih*0.7', get_test_example_file('source-70crop.jpg') ])
+	subprocess.run([ 'ffmpeg', '-i', get_test_example_file('source.jpg'), '-vf', 'crop=iw*0.6:ih*0.6', get_test_example_file('source-60crop.jpg') ])
+	state_manager.init_item('execution_device_id', '0')
+	state_manager.init_item('execution_providers', [ 'cpu' ])
+	state_manager.init_item('download_providers', [ 'github' ])
+	state_manager.init_item('face_detector_angles', [ 0 ])
+	state_manager.init_item('face_detector_model', 'many')
+	state_manager.init_item('face_detector_score', 0.5)
+	state_manager.init_item('face_landmarker_model', 'many')
+	state_manager.init_item('face_landmarker_score', 0.5)
+	face_classifier.pre_check()
+	face_landmarker.pre_check()
+	face_recognizer.pre_check()
+
+
+@pytest.fixture(autouse = True)
+def before_each() -> None:
+	face_classifier.clear_inference_pool()
+	face_detector.clear_inference_pool()
+	face_landmarker.clear_inference_pool()
+	face_recognizer.clear_inference_pool()
+
+
+def test_get_one_face_with_retinaface() -> None:
+	state_manager.init_item('face_detector_model', 'retinaface')
+	state_manager.init_item('face_detector_size', '320x320')
+	face_detector.pre_check()
+
+	source_paths =\
+	[
+		get_test_example_file('source.jpg'),
+		get_test_example_file('source-80crop.jpg'),
+		get_test_example_file('source-70crop.jpg'),
+		get_test_example_file('source-60crop.jpg')
+	]
+
+	for source_path in source_paths:
+		source_frame = read_static_image(source_path)
+		many_faces = get_many_faces([ source_frame ])
+		face = get_one_face(many_faces)
+
+		assert isinstance(face, Face)
+
+
+def test_get_one_face_with_scrfd() -> None:
+	state_manager.init_item('face_detector_model', 'scrfd')
+	state_manager.init_item('face_detector_size', '640x640')
+	face_detector.pre_check()
+
+	source_paths =\
+	[
+		get_test_example_file('source.jpg'),
+		get_test_example_file('source-80crop.jpg'),
+		get_test_example_file('source-70crop.jpg'),
+		get_test_example_file('source-60crop.jpg')
+	]
+
+	for source_path in source_paths:
+		source_frame = read_static_image(source_path)
+		many_faces = get_many_faces([ source_frame ])
+		face = get_one_face(many_faces)
+
+		assert isinstance(face, Face)
+
+
+def test_get_one_face_with_yoloface() -> None:
+	state_manager.init_item('face_detector_model', 'yoloface')
+	state_manager.init_item('face_detector_size', '640x640')
+	face_detector.pre_check()
+
+	source_paths =\
+	[
+		get_test_example_file('source.jpg'),
+		get_test_example_file('source-80crop.jpg'),
+		get_test_example_file('source-70crop.jpg'),
+		get_test_example_file('source-60crop.jpg')
+	]
+
+	for source_path in source_paths:
+		source_frame = read_static_image(source_path)
+		many_faces = get_many_faces([ source_frame ])
+		face = get_one_face(many_faces)
+
+		assert isinstance(face, Face)
+
+
+def test_get_many_faces() -> None:
+	source_path = get_test_example_file('source.jpg')
+	source_frame = read_static_image(source_path)
+	many_faces = get_many_faces([ source_frame, source_frame, source_frame ])
+
+	assert isinstance(many_faces[0], Face)
+	assert isinstance(many_faces[1], Face)
+	assert isinstance(many_faces[2], Face)

tests/test_ffmpeg.py

@@ -0,0 +1,119 @@
+import subprocess
+import tempfile
+
+import pytest
+
+from facefusion import process_manager, state_manager
+from facefusion.download import conditional_download
+from facefusion.ffmpeg import concat_video, extract_frames, get_available_encoder_set, read_audio_buffer, replace_audio, restore_audio
+from facefusion.filesystem import copy_file
+from facefusion.temp_helper import clear_temp_directory, create_temp_directory, get_temp_file_path, resolve_temp_frame_paths
+from .helper import get_test_example_file, get_test_examples_directory, get_test_output_file, prepare_test_output_directory
+
+
+@pytest.fixture(scope = 'module', autouse = True)
+def before_all() -> None:
+	process_manager.start()
+	conditional_download(get_test_examples_directory(),
+	[
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/source.jpg',
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/source.mp3',
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/target-240p.mp4'
+	])
+	subprocess.run([ 'ffmpeg', '-i', get_test_example_file('source.mp3'), get_test_example_file('source.wav') ])
+	subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), '-vf', 'fps=25', get_test_example_file('target-240p-25fps.mp4') ])
+	subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), '-vf', 'fps=30', get_test_example_file('target-240p-30fps.mp4') ])
+	subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), '-vf', 'fps=60', get_test_example_file('target-240p-60fps.mp4') ])
+	subprocess.run([ 'ffmpeg', '-i', get_test_example_file('source.mp3'), '-i', get_test_example_file('target-240p.mp4'), '-ar', '16000', get_test_example_file('target-240p-16khz.mp4') ])
+	subprocess.run([ 'ffmpeg', '-i', get_test_example_file('source.mp3'), '-i', get_test_example_file('target-240p.mp4'), '-ar', '48000', get_test_example_file('target-240p-48khz.mp4') ])
+	state_manager.init_item('temp_path', tempfile.gettempdir())
+	state_manager.init_item('temp_frame_format', 'png')
+	state_manager.init_item('output_audio_encoder', 'aac')
+	state_manager.init_item('output_audio_quality', 80)
+	state_manager.init_item('output_audio_volume', 100)
+
+
+@pytest.fixture(scope = 'function', autouse = True)
+def before_each() -> None:
+	prepare_test_output_directory()
+
+
+@pytest.mark.skip()
+def test_get_available_encoder_set() -> None:
+	available_encoder_set = get_available_encoder_set()
+
+	assert 'aac' in available_encoder_set.get('audio')
+	assert 'libx264' in available_encoder_set.get('video')
+
+
+def test_extract_frames() -> None:
+	extract_set =\
+	[
+		(get_test_example_file('target-240p-25fps.mp4'), 0, 270, 324),
+		(get_test_example_file('target-240p-25fps.mp4'), 224, 270, 55),
+		(get_test_example_file('target-240p-25fps.mp4'), 124, 224, 120),
+		(get_test_example_file('target-240p-25fps.mp4'), 0, 100, 120),
+		(get_test_example_file('target-240p-30fps.mp4'), 0, 324, 324),
+		(get_test_example_file('target-240p-30fps.mp4'), 224, 324, 100),
+		(get_test_example_file('target-240p-30fps.mp4'), 124, 224, 100),
+		(get_test_example_file('target-240p-30fps.mp4'), 0, 100, 100),
+		(get_test_example_file('target-240p-60fps.mp4'), 0, 648, 324),
+		(get_test_example_file('target-240p-60fps.mp4'), 224, 648, 212),
+		(get_test_example_file('target-240p-60fps.mp4'), 124, 224, 50),
+		(get_test_example_file('target-240p-60fps.mp4'), 0, 100, 50)
+	]
+
+	for target_path, trim_frame_start, trim_frame_end, frame_total in extract_set:
+		create_temp_directory(target_path)
+
+		assert extract_frames(target_path, '452x240', 30.0, trim_frame_start, trim_frame_end) is True
+		assert len(resolve_temp_frame_paths(target_path)) == frame_total
+
+		clear_temp_directory(target_path)
+
+
+def test_concat_video() -> None:
+	output_path = get_test_output_file('test-concat-video.mp4')
+	temp_output_paths =\
+	[
+		get_test_example_file('target-240p.mp4'),
+		get_test_example_file('target-240p.mp4')
+	]
+
+	assert concat_video(output_path, temp_output_paths) is True
+
+
+def test_read_audio_buffer() -> None:
+	assert isinstance(read_audio_buffer(get_test_example_file('source.mp3'), 1, 16, 1), bytes)
+	assert isinstance(read_audio_buffer(get_test_example_file('source.wav'), 1, 16, 1), bytes)
+	assert read_audio_buffer(get_test_example_file('invalid.mp3'), 1, 16, 1) is None
+
+
+def test_restore_audio() -> None:
+	target_paths =\
+	[
+		get_test_example_file('target-240p-16khz.mp4'),
+		get_test_example_file('target-240p-48khz.mp4')
+	]
+	output_path = get_test_output_file('test-restore-audio.mp4')
+
+	for target_path in target_paths:
+		create_temp_directory(target_path)
+		copy_file(target_path, get_temp_file_path(target_path))
+
+		assert restore_audio(target_path, output_path, 0, 270) is True
+
+		clear_temp_directory(target_path)
+
+
+def test_replace_audio() -> None:
+	target_path = get_test_example_file('target-240p.mp4')
+	output_path = get_test_output_file('test-replace-audio.mp4')
+
+	create_temp_directory(target_path)
+	copy_file(target_path, get_temp_file_path(target_path))
+
+	assert replace_audio(target_path, get_test_example_file('source.mp3'), output_path) is True
+	assert replace_audio(target_path, get_test_example_file('source.wav'), output_path) is True
+
+	clear_temp_directory(target_path)

tests/test_ffmpeg_builder.py

@@ -0,0 +1,83 @@
+from shutil import which
+
+from facefusion import ffmpeg_builder
+from facefusion.ffmpeg_builder import chain, run, select_frame_range, set_audio_quality, set_audio_sample_size, set_stream_mode, set_video_quality
+
+
+def test_run() -> None:
+	assert run([]) == [ which('ffmpeg'), '-loglevel', 'error' ]
+
+
+def test_chain() -> None:
+	assert chain(ffmpeg_builder.set_progress()) == [ '-progress' ]
+
+
+def test_set_stream_mode() -> None:
+	assert set_stream_mode('udp') == [ '-f', 'mpegts' ]
+	assert set_stream_mode('v4l2') == [ '-f', 'v4l2' ]
+
+
+def test_select_frame_range() -> None:
+	assert select_frame_range(0, None, 30) == [ '-vf', 'trim=start_frame=0,fps=30' ]
+	assert select_frame_range(None, 100, 30) == [ '-vf', 'trim=end_frame=100,fps=30' ]
+	assert select_frame_range(0, 100, 30) == [ '-vf', 'trim=start_frame=0:end_frame=100,fps=30' ]
+	assert select_frame_range(None, None, 30) == [ '-vf', 'fps=30' ]
+
+
+def test_set_audio_sample_size() -> None:
+	assert set_audio_sample_size(16) == [ '-f', 's16le' ]
+	assert set_audio_sample_size(32) == [ '-f', 's32le' ]
+
+
+def test_set_audio_quality() -> None:
+	assert set_audio_quality('aac', 0) == [ '-q:a', '0.1' ]
+	assert set_audio_quality('aac', 50) == [ '-q:a', '1.0' ]
+	assert set_audio_quality('aac', 100) == [ '-q:a', '2.0' ]
+	assert set_audio_quality('libmp3lame', 0) == [ '-q:a', '9' ]
+	assert set_audio_quality('libmp3lame', 50) == [ '-q:a', '4' ]
+	assert set_audio_quality('libmp3lame', 100) == [ '-q:a', '0' ]
+	assert set_audio_quality('libopus', 0) == [ '-b:a', '64k' ]
+	assert set_audio_quality('libopus', 50) == [ '-b:a', '192k' ]
+	assert set_audio_quality('libopus', 100) == [ '-b:a', '320k' ]
+	assert set_audio_quality('libvorbis', 0) == [ '-q:a', '-1.0' ]
+	assert set_audio_quality('libvorbis', 50) == [ '-q:a', '4.5' ]
+	assert set_audio_quality('libvorbis', 100) == [ '-q:a', '10.0' ]
+	assert set_audio_quality('flac', 0) == []
+	assert set_audio_quality('flac', 50) == []
+	assert set_audio_quality('flac', 100) == []
+
+
+def test_set_video_quality() -> None:
+	assert set_video_quality('libx264', 0) == [ '-crf', '51' ]
+	assert set_video_quality('libx264', 50) == [ '-crf', '26' ]
+	assert set_video_quality('libx264', 100) == [ '-crf', '0' ]
+	assert set_video_quality('libx265', 0) == [ '-crf', '51' ]
+	assert set_video_quality('libx265', 50) == [ '-crf', '26' ]
+	assert set_video_quality('libx265', 100) == [ '-crf', '0' ]
+	assert set_video_quality('libvpx-vp9', 0) == [ '-crf', '63' ]
+	assert set_video_quality('libvpx-vp9', 50) == [ '-crf', '32' ]
+	assert set_video_quality('libvpx-vp9', 100) == [ '-crf', '0' ]
+	assert set_video_quality('h264_nvenc', 0) == [ '-cq' , '51' ]
+	assert set_video_quality('h264_nvenc', 50) == [ '-cq' , '26' ]
+	assert set_video_quality('h264_nvenc', 100) == [ '-cq' , '0' ]
+	assert set_video_quality('hevc_nvenc', 0) == [ '-cq' , '51' ]
+	assert set_video_quality('hevc_nvenc', 50) == [ '-cq' , '26' ]
+	assert set_video_quality('hevc_nvenc', 100) == [ '-cq' , '0' ]
+	assert set_video_quality('h264_amf', 0) == [ '-qp_i', '51', '-qp_p', '51', '-qp_b', '51' ]
+	assert set_video_quality('h264_amf', 50) == [ '-qp_i', '26', '-qp_p', '26', '-qp_b', '26' ]
+	assert set_video_quality('h264_amf', 100) == [ '-qp_i', '0', '-qp_p', '0', '-qp_b', '0' ]
+	assert set_video_quality('hevc_amf', 0) == [ '-qp_i', '51', '-qp_p', '51', '-qp_b', '51' ]
+	assert set_video_quality('hevc_amf', 50) == [ '-qp_i', '26', '-qp_p', '26', '-qp_b', '26' ]
+	assert set_video_quality('hevc_amf', 100) == [ '-qp_i', '0', '-qp_p', '0', '-qp_b', '0' ]
+	assert set_video_quality('h264_qsv', 0) == [ '-qp', '51' ]
+	assert set_video_quality('h264_qsv', 50) == [ '-qp', '26' ]
+	assert set_video_quality('h264_qsv', 100) == [ '-qp', '0' ]
+	assert set_video_quality('hevc_qsv', 0) == [ '-qp', '51' ]
+	assert set_video_quality('hevc_qsv', 50) == [ '-qp', '26' ]
+	assert set_video_quality('hevc_qsv', 100) == [ '-qp', '0' ]
+	assert set_video_quality('h264_videotoolbox', 0) == [ '-b:v', '1024k' ]
+	assert set_video_quality('h264_videotoolbox', 50) == [ '-b:v', '25768k' ]
+	assert set_video_quality('h264_videotoolbox', 100) == [ '-b:v', '50512k' ]
+	assert set_video_quality('hevc_videotoolbox', 0) == [ '-b:v', '1024k' ]
+	assert set_video_quality('hevc_videotoolbox', 50) == [ '-b:v', '25768k' ]
+	assert set_video_quality('hevc_videotoolbox', 100) == [ '-b:v', '50512k' ]

tests/test_filesystem.py

@@ -0,0 +1,135 @@
+import os.path
+
+import pytest
+
+from facefusion.download import conditional_download
+from facefusion.filesystem import create_directory, filter_audio_paths, filter_image_paths, get_file_extension, get_file_format, get_file_size, has_audio, has_image, has_video, in_directory, is_audio, is_directory, is_file, is_image, is_video, remove_directory, resolve_file_paths, same_file_extension
+from .helper import get_test_example_file, get_test_examples_directory, get_test_outputs_directory
+
+
+@pytest.fixture(scope = 'module', autouse = True)
+def before_all() -> None:
+	conditional_download(get_test_examples_directory(),
+	[
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/source.jpg',
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/source.mp3',
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/target-240p.mp4'
+	])
+
+
+def test_get_file_size() -> None:
+	assert get_file_size(get_test_example_file('source.jpg')) == 549458
+	assert get_file_size('invalid') == 0
+
+
+def test_get_file_extension() -> None:
+	assert get_file_extension('source.jpg') == '.jpg'
+	assert get_file_extension('source.mp3') == '.mp3'
+	assert get_file_extension('invalid') is None
+
+
+def test_get_file_format() -> None:
+	assert get_file_format('source.jpg') == 'jpeg'
+	assert get_file_format('source.jpeg') == 'jpeg'
+	assert get_file_format('source.mp3') == 'mp3'
+	assert get_file_format('invalid') is None
+
+
+def test_same_file_extension() -> None:
+	assert same_file_extension('source.jpg', 'source.jpg') is True
+	assert same_file_extension('source.jpg', 'source.mp3') is False
+	assert same_file_extension('invalid', 'invalid') is False
+
+
+def test_is_file() -> None:
+	assert is_file(get_test_example_file('source.jpg')) is True
+	assert is_file(get_test_examples_directory()) is False
+	assert is_file('invalid') is False
+
+
+def test_is_audio() -> None:
+	assert is_audio(get_test_example_file('source.mp3')) is True
+	assert is_audio(get_test_example_file('source.jpg')) is False
+	assert is_audio('invalid') is False
+
+
+def test_has_audio() -> None:
+	assert has_audio([ get_test_example_file('source.mp3') ]) is True
+	assert has_audio([ get_test_example_file('source.mp3'), get_test_example_file('source.jpg') ]) is True
+	assert has_audio([ get_test_example_file('source.jpg'), get_test_example_file('source.jpg') ]) is False
+	assert has_audio([ 'invalid' ]) is False
+
+
+def test_is_image() -> None:
+	assert is_image(get_test_example_file('source.jpg')) is True
+	assert is_image(get_test_example_file('target-240p.mp4')) is False
+	assert is_image('invalid') is False
+
+
+def test_has_image() -> None:
+	assert has_image([ get_test_example_file('source.jpg') ]) is True
+	assert has_image([ get_test_example_file('source.jpg'), get_test_example_file('source.mp3') ]) is True
+	assert has_image([ get_test_example_file('source.mp3'), get_test_example_file('source.mp3') ]) is False
+	assert has_image([ 'invalid' ]) is False
+
+
+def test_is_video() -> None:
+	assert is_video(get_test_example_file('target-240p.mp4')) is True
+	assert is_video(get_test_example_file('source.jpg')) is False
+	assert is_video('invalid') is False
+
+
+def test_has_video() -> None:
+	assert has_video([ get_test_example_file('target-240p.mp4') ]) is True
+	assert has_video([ get_test_example_file('target-240p.mp4'), get_test_example_file('source.mp3') ]) is True
+	assert has_video([ get_test_example_file('source.mp3'), get_test_example_file('source.mp3') ]) is False
+	assert has_video([ 'invalid' ]) is False
+
+
+def test_filter_audio_paths() -> None:
+	assert filter_audio_paths([ get_test_example_file('source.jpg'), get_test_example_file('source.mp3') ]) == [ get_test_example_file('source.mp3') ]
+	assert filter_audio_paths([ get_test_example_file('source.jpg'), get_test_example_file('source.jpg') ]) == []
+	assert filter_audio_paths([ 'invalid' ]) == []
+
+
+def test_filter_image_paths() -> None:
+	assert filter_image_paths([ get_test_example_file('source.jpg'), get_test_example_file('source.mp3') ]) == [ get_test_example_file('source.jpg') ]
+	assert filter_image_paths([ get_test_example_file('source.mp3'), get_test_example_file('source.mp3') ]) == []
+	assert filter_audio_paths([ 'invalid' ]) == []
+
+
+def test_resolve_file_paths() -> None:
+	file_paths = resolve_file_paths(get_test_examples_directory())
+
+	for file_path in file_paths:
+		assert file_path == get_test_example_file(file_path)
+
+	assert resolve_file_paths('invalid') == []
+
+
+def test_create_directory() -> None:
+	create_directory_path = os.path.join(get_test_outputs_directory(), 'create_directory')
+
+	assert create_directory(create_directory_path) is True
+	assert create_directory(get_test_example_file('source.jpg')) is False
+
+
+def test_remove_directory() -> None:
+	remove_directory_path = os.path.join(get_test_outputs_directory(), 'remove_directory')
+	create_directory(remove_directory_path)
+
+	assert remove_directory(remove_directory_path) is True
+	assert remove_directory(get_test_example_file('source.jpg')) is False
+	assert remove_directory('invalid') is False
+
+
+def test_is_directory() -> None:
+	assert is_directory(get_test_examples_directory()) is True
+	assert is_directory(get_test_example_file('source.jpg')) is False
+	assert is_directory('invalid') is False
+
+
+def test_in_directory() -> None:
+	assert in_directory(get_test_example_file('source.jpg')) is True
+	assert in_directory('source.jpg') is False
+	assert in_directory('invalid') is False

tests/test_inference_manager.py

@@ -0,0 +1,32 @@
+from unittest.mock import patch
+
+import pytest
+from onnxruntime import InferenceSession
+
+from facefusion import content_analyser, state_manager
+from facefusion.inference_manager import INFERENCE_POOL_SET, get_inference_pool
+
+
+@pytest.fixture(scope = 'module', autouse = True)
+def before_all() -> None:
+	state_manager.init_item('execution_device_id', '0')
+	state_manager.init_item('execution_providers', [ 'cpu' ])
+	state_manager.init_item('download_providers', [ 'github' ])
+	content_analyser.pre_check()
+
+
+def test_get_inference_pool() -> None:
+	model_names = [ 'yolo_nsfw' ]
+	model_source_set = content_analyser.get_model_options().get('sources')
+
+	with patch('facefusion.inference_manager.detect_app_context', return_value = 'cli'):
+		get_inference_pool('facefusion.content_analyser', model_names, model_source_set)
+
+		assert isinstance(INFERENCE_POOL_SET.get('cli').get('facefusion.content_analyser.yolo_nsfw.0.cpu').get('content_analyser'), InferenceSession)
+
+	with patch('facefusion.inference_manager.detect_app_context', return_value = 'ui'):
+		get_inference_pool('facefusion.content_analyser', model_names, model_source_set)
+
+		assert isinstance(INFERENCE_POOL_SET.get('ui').get('facefusion.content_analyser.yolo_nsfw.0.cpu').get('content_analyser'), InferenceSession)
+
+	assert INFERENCE_POOL_SET.get('cli').get('facefusion.content_analyser.yolo_nsfw.0.cpu').get('content_analyser') == INFERENCE_POOL_SET.get('ui').get('facefusion.content_analyser.yolo_nsfw.0.cpu').get('content_analyser')

tests/test_job_helper.py

@@ -0,0 +1,8 @@
+import os
+
+from facefusion.jobs.job_helper import get_step_output_path
+
+
+def test_get_step_output_path() -> None:
+	assert get_step_output_path('test-job', 0, 'test.mp4') == 'test-test-job-0.mp4'
+	assert get_step_output_path('test-job', 0, 'test/test.mp4') == os.path.join('test', 'test-test-job-0.mp4')

tests/test_job_list.py

@@ -0,0 +1,24 @@
+from time import sleep
+
+import pytest
+
+from facefusion.jobs.job_list import compose_job_list
+from facefusion.jobs.job_manager import clear_jobs, create_job, init_jobs
+from .helper import get_test_jobs_directory
+
+
+@pytest.fixture(scope = 'function', autouse = True)
+def before_each() -> None:
+	clear_jobs(get_test_jobs_directory())
+	init_jobs(get_test_jobs_directory())
+
+
+def test_compose_job_list() -> None:
+	create_job('job-test-compose-job-list-1')
+	sleep(0.5)
+	create_job('job-test-compose-job-list-2')
+	job_headers, job_contents = compose_job_list('drafted')
+
+	assert job_headers == [ 'job id', 'steps', 'date created', 'date updated', 'job status' ]
+	assert job_contents[0] == [ 'job-test-compose-job-list-1', 0, 'just now', None, 'drafted' ]
+	assert job_contents[1] == [ 'job-test-compose-job-list-2', 0, 'just now', None, 'drafted' ]

tests/test_job_manager.py

@@ -0,0 +1,376 @@
+from time import sleep
+
+import pytest
+
+from facefusion.jobs.job_helper import get_step_output_path
+from facefusion.jobs.job_manager import add_step, clear_jobs, count_step_total, create_job, delete_job, delete_jobs, find_job_ids, get_steps, init_jobs, insert_step, move_job_file, remix_step, remove_step, set_step_status, set_steps_status, submit_job, submit_jobs
+from .helper import get_test_jobs_directory
+
+
+@pytest.fixture(scope = 'function', autouse = True)
+def before_each() -> None:
+	clear_jobs(get_test_jobs_directory())
+	init_jobs(get_test_jobs_directory())
+
+
+def test_create_job() -> None:
+	args_1 =\
+	{
+		'source_path': 'source-1.jpg',
+		'target_path': 'target-1.jpg',
+		'output_path': 'output-1.jpg'
+	}
+
+	assert create_job('job-test-create-job') is True
+	assert create_job('job-test-create-job') is False
+
+	add_step('job-test-submit-job', args_1)
+	submit_job('job-test-create-job')
+
+	assert create_job('job-test-create-job') is False
+
+
+def test_submit_job() -> None:
+	args_1 =\
+	{
+		'source_path': 'source-1.jpg',
+		'target_path': 'target-1.jpg',
+		'output_path': 'output-1.jpg'
+	}
+
+	assert submit_job('job-invalid') is False
+
+	create_job('job-test-submit-job')
+
+	assert submit_job('job-test-submit-job') is False
+
+	add_step('job-test-submit-job', args_1)
+
+	assert submit_job('job-test-submit-job') is True
+	assert submit_job('job-test-submit-job') is False
+
+
+def test_submit_jobs() -> None:
+	args_1 =\
+	{
+		'source_path': 'source-1.jpg',
+		'target_path': 'target-1.jpg',
+		'output_path': 'output-1.jpg'
+	}
+	args_2 =\
+	{
+		'source_path': 'source-2.jpg',
+		'target_path': 'target-2.jpg',
+		'output_path': 'output-2.jpg'
+	}
+	halt_on_error = True
+
+	assert submit_jobs(halt_on_error) is False
+
+	create_job('job-test-submit-jobs-1')
+	create_job('job-test-submit-jobs-2')
+
+	assert submit_jobs(halt_on_error) is False
+
+	add_step('job-test-submit-jobs-1', args_1)
+	add_step('job-test-submit-jobs-2', args_2)
+
+	assert submit_jobs(halt_on_error) is True
+	assert submit_jobs(halt_on_error) is False
+
+
+def test_delete_job() -> None:
+	assert delete_job('job-invalid') is False
+
+	create_job('job-test-delete-job')
+
+	assert delete_job('job-test-delete-job') is True
+	assert delete_job('job-test-delete-job') is False
+
+
+def test_delete_jobs() -> None:
+	halt_on_error = True
+
+	assert delete_jobs(halt_on_error) is False
+
+	create_job('job-test-delete-jobs-1')
+	create_job('job-test-delete-jobs-2')
+
+	assert delete_jobs(halt_on_error) is True
+
+
+@pytest.mark.skip()
+def test_find_jobs() -> None:
+	pass
+
+
+def test_find_job_ids() -> None:
+	create_job('job-test-find-job-ids-1')
+	sleep(0.5)
+	create_job('job-test-find-job-ids-2')
+	sleep(0.5)
+	create_job('job-test-find-job-ids-3')
+
+	assert find_job_ids('drafted') == [ 'job-test-find-job-ids-1', 'job-test-find-job-ids-2', 'job-test-find-job-ids-3' ]
+	assert find_job_ids('queued') == []
+	assert find_job_ids('completed') == []
+	assert find_job_ids('failed') == []
+
+	move_job_file('job-test-find-job-ids-1', 'queued')
+	move_job_file('job-test-find-job-ids-2', 'queued')
+	move_job_file('job-test-find-job-ids-3', 'queued')
+
+	assert find_job_ids('drafted') == []
+	assert find_job_ids('queued') == [ 'job-test-find-job-ids-1', 'job-test-find-job-ids-2', 'job-test-find-job-ids-3' ]
+	assert find_job_ids('completed') == []
+	assert find_job_ids('failed') == []
+
+	move_job_file('job-test-find-job-ids-1', 'completed')
+
+	assert find_job_ids('drafted') == []
+	assert find_job_ids('queued') == [ 'job-test-find-job-ids-2', 'job-test-find-job-ids-3' ]
+	assert find_job_ids('completed') == [ 'job-test-find-job-ids-1' ]
+	assert find_job_ids('failed') == []
+
+	move_job_file('job-test-find-job-ids-2', 'failed')
+
+	assert find_job_ids('drafted') == []
+	assert find_job_ids('queued') == [ 'job-test-find-job-ids-3' ]
+	assert find_job_ids('completed') == [ 'job-test-find-job-ids-1' ]
+	assert find_job_ids('failed') == [ 'job-test-find-job-ids-2' ]
+
+	move_job_file('job-test-find-job-ids-3', 'completed')
+
+	assert find_job_ids('drafted') == []
+	assert find_job_ids('queued') == []
+	assert find_job_ids('completed') == [ 'job-test-find-job-ids-1', 'job-test-find-job-ids-3' ]
+	assert find_job_ids('failed') == [ 'job-test-find-job-ids-2' ]
+
+
+def test_add_step() -> None:
+	args_1 =\
+	{
+		'source_path': 'source-1.jpg',
+		'target_path': 'target-1.jpg',
+		'output_path': 'output-1.jpg'
+	}
+	args_2 =\
+	{
+		'source_path': 'source-2.jpg',
+		'target_path': 'target-2.jpg',
+		'output_path': 'output-2.jpg'
+	}
+
+	assert add_step('job-invalid', args_1) is False
+
+	create_job('job-test-add-step')
+
+	assert add_step('job-test-add-step', args_1) is True
+	assert add_step('job-test-add-step', args_2) is True
+
+	steps = get_steps('job-test-add-step')
+
+	assert steps[0].get('args') == args_1
+	assert steps[1].get('args') == args_2
+	assert count_step_total('job-test-add-step') == 2
+
+
+def test_remix_step() -> None:
+	args_1 =\
+	{
+		'source_path': 'source-1.jpg',
+		'target_path': 'target-1.jpg',
+		'output_path': 'output-1.jpg'
+	}
+	args_2 =\
+	{
+		'source_path': 'source-2.jpg',
+		'target_path': 'target-2.jpg',
+		'output_path': 'output-2.jpg'
+	}
+
+	assert remix_step('job-invalid', 0, args_1) is False
+
+	create_job('job-test-remix-step')
+	add_step('job-test-remix-step', args_1)
+	add_step('job-test-remix-step', args_2)
+
+	assert remix_step('job-test-remix-step', 99, args_1) is False
+	assert remix_step('job-test-remix-step', 0, args_2) is True
+	assert remix_step('job-test-remix-step', -1, args_2) is True
+
+	steps = get_steps('job-test-remix-step')
+
+	assert steps[0].get('args') == args_1
+	assert steps[1].get('args') == args_2
+	assert steps[2].get('args').get('source_path') == args_2.get('source_path')
+	assert steps[2].get('args').get('target_path') == get_step_output_path('job-test-remix-step', 0, args_1.get('output_path'))
+	assert steps[2].get('args').get('output_path') == args_2.get('output_path')
+	assert steps[3].get('args').get('source_path') == args_2.get('source_path')
+	assert steps[3].get('args').get('target_path') == get_step_output_path('job-test-remix-step', 2, args_2.get('output_path'))
+	assert steps[3].get('args').get('output_path') == args_2.get('output_path')
+	assert count_step_total('job-test-remix-step') == 4
+
+
+def test_insert_step() -> None:
+	args_1 =\
+	{
+		'source_path': 'source-1.jpg',
+		'target_path': 'target-1.jpg',
+		'output_path': 'output-1.jpg'
+	}
+	args_2 =\
+	{
+		'source_path': 'source-2.jpg',
+		'target_path': 'target-2.jpg',
+		'output_path': 'output-2.jpg'
+	}
+	args_3 =\
+	{
+		'source_path': 'source-3.jpg',
+		'target_path': 'target-3.jpg',
+		'output_path': 'output-3.jpg'
+	}
+
+	assert insert_step('job-invalid', 0, args_1) is False
+
+	create_job('job-test-insert-step')
+	add_step('job-test-insert-step', args_1)
+	add_step('job-test-insert-step', args_1)
+
+	assert insert_step('job-test-insert-step', 99, args_1) is False
+	assert insert_step('job-test-insert-step', 0, args_2) is True
+	assert insert_step('job-test-insert-step', -1, args_3) is True
+
+	steps = get_steps('job-test-insert-step')
+
+	assert steps[0].get('args') == args_2
+	assert steps[1].get('args') == args_1
+	assert steps[2].get('args') == args_3
+	assert steps[3].get('args') == args_1
+	assert count_step_total('job-test-insert-step') == 4
+
+
+def test_remove_step() -> None:
+	args_1 =\
+	{
+		'source_path': 'source-1.jpg',
+		'target_path': 'target-1.jpg',
+		'output_path': 'output-1.jpg'
+	}
+	args_2 =\
+	{
+		'source_path': 'source-2.jpg',
+		'target_path': 'target-2.jpg',
+		'output_path': 'output-2.jpg'
+	}
+	args_3 =\
+	{
+		'source_path': 'source-3.jpg',
+		'target_path': 'target-3.jpg',
+		'output_path': 'output-3.jpg'
+	}
+
+	assert remove_step('job-invalid', 0) is False
+
+	create_job('job-test-remove-step')
+	add_step('job-test-remove-step', args_1)
+	add_step('job-test-remove-step', args_2)
+	add_step('job-test-remove-step', args_1)
+	add_step('job-test-remove-step', args_3)
+
+	assert remove_step('job-test-remove-step', 99) is False
+	assert remove_step('job-test-remove-step', 0) is True
+	assert remove_step('job-test-remove-step', -1) is True
+
+	steps = get_steps('job-test-remove-step')
+
+	assert steps[0].get('args') == args_2
+	assert steps[1].get('args') == args_1
+	assert count_step_total('job-test-remove-step') == 2
+
+
+def test_get_steps() -> None:
+	args_1 =\
+	{
+		'source_path': 'source-1.jpg',
+		'target_path': 'target-1.jpg',
+		'output_path': 'output-1.jpg'
+	}
+	args_2 =\
+	{
+		'source_path': 'source-2.jpg',
+		'target_path': 'target-2.jpg',
+		'output_path': 'output-2.jpg'
+	}
+
+	assert get_steps('job-invalid') == []
+
+	create_job('job-test-get-steps')
+	add_step('job-test-get-steps', args_1)
+	add_step('job-test-get-steps', args_2)
+	steps = get_steps('job-test-get-steps')
+
+	assert steps[0].get('args') == args_1
+	assert steps[1].get('args') == args_2
+	assert count_step_total('job-test-get-steps') == 2
+
+
+def test_set_step_status() -> None:
+	args_1 =\
+	{
+		'source_path': 'source-1.jpg',
+		'target_path': 'target-1.jpg',
+		'output_path': 'output-1.jpg'
+	}
+	args_2 =\
+	{
+		'source_path': 'source-2.jpg',
+		'target_path': 'target-2.jpg',
+		'output_path': 'output-2.jpg'
+	}
+
+	assert set_step_status('job-invalid', 0, 'completed') is False
+
+	create_job('job-test-set-step-status')
+	add_step('job-test-set-step-status', args_1)
+	add_step('job-test-set-step-status', args_2)
+
+	assert set_step_status('job-test-set-step-status', 99, 'completed') is False
+	assert set_step_status('job-test-set-step-status', 0, 'completed') is True
+	assert set_step_status('job-test-set-step-status', 1, 'failed') is True
+
+	steps = get_steps('job-test-set-step-status')
+
+	assert steps[0].get('status') == 'completed'
+	assert steps[1].get('status') == 'failed'
+	assert count_step_total('job-test-set-step-status') == 2
+
+
+def test_set_steps_status() -> None:
+	args_1 =\
+	{
+		'source_path': 'source-1.jpg',
+		'target_path': 'target-1.jpg',
+		'output_path': 'output-1.jpg'
+	}
+	args_2 =\
+	{
+		'source_path': 'source-2.jpg',
+		'target_path': 'target-2.jpg',
+		'output_path': 'output-2.jpg'
+	}
+
+	assert set_steps_status('job-invalid', 'queued') is False
+
+	create_job('job-test-set-steps-status')
+	add_step('job-test-set-steps-status', args_1)
+	add_step('job-test-set-steps-status', args_2)
+
+	assert set_steps_status('job-test-set-steps-status', 'queued') is True
+
+	steps = get_steps('job-test-set-steps-status')
+
+	assert steps[0].get('status') == 'queued'
+	assert steps[1].get('status') == 'queued'
+	assert count_step_total('job-test-set-steps-status') == 2

tests/test_job_runner.py

@@ -0,0 +1,227 @@
+import subprocess
+
+import pytest
+
+from facefusion.download import conditional_download
+from facefusion.filesystem import copy_file
+from facefusion.jobs.job_manager import add_step, clear_jobs, create_job, init_jobs, submit_job, submit_jobs
+from facefusion.jobs.job_runner import collect_output_set, finalize_steps, run_job, run_jobs, run_steps
+from facefusion.types import Args
+from .helper import get_test_example_file, get_test_examples_directory, get_test_jobs_directory, get_test_output_file, is_test_output_file, prepare_test_output_directory
+
+
+@pytest.fixture(scope = 'module', autouse = True)
+def before_all() -> None:
+	conditional_download(get_test_examples_directory(),
+	[
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/source.jpg',
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/target-240p.mp4'
+	])
+	subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), '-vframes', '1', get_test_example_file('target-240p.jpg') ])
+
+
+@pytest.fixture(scope = 'function', autouse = True)
+def before_each() -> None:
+	clear_jobs(get_test_jobs_directory())
+	init_jobs(get_test_jobs_directory())
+	prepare_test_output_directory()
+
+
+def process_step(job_id : str, step_index : int, step_args : Args) -> bool:
+	return copy_file(step_args.get('target_path'), step_args.get('output_path'))
+
+
+def test_run_job() -> None:
+	args_1 =\
+	{
+		'source_path': get_test_example_file('source.jpg'),
+		'target_path': get_test_example_file('target-240p.mp4'),
+		'output_path': get_test_output_file('output-1.mp4')
+	}
+	args_2 =\
+	{
+		'source_path': get_test_example_file('source.jpg'),
+		'target_path': get_test_example_file('target-240p.mp4'),
+		'output_path': get_test_output_file('output-2.mp4')
+	}
+	args_3 =\
+	{
+		'source_path': get_test_example_file('source.jpg'),
+		'target_path': get_test_example_file('target-240p.jpg'),
+		'output_path': get_test_output_file('output-1.jpg')
+	}
+
+	assert run_job('job-invalid', process_step) is False
+
+	create_job('job-test-run-job')
+	add_step('job-test-run-job', args_1)
+	add_step('job-test-run-job', args_2)
+	add_step('job-test-run-job', args_2)
+	add_step('job-test-run-job', args_3)
+
+	assert run_job('job-test-run-job', process_step) is False
+
+	submit_job('job-test-run-job')
+
+	assert run_job('job-test-run-job', process_step) is True
+
+
+def test_run_jobs() -> None:
+	args_1 =\
+	{
+		'source_path': get_test_example_file('source.jpg'),
+		'target_path': get_test_example_file('target-240p.mp4'),
+		'output_path': get_test_output_file('output-1.mp4')
+	}
+	args_2 =\
+	{
+		'source_path': get_test_example_file('source.jpg'),
+		'target_path': get_test_example_file('target-240p.mp4'),
+		'output_path': get_test_output_file('output-2.mp4')
+	}
+	args_3 =\
+	{
+		'source_path': get_test_example_file('source.jpg'),
+		'target_path': get_test_example_file('target-240p.jpg'),
+		'output_path': get_test_output_file('output-1.jpg')
+	}
+	halt_on_error = True
+
+	assert run_jobs(process_step, halt_on_error) is False
+
+	create_job('job-test-run-jobs-1')
+	create_job('job-test-run-jobs-2')
+	add_step('job-test-run-jobs-1', args_1)
+	add_step('job-test-run-jobs-1', args_1)
+	add_step('job-test-run-jobs-2', args_2)
+	add_step('job-test-run-jobs-3', args_3)
+
+	assert run_jobs(process_step, halt_on_error) is False
+
+	submit_jobs(halt_on_error)
+
+	assert run_jobs(process_step, halt_on_error) is True
+
+
+@pytest.mark.skip()
+def test_retry_job() -> None:
+	pass
+
+
+@pytest.mark.skip()
+def test_retry_jobs() -> None:
+	pass
+
+
+def test_run_steps() -> None:
+	args_1 =\
+	{
+		'source_path': get_test_example_file('source.jpg'),
+		'target_path': get_test_example_file('target-240p.mp4'),
+		'output_path': get_test_output_file('output-1.mp4')
+	}
+	args_2 =\
+	{
+		'source_path': get_test_example_file('source.jpg'),
+		'target_path': get_test_example_file('target-240p.mp4'),
+		'output_path': get_test_output_file('output-2.mp4')
+	}
+	args_3 =\
+	{
+		'source_path': get_test_example_file('source.jpg'),
+		'target_path': get_test_example_file('target-240p.jpg'),
+		'output_path': get_test_output_file('output-1.jpg')
+	}
+
+	assert run_steps('job-invalid', process_step) is False
+
+	create_job('job-test-run-steps')
+	add_step('job-test-run-steps', args_1)
+	add_step('job-test-run-steps', args_1)
+	add_step('job-test-run-steps', args_2)
+	add_step('job-test-run-steps', args_3)
+
+	assert run_steps('job-test-run-steps', process_step) is True
+
+
+def test_finalize_steps() -> None:
+	args_1 =\
+	{
+		'source_path': get_test_example_file('source.jpg'),
+		'target_path': get_test_example_file('target-240p.mp4'),
+		'output_path': get_test_output_file('output-1.mp4')
+	}
+	args_2 =\
+	{
+		'source_path': get_test_example_file('source.jpg'),
+		'target_path': get_test_example_file('target-240p.mp4'),
+		'output_path': get_test_output_file('output-2.mp4')
+	}
+	args_3 =\
+	{
+		'source_path': get_test_example_file('source.jpg'),
+		'target_path': get_test_example_file('target-240p.jpg'),
+		'output_path': get_test_output_file('output-1.jpg')
+	}
+
+	create_job('job-test-finalize-steps')
+	add_step('job-test-finalize-steps', args_1)
+	add_step('job-test-finalize-steps', args_1)
+	add_step('job-test-finalize-steps', args_2)
+	add_step('job-test-finalize-steps', args_3)
+
+	copy_file(args_1.get('target_path'), get_test_output_file('output-1-job-test-finalize-steps-0.mp4'))
+	copy_file(args_1.get('target_path'), get_test_output_file('output-1-job-test-finalize-steps-1.mp4'))
+	copy_file(args_2.get('target_path'), get_test_output_file('output-2-job-test-finalize-steps-2.mp4'))
+	copy_file(args_3.get('target_path'), get_test_output_file('output-1-job-test-finalize-steps-3.jpg'))
+
+	assert finalize_steps('job-test-finalize-steps') is True
+	assert is_test_output_file('output-1.mp4') is True
+	assert is_test_output_file('output-2.mp4') is True
+	assert is_test_output_file('output-1.jpg') is True
+
+
+def test_collect_output_set() -> None:
+	args_1 =\
+	{
+		'source_path': get_test_example_file('source.jpg'),
+		'target_path': get_test_example_file('target-240p.mp4'),
+		'output_path': get_test_output_file('output-1.mp4')
+	}
+	args_2 =\
+	{
+		'source_path': get_test_example_file('source.jpg'),
+		'target_path': get_test_example_file('target-240p.mp4'),
+		'output_path': get_test_output_file('output-2.mp4')
+	}
+	args_3 =\
+	{
+		'source_path': get_test_example_file('source.jpg'),
+		'target_path': get_test_example_file('target-240p.jpg'),
+		'output_path': get_test_output_file('output-1.jpg')
+	}
+
+	create_job('job-test-collect-output-set')
+	add_step('job-test-collect-output-set', args_1)
+	add_step('job-test-collect-output-set', args_1)
+	add_step('job-test-collect-output-set', args_2)
+	add_step('job-test-collect-output-set', args_3)
+
+	output_set =\
+	{
+		get_test_output_file('output-1.mp4'):
+		[
+			get_test_output_file('output-1-job-test-collect-output-set-0.mp4'),
+			get_test_output_file('output-1-job-test-collect-output-set-1.mp4')
+		],
+		get_test_output_file('output-2.mp4'):
+		[
+			get_test_output_file('output-2-job-test-collect-output-set-2.mp4')
+		],
+		get_test_output_file('output-1.jpg'):
+		[
+			get_test_output_file('output-1-job-test-collect-output-set-3.jpg')
+		]
+	}
+
+	assert collect_output_set('job-test-collect-output-set') == output_set

tests/test_json.py

@@ -0,0 +1,19 @@
+import tempfile
+
+from facefusion.json import read_json, write_json
+
+
+def test_read_json() -> None:
+	_, json_path = tempfile.mkstemp(suffix = '.json')
+
+	assert not read_json(json_path)
+
+	write_json(json_path, {})
+
+	assert read_json(json_path) == {}
+
+
+def test_write_json() -> None:
+	_, json_path = tempfile.mkstemp(suffix = '.json')
+
+	assert write_json(json_path, {})

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,16 @@
+from facefusion.normalizer import normalize_fps, normalize_padding
+
+
+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 end, is_pending, is_processing, is_stopping, set_process_state, start, stop
+
+
+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_program_helper.py

@@ -0,0 +1,40 @@
+from argparse import ArgumentParser
+
+import pytest
+
+from facefusion.program_helper import find_argument_group, validate_actions
+
+
+def test_find_argument_group() -> None:
+	program = ArgumentParser()
+	program.add_argument_group('test-1')
+	program.add_argument_group('test-2')
+
+	assert find_argument_group(program, 'test-1')
+	assert find_argument_group(program, 'test-2')
+	assert find_argument_group(program, 'invalid') is None
+
+
+@pytest.mark.skip()
+def test_validate_args() -> None:
+	pass
+
+
+def test_validate_actions() -> None:
+	program = ArgumentParser()
+	program.add_argument('--test-1', default = 'test_1', choices = [ 'test_1', 'test_2' ])
+	program.add_argument('--test-2', default = 'test_2', choices= [ 'test_1', 'test_2' ], nargs = '+')
+
+	assert validate_actions(program) is True
+
+	args =\
+	{
+		'test_1': 'test_2',
+		'test_2': [ 'test_1', 'test_3' ]
+	}
+
+	for action in program._actions:
+		if action.dest in args:
+			action.default = args[action.dest]
+
+	assert validate_actions(program) is False

tests/test_temp_helper.py

@@ -0,0 +1,34 @@
+import os.path
+import tempfile
+
+import pytest
+
+from facefusion import state_manager
+from facefusion.download import conditional_download
+from facefusion.temp_helper import get_temp_directory_path, get_temp_file_path, get_temp_frames_pattern
+from .helper import get_test_example_file, get_test_examples_directory
+
+
+@pytest.fixture(scope = 'module', autouse = True)
+def before_all() -> None:
+	conditional_download(get_test_examples_directory(),
+	[
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/target-240p.mp4'
+	])
+	state_manager.init_item('temp_path', tempfile.gettempdir())
+	state_manager.init_item('temp_frame_format', 'png')
+
+
+def test_get_temp_file_path() -> None:
+	temp_directory = tempfile.gettempdir()
+	assert get_temp_file_path(get_test_example_file('target-240p.mp4')) == os.path.join(temp_directory, 'facefusion', 'target-240p', 'temp.mp4')
+
+
+def test_get_temp_directory_path() -> None:
+	temp_directory = tempfile.gettempdir()
+	assert get_temp_directory_path(get_test_example_file('target-240p.mp4')) == os.path.join(temp_directory, 'facefusion', 'target-240p')
+
+
+def test_get_temp_frames_pattern() -> None:
+	temp_directory = tempfile.gettempdir()
+	assert get_temp_frames_pattern(get_test_example_file('target-240p.mp4'), '%04d') == os.path.join(temp_directory, 'facefusion', 'target-240p', '%04d.png')

tests/test_vision.py

@@ -0,0 +1,179 @@
+import subprocess
+
+import pytest
+
+from facefusion.download import conditional_download
+from facefusion.vision import calc_histogram_difference, count_trim_frame_total, count_video_frame_total, create_image_resolutions, create_video_resolutions, detect_image_resolution, detect_video_duration, detect_video_fps, detect_video_resolution, match_frame_color, normalize_resolution, pack_resolution, predict_video_frame_total, read_image, read_video_frame, restrict_image_resolution, restrict_trim_frame, restrict_video_fps, restrict_video_resolution, unpack_resolution, write_image
+from .helper import get_test_example_file, get_test_examples_directory, get_test_output_file, prepare_test_output_directory
+
+
+@pytest.fixture(scope = 'module', autouse = True)
+def before_all() -> None:
+	conditional_download(get_test_examples_directory(),
+	[
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/source.jpg',
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/target-240p.mp4',
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/target-1080p.mp4'
+	])
+	subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), '-vframes', '1', get_test_example_file('target-240p.jpg') ])
+	subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), '-vframes', '1', get_test_example_file('目标-240p.webp') ])
+	subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-1080p.mp4'), '-vframes', '1', get_test_example_file('target-1080p.jpg') ])
+	subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), '-vframes', '1', '-vf', 'hue=s=0', get_test_example_file('target-240p-0sat.jpg') ])
+	subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), '-vframes', '1', '-vf', 'transpose=0', get_test_example_file('target-240p-90deg.jpg') ])
+	subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-1080p.mp4'), '-vframes', '1', '-vf', 'transpose=0', get_test_example_file('target-1080p-90deg.jpg') ])
+	subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), '-vf', 'fps=25', get_test_example_file('target-240p-25fps.mp4') ])
+	subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), '-vf', 'fps=30', get_test_example_file('target-240p-30fps.mp4') ])
+	subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), '-vf', 'fps=60', get_test_example_file('target-240p-60fps.mp4') ])
+	subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), '-vf', 'transpose=0', get_test_example_file('target-240p-90deg.mp4') ])
+	subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-1080p.mp4'), '-vf', 'transpose=0', get_test_example_file('target-1080p-90deg.mp4') ])
+
+
+@pytest.fixture(scope = 'function', autouse = True)
+def before_each() -> None:
+	prepare_test_output_directory()
+
+
+def test_read_image() -> None:
+	assert read_image(get_test_example_file('target-240p.jpg')).shape == (226, 426, 3)
+	assert read_image(get_test_example_file('目标-240p.webp')).shape == (226, 426, 3)
+	assert read_image('invalid') is None
+
+
+def test_write_image() -> None:
+	vision_frame = read_image(get_test_example_file('target-240p.jpg'))
+
+	assert write_image(get_test_output_file('target-240p.jpg'), vision_frame) is True
+	assert write_image(get_test_output_file('目标-240p.webp'), vision_frame) is True
+
+
+def test_detect_image_resolution() -> None:
+	assert detect_image_resolution(get_test_example_file('target-240p.jpg')) == (426, 226)
+	assert detect_image_resolution(get_test_example_file('target-240p-90deg.jpg')) == (226, 426)
+	assert detect_image_resolution(get_test_example_file('target-1080p.jpg')) == (2048, 1080)
+	assert detect_image_resolution(get_test_example_file('target-1080p-90deg.jpg')) == (1080, 2048)
+	assert detect_image_resolution('invalid') is None
+
+
+def test_restrict_image_resolution() -> None:
+	assert restrict_image_resolution(get_test_example_file('target-1080p.jpg'), (426, 226)) == (426, 226)
+	assert restrict_image_resolution(get_test_example_file('target-1080p.jpg'), (2048, 1080)) == (2048, 1080)
+	assert restrict_image_resolution(get_test_example_file('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_read_video_frame() -> None:
+	assert hasattr(read_video_frame(get_test_example_file('target-240p-25fps.mp4')), '__array_interface__')
+	assert read_video_frame('invalid') is None
+
+
+def test_count_video_frame_total() -> None:
+	assert count_video_frame_total(get_test_example_file('target-240p-25fps.mp4')) == 270
+	assert count_video_frame_total(get_test_example_file('target-240p-30fps.mp4')) == 324
+	assert count_video_frame_total(get_test_example_file('target-240p-60fps.mp4')) == 648
+	assert count_video_frame_total('invalid') == 0
+
+
+def test_predict_video_frame_total() -> None:
+	assert predict_video_frame_total(get_test_example_file('target-240p-25fps.mp4'), 12.5, 0, 100) == 50
+	assert predict_video_frame_total(get_test_example_file('target-240p-25fps.mp4'), 25, 0, 100) == 100
+	assert predict_video_frame_total(get_test_example_file('target-240p-25fps.mp4'), 25, 0, 200) == 200
+	assert predict_video_frame_total('invalid', 25, 0, 100) == 0
+
+
+def test_detect_video_fps() -> None:
+	assert detect_video_fps(get_test_example_file('target-240p-25fps.mp4')) == 25.0
+	assert detect_video_fps(get_test_example_file('target-240p-30fps.mp4')) == 30.0
+	assert detect_video_fps(get_test_example_file('target-240p-60fps.mp4')) == 60.0
+	assert detect_video_fps('invalid') is None
+
+
+def test_restrict_video_fps() -> None:
+	assert restrict_video_fps(get_test_example_file('target-1080p.mp4'), 20.0) == 20.0
+	assert restrict_video_fps(get_test_example_file('target-1080p.mp4'), 25.0) == 25.0
+	assert restrict_video_fps(get_test_example_file('target-1080p.mp4'), 60.0) == 25.0
+
+
+def test_detect_video_duration() -> None:
+	assert detect_video_duration(get_test_example_file('target-240p.mp4')) == 10.8
+	assert detect_video_duration('invalid') == 0
+
+
+def test_count_trim_frame_total() -> None:
+	assert count_trim_frame_total(get_test_example_file('target-240p.mp4'), 0, 200) == 200
+	assert count_trim_frame_total(get_test_example_file('target-240p.mp4'), 70, 270) == 200
+	assert count_trim_frame_total(get_test_example_file('target-240p.mp4'), -10, None) == 270
+	assert count_trim_frame_total(get_test_example_file('target-240p.mp4'), None, -10) == 0
+	assert count_trim_frame_total(get_test_example_file('target-240p.mp4'), 280, None) == 0
+	assert count_trim_frame_total(get_test_example_file('target-240p.mp4'), None, 280) == 270
+	assert count_trim_frame_total(get_test_example_file('target-240p.mp4'), None, None) == 270
+
+
+def test_restrict_trim_frame() -> None:
+	assert restrict_trim_frame(get_test_example_file('target-240p.mp4'), 0, 200) == (0, 200)
+	assert restrict_trim_frame(get_test_example_file('target-240p.mp4'), 70, 270) == (70, 270)
+	assert restrict_trim_frame(get_test_example_file('target-240p.mp4'), -10, None) == (0, 270)
+	assert restrict_trim_frame(get_test_example_file('target-240p.mp4'), None, -10) == (0, 0)
+	assert restrict_trim_frame(get_test_example_file('target-240p.mp4'), 280, None) == (270, 270)
+	assert restrict_trim_frame(get_test_example_file('target-240p.mp4'), None, 280) == (0, 270)
+	assert restrict_trim_frame(get_test_example_file('target-240p.mp4'), None, None) == (0, 270)
+
+
+def test_detect_video_resolution() -> None:
+	assert detect_video_resolution(get_test_example_file('target-240p.mp4')) == (426, 226)
+	assert detect_video_resolution(get_test_example_file('target-240p-90deg.mp4')) == (226, 426)
+	assert detect_video_resolution(get_test_example_file('target-1080p.mp4')) == (2048, 1080)
+	assert detect_video_resolution(get_test_example_file('target-1080p-90deg.mp4')) == (1080, 2048)
+	assert detect_video_resolution('invalid') is None
+
+
+def test_restrict_video_resolution() -> None:
+	assert restrict_video_resolution(get_test_example_file('target-1080p.mp4'), (426, 226)) == (426, 226)
+	assert restrict_video_resolution(get_test_example_file('target-1080p.mp4'), (2048, 1080)) == (2048, 1080)
+	assert restrict_video_resolution(get_test_example_file('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)
+
+
+def test_calc_histogram_difference() -> None:
+	source_vision_frame = read_image(get_test_example_file('target-240p.jpg'))
+	target_vision_frame = read_image(get_test_example_file('target-240p-0sat.jpg'))
+
+	assert calc_histogram_difference(source_vision_frame, source_vision_frame) == 1.0
+	assert calc_histogram_difference(source_vision_frame, target_vision_frame) < 0.5
+
+
+def test_match_frame_color() -> None:
+	source_vision_frame = read_image(get_test_example_file('target-240p.jpg'))
+	target_vision_frame = read_image(get_test_example_file('target-240p-0sat.jpg'))
+	output_vision_frame = match_frame_color(source_vision_frame, target_vision_frame)
+
+	assert calc_histogram_difference(source_vision_frame, output_vision_frame) > 0.5

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_paths')
+	assert wording.get('invalid') is None