From c010cdef5912a7538c04302eebb44dbe1886c08f Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 18 Mar 2025 17:26:23 -0700 Subject: [PATCH] test_dynamic_azure_params --- tests/local_testing/test_completion.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/tests/local_testing/test_completion.py b/tests/local_testing/test_completion.py index 5fe4984c17..59f5a38f08 100644 --- a/tests/local_testing/test_completion.py +++ b/tests/local_testing/test_completion.py @@ -4364,14 +4364,14 @@ async def test_dynamic_azure_params(stream, sync_mode): ## recreate mock client if sync_mode: - mock_client = MagicMock(return_value="Hello world!") + new_mock_client = MagicMock(return_value="Hello world!") else: - mock_client = AsyncMock(return_value="Hello world!") + new_mock_client = AsyncMock(return_value="Hello world!") ## CHECK IF NEW CLIENT IS USED (PARAM CHANGE) with patch.object( - client.chat.completions.with_raw_response, "create", new=mock_client - ) as mock_client: + client.chat.completions.with_raw_response, "create", new=new_mock_client + ) as new_mock_client: try: if sync_mode: _ = completion( @@ -4393,7 +4393,7 @@ async def test_dynamic_azure_params(stream, sync_mode): pass try: - mock_client.assert_not_called() + new_mock_client.assert_called() except Exception as e: raise e