From 8dba29ef18e6adb225fbab503cc35b8c256bb0bc Mon Sep 17 00:00:00 2001 From: harisreedhar Date: Tue, 17 Feb 2026 15:32:29 +0530 Subject: [PATCH] remove unwanted assertions --- tests/test_api_capabilities.py | 27 +++++++++++++-------------- 1 file changed, 13 insertions(+), 14 deletions(-) diff --git a/tests/test_api_capabilities.py b/tests/test_api_capabilities.py index 6106d7d4..7b1b3237 100644 --- a/tests/test_api_capabilities.py +++ b/tests/test_api_capabilities.py @@ -11,12 +11,19 @@ from facefusion.apis.core import create_api @pytest.fixture(scope = 'module') def test_client() -> Iterator[TestClient]: program = ArgumentParser() - source_paths_action = program.add_argument('--source-paths', nargs = '+') - target_path_action = program.add_argument('--target-path') - output_format_action = program.add_argument('--output-format', default = 'mp4', choices = [ 'mp4', 'mkv', 'webm' ]) - args_store.register_argument(source_paths_action, scopes = [ 'api' ]) - args_store.register_argument(target_path_action, scopes = [ 'api' ]) - args_store.register_argument(output_format_action, scopes = [ 'api' ]) + args_store.register_argument( + program.add_argument( + '--source-paths', + nargs = '+'), + scopes = [ 'api' ] + ) + args_store.register_argument( + program.add_argument( + '--output-format', + default = 'mp4', + choices = [ 'mp4', 'mkv', 'webm' ]), + scopes = [ 'api' ] + ) with TestClient(create_api()) as test_client: yield test_client @@ -34,18 +41,10 @@ def test_get_capabilities(test_client : TestClient) -> None: assert capabilities_response.status_code == 200 assert 'mp3' in capabilities_body.get('formats').get('audio') - assert 'wav' in capabilities_body.get('formats').get('audio') - assert 'flac' in capabilities_body.get('formats').get('audio') assert 'jpeg' in capabilities_body.get('formats').get('image') - assert 'png' in capabilities_body.get('formats').get('image') - assert 'webp' in capabilities_body.get('formats').get('image') assert 'mp4' in capabilities_body.get('formats').get('video') - assert 'mkv' in capabilities_body.get('formats').get('video') - assert 'webm' in capabilities_body.get('formats').get('video') assert capabilities_body.get('arguments').get('source_paths').get('default') is None - assert capabilities_body.get('arguments').get('target_path').get('default') is None - assert capabilities_body.get('arguments').get('output_format').get('default') == 'mp4' assert capabilities_body.get('arguments').get('output_format').get('choices') == [ 'mp4', 'mkv', 'webm' ]