From ed26fc5ce59fe190ccb4b07d46d0f6fafd2fefa9 Mon Sep 17 00:00:00 2001 From: yujonglee Date: Mon, 19 Oct 2026 18:26:09 -0700 Subject: [PATCH] test(images): stop writing image edit responses to disk (#47463) Co-authored-by: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- tests/unit/images/test_image_edit_utils.py | 18 ------------------ 1 file changed, 18 deletions(-) diff --git a/tests/unit/images/test_image_edit_utils.py b/tests/unit/images/test_image_edit_utils.py index 109c4b89101..2f240b0fed4 100644 --- a/tests/unit/images/test_image_edit_utils.py +++ b/tests/unit/images/test_image_edit_utils.py @@ -536,15 +536,6 @@ async def test_azure_image_edit_litellm_sdk(): # Validate the response meets expected schema ImageResponse.model_validate(result) - if isinstance(result, ImageResponse) and result.data: - image_base64 = result.data[0].b64_json - if image_base64: - image_bytes = base64.b64decode(image_base64) - - # Save the image to a file - with open("test_image_edit.png", "wb") as f: - f.write(image_bytes) - @pytest.mark.asyncio async def test_openai_image_edit_cost_tracking(): """Test OpenAI image edit cost tracking with custom logger""" @@ -607,15 +598,6 @@ async def test_openai_image_edit_cost_tracking(): # Validate the response meets expected schema ImageResponse.model_validate(result) - if isinstance(result, ImageResponse) and result.data: - image_base64 = result.data[0].b64_json - if image_base64: - image_bytes = base64.b64decode(image_base64) - - # Save the image to a file - with open("test_image_edit.png", "wb") as f: - f.write(image_bytes) - await asyncio.sleep(5) print( "standard logging payload",