From 2c79e78d188f6f0bb15bec791e19635367345f38 Mon Sep 17 00:00:00 2001 From: Katie McLaughlin Date: Fri, 10 Oct 2025 15:44:18 +1100 Subject: [PATCH 1/2] ci(vertexai): test for type, not for value --- vertexai/snippets/src/test/java/vertexai/gemini/SnippetsIT.java | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vertexai/snippets/src/test/java/vertexai/gemini/SnippetsIT.java b/vertexai/snippets/src/test/java/vertexai/gemini/SnippetsIT.java index b74908e59ec..bd47f88a616 100644 --- a/vertexai/snippets/src/test/java/vertexai/gemini/SnippetsIT.java +++ b/vertexai/snippets/src/test/java/vertexai/gemini/SnippetsIT.java @@ -175,7 +175,7 @@ public void testTokenCount() throws Exception { @Test public void testMediaTokenCount() throws Exception { int tokenCount = GetMediaTokenCount.getMediaTokenCount(PROJECT_ID, LOCATION, GEMINI_FLASH); - assertThat(tokenCount).isEqualTo(16252); + assertThat(tokenCount).isInstanceOf(int.class); } @Test From be0bb48a89ee6b01a8c3e525431d1de3fa76cebe Mon Sep 17 00:00:00 2001 From: Katie McLaughlin Date: Fri, 10 Oct 2025 16:40:59 +1100 Subject: [PATCH 2/2] use isNotNull instead --- vertexai/snippets/src/test/java/vertexai/gemini/SnippetsIT.java | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vertexai/snippets/src/test/java/vertexai/gemini/SnippetsIT.java b/vertexai/snippets/src/test/java/vertexai/gemini/SnippetsIT.java index bd47f88a616..8f43bd98a01 100644 --- a/vertexai/snippets/src/test/java/vertexai/gemini/SnippetsIT.java +++ b/vertexai/snippets/src/test/java/vertexai/gemini/SnippetsIT.java @@ -175,7 +175,7 @@ public void testTokenCount() throws Exception { @Test public void testMediaTokenCount() throws Exception { int tokenCount = GetMediaTokenCount.getMediaTokenCount(PROJECT_ID, LOCATION, GEMINI_FLASH); - assertThat(tokenCount).isInstanceOf(int.class); + assertThat(tokenCount).isNotNull(); } @Test