vision: makes tests more generic (#2475)

Fix: #2424 Fix: #2408
GoogleCloudPlatform · Mar 23, 2020 · b2fcee6 · b2fcee6
1 parent 9bd162d
commit b2fcee6
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/vision/cloud-client/src/test/java/com/example/vision/DetectIT.java b/vision/cloud-client/src/test/java/com/example/vision/DetectIT.java
@@ -244,8 +244,8 @@ public void detectWebAnnotations() throws Exception {
 
     // Assert
     String got = bout.toString().toLowerCase();
-    assertThat(got).contains("history");
-    assertThat(got).contains("best guess label: palace of fine arts");
+    assertThat(got).contains("entity:id:score");
+    assertThat(got).contains("best guess label");
   }
 
   @Test
@@ -256,8 +256,8 @@ public void detectWebAnnotationsGcs() throws Exception {
 
     // Assert
     String got = bout.toString().toLowerCase();
-    assertThat(got).contains("history");
-    assertThat(got).contains("best guess label: palace of fine arts");
+    assertThat(got).contains("entity:id:score");
+    assertThat(got).contains("best guess label");
   }
 
   @Test