Files
facefusion/tests/assert_helper.py
T
Harisreedhar ab24cd3f2e Implement basic webrtc stream (#1054)
* implement basic webrtc_stream

* add aiortc to requirements.txt

* update aiortc version

* rename variables with rtc_ prefix

* changes

* changes

* change helper to assert_helper and stream_helper

* rename variables with rtc_ prefix

* add error handling

* return whole connection

* remove monkey patch and some cleaning

* cleanup

* tiny adjustments

* tiny adjustments

* proper typing and naming for rtc offer set

* - remove async from on_video_track method
- rename source -> target
- add audio

* audio always before video

---------

Co-authored-by: henryruhs <info@henryruhs.com>
2026-03-17 14:02:40 +01:00

49 lines
1.4 KiB
Python

import os
import tempfile
from facefusion.filesystem import are_images, create_directory, is_directory, is_file, remove_directory, resolve_file_paths
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_path(file_path))
def is_test_output_sequence(directory_path : str) -> bool:
return are_images(resolve_file_paths(directory_path))
def get_test_output_path(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)