From 74504e9e4a0e7518d40f733030e8a1069c4538b0 Mon Sep 17 00:00:00 2001 From: fanhaoxuee <1470600398@qq.com> Date: Wed, 29 Oct 2025 01:46:01 +0800 Subject: [PATCH] 1029-9 --- .../unit/agent/action/tool/test_readimage_tool.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/tests/test_agentuniverse/unit/agent/action/tool/test_readimage_tool.py b/tests/test_agentuniverse/unit/agent/action/tool/test_readimage_tool.py index 16d4d912..b7b996e8 100644 --- a/tests/test_agentuniverse/unit/agent/action/tool/test_readimage_tool.py +++ b/tests/test_agentuniverse/unit/agent/action/tool/test_readimage_tool.py @@ -36,11 +36,15 @@ class TestReadImageTool(unittest.TestCase): test_text = "Sample text for testing." test_filename = "test_extracted_text.txt" save_text_to_file(test_text, test_filename) - self.assertTrue(os.path.exists(test_filename)) - with open(test_filename, 'r', encoding='utf-8') as f: - content = f.read() - self.assertEqual(content, test_text) - os.remove(test_filename) + try: + self.assertTrue(os.path.exists(test_filename)) + with open(test_filename, 'r', encoding='utf-8') as f: + content = f.read() + self.assertEqual(content, test_text) + finally: + if os.path.exists(test_filename): + os.remove(test_filename) + def test_extract_text_from_image_without_east(self): # Save the test OCR image as a temporary file