From eed5e374ae29ea3b6abcb2ec28e42236570b6ab5 Mon Sep 17 00:00:00 2001
From: Merwane Hamadi <merwanehamadi@gmail.com>
Date: Mon, 29 May 2023 19:01:10 -0700
Subject: [PATCH] add vcr to dalle

---
 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 5c04921b5a52..0156c9e5bdc9 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,