Apply black formatting to test files

This commit is contained in:
Russell Ballestrini 2025-11-10 17:05:08 -05:00
parent 21acc90f2d
commit 4550407d7b
2 changed files with 6 additions and 2 deletions

View file

@ -163,7 +163,9 @@ script_result = {
mock_metadata["ai_board"][90:92] = ["Destroyer"] * 2 # Destroyer
with patch.object(
activity, "execute_processing_script", return_value={"metadata": mock_metadata}
activity,
"execute_processing_script",
return_value={"metadata": mock_metadata},
) as mock_exec:
metadata = {}
result = activity.execute_processing_script(metadata, setup_script)

View file

@ -311,7 +311,9 @@ class TestGuardedAI(unittest.TestCase):
# Verify client was created and stored with actual model ID
mock_get_client.assert_called_with("http://test.com", "test-key")
self.assertIn("test-model-id", guarded_ai.MODEL_CLIENT_MAP)
self.assertEqual(guarded_ai.MODEL_CLIENT_MAP["test-model-id"][0], mock_client)
self.assertEqual(
guarded_ai.MODEL_CLIENT_MAP["test-model-id"][0], mock_client
)
self.assertEqual(
guarded_ai.MODEL_CLIENT_MAP["test-model-id"][1], "http://test.com"
)