Corrected integration tests

jodendaal · Nov 10, 2023 · f035c9d · f035c9d
1 parent 085238e
commit f035c9d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/OpenAI.Net.Integration.Tests/ImageService_Variation.cs b/src/OpenAI.Net.Integration.Tests/ImageService_Variation.cs
@@ -32,7 +32,7 @@ public async Task Variation_Base64ToFileContent(bool isSuccess, HttpStatusCode s
                 o.Size = "256x256";
                 o.ResponseFormat = ImageResponseFormat.Base64;
                 o.Style = ImageStyleOptions.Vivid;
-                o.Quality = ImageQualityOptions.HighDefinition;
+                o.Quality = ImageQualityOptions.Standard;
                 o.Model = ImageModelOptions.Dalle2;
             });
 
@@ -47,7 +47,7 @@ public async Task Variation_Base64ToFileContent(bool isSuccess, HttpStatusCode s
             Assert.That(response.StatusCode, Is.EqualTo(statusCode));
             Assert.That(response.Result?.Data?.Count() == 1, Is.EqualTo(isSuccess), "Data is not mapped correctly");
             Assert.That(response.Result?.Data?[0].Url?.Contains("https://"), isSuccess ? Is.EqualTo(isSuccess) : Is.EqualTo(null), "Choice text not set");
-            Assert.That(response.ErrorResponse?.Error?.Message?.Contains("is not one of ['256x256', '512x512', '1024x1024']"), isSuccess ? Is.EqualTo(null) : Is.EqualTo(true), "Error message not returned");
+            Assert.That(response.ErrorResponse?.Error?.Message, isSuccess ? Is.EqualTo(null) : Contains.Substring("is not one of ['256x256', '512x512', '1024x1024']"), "Error message not returned");
         }
 
         [TestCase(true, HttpStatusCode.OK, "256x256", TestName = "Variation_FileInfoFileContent_When_Success")]