From 419da8a9b52caa53c611ee1656675be505f94105 Mon Sep 17 00:00:00 2001 From: Maverick-F35 Date: Wed, 26 Jul 2023 10:01:11 +0530 Subject: [PATCH] test cases refactored --- .../tools/image_generation/test_stable_diffusion_image_gen.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/tests/unit_tests/tools/image_generation/test_stable_diffusion_image_gen.py b/tests/unit_tests/tools/image_generation/test_stable_diffusion_image_gen.py index 537223f81..3f34cec6f 100644 --- a/tests/unit_tests/tools/image_generation/test_stable_diffusion_image_gen.py +++ b/tests/unit_tests/tools/image_generation/test_stable_diffusion_image_gen.py @@ -57,14 +57,10 @@ def test_execute(stable_diffusion_tool): tool.resource_manager = Mock() tool.agent_id = 123 # Use a dummy agent_id for testing purposes tool.toolkit_config.get_tool_config = lambda key: 'fake_api_key' if key == 'STABILITY_API_KEY' else 'engine_id_1' - # Change 'engine_id_1' to the appropriate value for the "ENGINE_ID" key in your real code - prompt = 'Test prompt' image_names = ['img1.png', 'img2.png'] expected_result = 'Images downloaded and saved successfully' - result = tool._execute(prompt, image_names) - assert result.startswith(expected_result) tool.resource_manager.write_binary_file.assert_called()