Apply black formatting to test files
This commit is contained in:
parent
21acc90f2d
commit
4550407d7b
2 changed files with 6 additions and 2 deletions
|
|
@ -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)
|
||||
|
|
|
|||
|
|
@ -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"
|
||||
)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue