vision: makes tests more generic

GoogleCloudPlatform · Mar 23, 2020 · 83f105d · 83f105d
1 parent c7a74fb
commit 83f105d
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