Skip to content

Commit de68dba

Browse files
committed
Fix tests
1 parent b9fc9e1 commit de68dba

File tree

1 file changed

+3
-16
lines changed

1 file changed

+3
-16
lines changed

python_gpt_po/tests/test_multi_provider.py

Lines changed: 3 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -312,14 +312,8 @@ def test_get_deepseek_models(mock_provider_clients: ProviderClients):
312312
assert "deepseek-coder" in models
313313

314314

315-
@patch('python_gpt_po.services.translation_service.requests.post')
316-
def test_translate_bulk_openai(mock_post, translation_service_openai: TranslationService):
315+
def test_translate_bulk_openai(translation_service_openai: TranslationService):
317316
"""Test bulk translation with OpenAI."""
318-
# Setup mock response
319-
mock_response = MagicMock()
320-
mock_response.json.return_value = OPENAI_TRANSLATION_RESPONSE
321-
mock_post.return_value = mock_response
322-
323317
# Call function
324318
translation_service_openai.config.provider_clients.openai_client.chat.completions.create.return_value = (
325319
MagicMock(
@@ -340,14 +334,8 @@ def test_translate_bulk_openai(mock_post, translation_service_openai: Translatio
340334
assert translations == ["Bonjour", "Monde", "Bienvenue dans notre application", "Au revoir"]
341335

342336

343-
@patch('python_gpt_po.services.translation_service.requests.post')
344-
def test_translate_bulk_azure_openai(mock_post, translation_service_azure_openai: TranslationService):
337+
def test_translate_bulk_azure_openai(translation_service_azure_openai: TranslationService):
345338
"""Test bulk translation with OpenAI."""
346-
# Setup mock response
347-
mock_response = MagicMock()
348-
mock_response.json.return_value = AZURE_OPENAI_TRANSLATION_RESPONSE
349-
mock_post.return_value = mock_response
350-
351339
# Call function
352340
azure_client = translation_service_azure_openai.config.provider_clients.azure_openai_client
353341
azure_response = azure_client.chat.completions.create
@@ -368,8 +356,7 @@ def test_translate_bulk_azure_openai(mock_post, translation_service_azure_openai
368356
assert translations == ["Bonjour", "Monde", "Bienvenue dans notre application", "Au revoir"]
369357

370358

371-
@patch('python_gpt_po.services.translation_service.requests.post')
372-
def test_translate_bulk_anthropic(mock_post, translation_service_anthropic: TranslationService):
359+
def test_translate_bulk_anthropic(translation_service_anthropic: TranslationService):
373360
"""Test bulk translation with Anthropic."""
374361
# Setup mock client response
375362
translation_service_anthropic.config.provider_clients.anthropic_client.messages.create.return_value = MagicMock(

0 commit comments

Comments
 (0)