Update testGenerateImage_inlineData to check size instead of contents

Author: andrewheard

FirebaseVertexAI/Tests/TestApp/Tests/Integration/IntegrationTests.swift

@@ -250,19 +250,22 @@ final class IntegrationTests: XCTestCase {
     the lion, shallow depth of field, detailed fur texture, DSLR, 85mm lens.
     """
 
-    let imageResponse = try await imagenModel.generateImages(prompt: imagePrompt)
-
-    XCTAssertNil(imageResponse.raiFilteredReason)
-    XCTAssertEqual(imageResponse.images.count, 1)
-    let image = try XCTUnwrap(imageResponse.images.first)
-
-    let textResponse = try await model.generateContent(
-      InlineDataPart(data: image.data, mimeType: "image/png"),
-      "What is the name of this animal? Answer with the animal name only."
-    )
-
-    let text = try XCTUnwrap(textResponse.text).trimmingCharacters(in: .whitespacesAndNewlines)
-    XCTAssertEqual(text, "Lion")
+    let response = try await imagenModel.generateImages(prompt: imagePrompt)
+
+    XCTAssertNil(response.raiFilteredReason)
+    XCTAssertEqual(response.images.count, 1)
+    let image = try XCTUnwrap(response.images.first)
+    XCTAssertEqual(image.mimeType, "image/png")
+    XCTAssertGreaterThan(image.data.count, 0)
+    let imagenImage = image.imagenImage
+    XCTAssertEqual(imagenImage.mimeType, image.mimeType)
+    XCTAssertEqual(imagenImage.bytesBase64Encoded, image.data.base64EncodedString())
+    XCTAssertNil(imagenImage.gcsURI)
+    #if canImport(UIKit)
+      let uiImage = try XCTUnwrap(UIImage(data: image.data))
+      XCTAssertEqual(uiImage.size.width, 1024.0)
+      XCTAssertEqual(uiImage.size.height, 1024.0)
+    #endif
   }
 }