chore: import upstream snapshot with attribution

python/tests/integration/text_to_image/test_text_to_image.py

@@ -0,0 +1,42 @@
+# Copyright (c) Microsoft. All rights reserved.
+
+
+import pytest
+
+from semantic_kernel.connectors.ai.text_to_image_client_base import TextToImageClientBase
+from tests.integration.text_to_image.text_to_image_test_base import TextToImageTestBase
+
+pytestmark = pytest.mark.parametrize(
+    "service_id, prompt",
+    [
+        pytest.param(
+            "openai",
+            "A cute tuxedo cat driving a race car.",
+            id="openai",
+        ),
+        pytest.param(
+            "azure_openai",
+            "A cute tuxedo cat driving a race car.",
+            id="azure_openai",
+            marks=[
+                pytest.mark.xfail(
+                    reason="Temporary failure due to Internal Server Error (500) from Azure OpenAI.",
+                ),
+            ],
+        ),
+    ],
+)
+
+
+class TestTextToImage(TextToImageTestBase):
+    """Test text-to-image services."""
+
+    async def test_text_to_image(
+        self,
+        services: dict[str, TextToImageClientBase],
+        service_id: str,
+        prompt: str,
+    ):
+        service = services[service_id]
+        image_url = await service.generate_image(prompt, 1024, 1024)
+        assert image_url

python/tests/integration/text_to_image/text_to_image_test_base.py

@@ -0,0 +1,20 @@
+# Copyright (c) Microsoft. All rights reserved.
+
+import pytest
+from azure.identity import AzureCliCredential
+
+from semantic_kernel.connectors.ai.open_ai.services.azure_text_to_image import AzureTextToImage
+from semantic_kernel.connectors.ai.open_ai.services.open_ai_text_to_image import OpenAITextToImage
+from semantic_kernel.connectors.ai.text_to_image_client_base import TextToImageClientBase
+
+
+class TextToImageTestBase:
+    """Base class for testing text-to-image services."""
+
+    @pytest.fixture(scope="module")
+    def services(self) -> dict[str, TextToImageClientBase]:
+        """Return text-to-image services."""
+        return {
+            "openai": OpenAITextToImage(),
+            "azure_openai": AzureTextToImage(credential=AzureCliCredential()),
+        }