Skip to content

Commit 4289e5f

Browse files
committed
Fix tests
1 parent fd29af9 commit 4289e5f

File tree

3 files changed

+6
-10
lines changed

3 files changed

+6
-10
lines changed

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

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -267,10 +267,6 @@ final class IntegrationTests: XCTestCase {
267267
let image = try XCTUnwrap(response.images.first)
268268
XCTAssertEqual(image.mimeType, "image/jpeg")
269269
XCTAssertGreaterThan(image.data.count, 0)
270-
let imagenImage = image.imagenImage
271-
XCTAssertEqual(imagenImage.mimeType, image.mimeType)
272-
XCTAssertEqual(imagenImage.bytesBase64Encoded, image.data.base64EncodedString())
273-
XCTAssertNil(imagenImage.gcsURI)
274270
#if canImport(UIKit)
275271
let uiImage = try XCTUnwrap(UIImage(data: image.data))
276272
XCTAssertEqual(uiImage.size.width, 1408.0)

FirebaseVertexAI/Tests/Unit/Types/Imagen/ImagenFileDataImageTests.swift

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -35,9 +35,9 @@ final class ImagenFileDataImageTests: XCTestCase {
3535

3636
XCTAssertEqual(image.mimeType, mimeType)
3737
XCTAssertEqual(image.gcsURI, gcsURI)
38-
XCTAssertEqual(image.imagenImage.mimeType, mimeType)
39-
XCTAssertEqual(image.imagenImage.gcsURI, gcsURI)
40-
XCTAssertNil(image.imagenImage.bytesBase64Encoded)
38+
XCTAssertEqual(image._imagenImage.mimeType, mimeType)
39+
XCTAssertEqual(image._imagenImage.gcsURI, gcsURI)
40+
XCTAssertNil(image._imagenImage.bytesBase64Encoded)
4141
}
4242

4343
func testDecodeImage_missingGCSURI_throws() throws {

FirebaseVertexAI/Tests/Unit/Types/Imagen/ImagenInlineDataImageTests.swift

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -35,9 +35,9 @@ final class ImagenInlineDataImageTests: XCTestCase {
3535

3636
XCTAssertEqual(image.mimeType, mimeType)
3737
XCTAssertEqual(image.data.base64EncodedString(), bytesBase64Encoded)
38-
XCTAssertEqual(image.imagenImage.mimeType, mimeType)
39-
XCTAssertEqual(image.imagenImage.bytesBase64Encoded, bytesBase64Encoded)
40-
XCTAssertNil(image.imagenImage.gcsURI)
38+
XCTAssertEqual(image._imagenImage.mimeType, mimeType)
39+
XCTAssertEqual(image._imagenImage.bytesBase64Encoded, bytesBase64Encoded)
40+
XCTAssertNil(image._imagenImage.gcsURI)
4141
}
4242

4343
func testDecodeImage_missingBytesBase64Encoded_throws() throws {

0 commit comments

Comments
 (0)