From 22429c6e98d71b67306f3cb81c2dee4577708e6c Mon Sep 17 00:00:00 2001
From: merwanehamadi <merwanehamadi@gmail.com>
Date: Tue, 30 May 2023 06:00:37 -0700
Subject: [PATCH] add vcr to dalle (#4474)

---
 tests/{ => integration}/test_image_gen.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)
 rename tests/{ => integration}/test_image_gen.py (97%)

diff --git a/tests/test_image_gen.py b/tests/integration/test_image_gen.py
similarity index 97%
rename from tests/test_image_gen.py
rename to tests/integration/test_image_gen.py
index 5c04921b..0156c9e5 100644
--- a/tests/test_image_gen.py
+++ b/tests/integration/test_image_gen.py
@@ -16,11 +16,9 @@ def image_size(request):
     return request.param
 
 
-@pytest.mark.xfail(
-    reason="The image is too big to be put in a cassette for a CI pipeline. We're looking into a solution."
-)
 @requires_api_key("OPENAI_API_KEY")
-def test_dalle(config, workspace, image_size):
+@pytest.mark.vcr
+def test_dalle(config, workspace, image_size, patched_api_requestor):
     """Test DALL-E image generation."""
     generate_and_validate(
         config,