automl: update error check due to changes (#2377)

GoogleCloudPlatform · Mar 11, 2020 · cbac09e · cbac09e
1 parent f4f43cc
commit cbac09e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/automl/beta/src/test/java/com/example/automl/BatchPredictTest.java b/automl/beta/src/test/java/com/example/automl/BatchPredictTest.java
@@ -74,10 +74,10 @@ public void testBatchPredict() {
       BatchPredict.batchPredict(PROJECT_ID, MODEL_ID, inputUri, outputUri);
       String got = bout.toString();
       assertThat(got)
-          .contains("The model is either not found or not supported for prediction yet.");
+          .contains("does not exist");
     } catch (IOException | ExecutionException | InterruptedException e) {
       assertThat(e.getMessage())
-          .contains("The model is either not found or not supported for prediction yet.");
+          .contains("does not exist");
     }
   }
 }
diff --git a/automl/cloud-client/src/test/java/com/example/automl/BatchPredictTest.java b/automl/cloud-client/src/test/java/com/example/automl/BatchPredictTest.java
@@ -83,10 +83,10 @@ public void testBatchPredict() {
       BatchPredict.batchPredict(PROJECT_ID, MODEL_ID, inputUri, outputUri);
       String got = bout.toString();
       assertThat(got)
-          .contains("The model is either not found or not supported for prediction yet.");
+          .contains("does not exist");
     } catch (IOException | ExecutionException | InterruptedException e) {
       assertThat(e.getMessage())
-          .contains("The model is either not found or not supported for prediction yet.");
+          .contains("does not exist");
     }
   }
 }