engineering/opencompletion.com
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Apply black formatting to test files

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

4
tests/unit/test_guarded_ai_functions.py
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"
)