Reformatted tests

Author: tisnik

tests/e2e/features/steps/llm_query_response.py

@@ -6,7 +6,6 @@
 from behave.runner import Context
 from tests.e2e.utils.utils import replace_placeholders
 
-
 DEFAULT_LLM_TIMEOUT = 60
 
 

tests/integration/endpoints/test_rlsapi_v1_integration.py

@@ -32,7 +32,6 @@
 from tests.unit.utils.auth_helpers import mock_authorization_resolvers
 from utils.suid import check_suid
 
-
 # ==========================================
 # Shared Fixtures
 # ==========================================

tests/unit/app/endpoints/test_a2a.py

@@ -39,7 +39,6 @@
 from configuration import AppConfig
 from models.config import Action
 
-
 # User ID must be proper UUID
 MOCK_AUTH = (
     "00000001-0001-0001-0001-000000000001",

tests/unit/app/endpoints/test_streaming_query.py

@@ -43,7 +43,6 @@
 from tests.unit.utils.auth_helpers import mock_authorization_resolvers
 from utils.token_counter import TokenCounter
 
-
 # Note: content_delta module doesn't exist in llama-stack-client 0.3.x
 # These are mock classes for backward compatibility with Agent API tests
 # pylint: disable=too-few-public-methods,redefined-builtin

tests/unit/authentication/test_api_key_token.py

@@ -2,7 +2,6 @@
 
 """Unit tests for functions defined in authentication/api_key_token.py"""
 
-
 import pytest
 from fastapi import HTTPException, Request
 from pydantic import SecretStr

tests/unit/cache/test_postgres_cache.py

@@ -18,7 +18,6 @@
 from cache.cache_error import CacheError
 from cache.postgres_cache import PostgresCache
 
-
 USER_ID_1 = suid.get_suid()
 USER_ID_2 = suid.get_suid()
 CONVERSATION_ID_1 = suid.get_suid()

tests/unit/models/rlsapi/test_requests.py

@@ -15,7 +15,6 @@
     RlsapiV1Terminal,
 )
 
-
 # ---------------------------------------------------------------------------
 # Fixtures
 # ---------------------------------------------------------------------------

tests/unit/models/rlsapi/test_responses.py

@@ -12,7 +12,6 @@
 )
 from models.responses import AbstractSuccessfulResponse
 
-
 # ---------------------------------------------------------------------------
 # Fixtures
 # ---------------------------------------------------------------------------

tests/unit/test_configuration.py

@@ -251,8 +251,7 @@ def test_load_proper_configuration(tmpdir: Path) -> None:
     """Test loading proper configuration from YAML file."""
     cfg_filename = tmpdir / "config.yaml"
     with open(cfg_filename, "w", encoding="utf-8") as fout:
-        fout.write(
-            """
+        fout.write("""
 name: foo bar baz
 service:
   host: localhost
@@ -268,8 +267,7 @@ def test_load_proper_configuration(tmpdir: Path) -> None:
 user_data_collection:
   feedback_enabled: false
 mcp_servers: []
-            """
-        )
+            """)
 
     cfg = AppConfig()
     cfg.load_configuration(str(cfg_filename))
@@ -283,8 +281,7 @@ def test_load_configuration_with_mcp_servers(tmpdir: Path) -> None:
     """Test loading configuration from YAML file with MCP servers."""
     cfg_filename = tmpdir / "config.yaml"
     with open(cfg_filename, "w", encoding="utf-8") as fout:
-        fout.write(
-            """
+        fout.write("""
 name: test service
 service:
   host: localhost
@@ -305,8 +302,7 @@ def test_load_configuration_with_mcp_servers(tmpdir: Path) -> None:
   - name: git-server
     provider_id: custom-git-provider
     url: https://git.example.com/mcp
-            """
-        )
+            """)
 
     cfg = AppConfig()
     cfg.load_configuration(str(cfg_filename))
@@ -447,8 +443,7 @@ def test_load_configuration_with_customization_system_prompt_path(tmpdir: Path)
 
     cfg_filename = tmpdir / "config.yaml"
     with open(cfg_filename, "w", encoding="utf-8") as fout:
-        fout.write(
-            f"""
+        fout.write(f"""
 name: test service
 service:
   host: localhost
@@ -472,8 +467,7 @@ def test_load_configuration_with_customization_system_prompt_path(tmpdir: Path)
 customization:
   disable_query_system_prompt: true
   system_prompt_path: {system_prompt_filename}
-            """
-        )
+            """)
 
     cfg = AppConfig()
     cfg.load_configuration(str(cfg_filename))
@@ -487,8 +481,7 @@ def test_load_configuration_with_customization_system_prompt(tmpdir: Path) -> No
     """Test loading configuration from YAML file with system_prompt in the customization."""
     cfg_filename = tmpdir / "config.yaml"
     with open(cfg_filename, "w", encoding="utf-8") as fout:
-        fout.write(
-            """
+        fout.write("""
 name: test service
 service:
   host: localhost
@@ -512,8 +505,7 @@ def test_load_configuration_with_customization_system_prompt(tmpdir: Path) -> No
 customization:
   system_prompt: |-
     this is system prompt in the customization section
-            """
-        )
+            """)
 
     cfg = AppConfig()
     cfg.load_configuration(str(cfg_filename))
@@ -532,8 +524,7 @@ def test_configuration_with_profile_customization(tmpdir: Path) -> None:
     expected_prompts = expected_profile.get_prompts()
     cfg_filename = tmpdir / "config.yaml"
     with open(cfg_filename, "w", encoding="utf-8") as fout:
-        fout.write(
-            """
+        fout.write("""
 name: test service
 service:
   host: localhost
@@ -550,8 +541,7 @@ def test_configuration_with_profile_customization(tmpdir: Path) -> None:
   feedback_enabled: false
 customization:
   profile_path: tests/profiles/test/profile.py
-            """
-        )
+            """)
 
     cfg = AppConfig()
     cfg.load_configuration(str(cfg_filename))
@@ -575,8 +565,7 @@ def test_configuration_with_all_customizations(tmpdir: Path) -> None:
 
     cfg_filename = tmpdir / "config.yaml"
     with open(cfg_filename, "w", encoding="utf-8") as fout:
-        fout.write(
-            f"""
+        fout.write(f"""
 name: test service
 service:
   host: localhost
@@ -595,8 +584,7 @@ def test_configuration_with_all_customizations(tmpdir: Path) -> None:
   profile_path: tests/profiles/test/profile.py
   system_prompt: custom prompt
   system_prompt_path: {system_prompt_filename}
-            """
-        )
+            """)
 
     cfg = AppConfig()
     cfg.load_configuration(str(cfg_filename))
@@ -614,8 +602,7 @@ def test_configuration_with_sqlite_conversation_cache(tmpdir: Path) -> None:
     """Test loading configuration from YAML file with conversation cache configuration."""
     cfg_filename = tmpdir / "config.yaml"
     with open(cfg_filename, "w", encoding="utf-8") as fout:
-        fout.write(
-            """
+        fout.write("""
 name: test service
 service:
   host: localhost
@@ -634,8 +621,7 @@ def test_configuration_with_sqlite_conversation_cache(tmpdir: Path) -> None:
   type: "sqlite"
   sqlite:
     db_path: ":memory:"
-            """
-        )
+            """)
 
     cfg = AppConfig()
     cfg.load_configuration(str(cfg_filename))
@@ -653,8 +639,7 @@ def test_configuration_with_in_memory_conversation_cache(tmpdir: Path) -> None:
     """Test loading configuration from YAML file with conversation cache configuration."""
     cfg_filename = tmpdir / "config.yaml"
     with open(cfg_filename, "w", encoding="utf-8") as fout:
-        fout.write(
-            """
+        fout.write("""
 name: test service
 service:
   host: localhost
@@ -673,8 +658,7 @@ def test_configuration_with_in_memory_conversation_cache(tmpdir: Path) -> None:
   type: "memory"
   memory:
     max_entries: 42
-            """
-        )
+            """)
 
     cfg = AppConfig()
     cfg.load_configuration(str(cfg_filename))
@@ -692,8 +676,7 @@ def test_configuration_with_quota_handlers_no_storage(tmpdir: Path) -> None:
     """Test loading configuration from YAML file with quota handlers configuration."""
     cfg_filename = tmpdir / "config.yaml"
     with open(cfg_filename, "w", encoding="utf-8") as fout:
-        fout.write(
-            """
+        fout.write("""
 name: test service
 service:
   host: localhost
@@ -723,8 +706,7 @@ def test_configuration_with_quota_handlers_no_storage(tmpdir: Path) -> None:
   scheduler:
     # scheduler ticks in seconds
     period: 1
-            """
-        )
+            """)
 
     cfg = AppConfig()
     cfg.load_configuration(str(cfg_filename))
@@ -746,8 +728,7 @@ def test_configuration_with_quota_handlers(tmpdir: Path) -> None:
     """Test loading configuration from YAML file with quota handlers configuration."""
     cfg_filename = tmpdir / "config.yaml"
     with open(cfg_filename, "w", encoding="utf-8") as fout:
-        fout.write(
-            """
+        fout.write("""
 name: test service
 service:
   host: localhost
@@ -779,8 +760,7 @@ def test_configuration_with_quota_handlers(tmpdir: Path) -> None:
   scheduler:
     # scheduler ticks in seconds
     period: 1
-            """
-        )
+            """)
 
     cfg = AppConfig()
     cfg.load_configuration(str(cfg_filename))
@@ -813,8 +793,7 @@ def test_load_configuration_with_azure_entra_id(tmpdir: Path) -> None:
     """Return Azure Entra ID configuration when provided in configuration."""
     cfg_filename = tmpdir / "config.yaml"
     with open(cfg_filename, "w", encoding="utf-8") as fout:
-        fout.write(
-            """
+        fout.write("""
 name: test service
 service:
   host: localhost
@@ -833,8 +812,7 @@ def test_load_configuration_with_azure_entra_id(tmpdir: Path) -> None:
   tenant_id: tenant
   client_id: client
   client_secret: secret
-            """
-        )
+            """)
 
     cfg = AppConfig()
     cfg.load_configuration(str(cfg_filename))
@@ -850,8 +828,7 @@ def test_load_configuration_with_incomplete_azure_entra_id_raises(tmpdir: Path)
     """Raise error if Azure Entra ID block is incomplete in configuration."""
     cfg_filename = tmpdir / "config.yaml"
     with open(cfg_filename, "w", encoding="utf-8") as fout:
-        fout.write(
-            """
+        fout.write("""
 name: test service
 service:
   host: localhost
@@ -869,8 +846,7 @@ def test_load_configuration_with_incomplete_azure_entra_id_raises(tmpdir: Path)
 azure_entra_id:
   tenant_id: tenant
   client_id: client
-            """
-        )
+            """)
 
     cfg = AppConfig()
     with pytest.raises(ValidationError):

tests/unit/test_llama_stack_configuration.py

@@ -19,7 +19,6 @@
     InferenceConfiguration,
 )
 
-
 # =============================================================================
 # Test construct_vector_dbs_section
 # =============================================================================