From 14c1d894033171c93b3a0ef3c1b66569d4eddb4a Mon Sep 17 00:00:00 2001 From: nnegrey Date: Wed, 11 Mar 2020 10:12:05 -0600 Subject: [PATCH] automl: update error check due to changes --- .../src/test/java/com/example/automl/BatchPredictTest.java | 4 ++-- .../src/test/java/com/example/automl/BatchPredictTest.java | 4 ++-- 2 files changed, 4 insertions(+), 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 index 688c6702c00..bcba7637ecd 100644 --- 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 index d46b669ac1d..974caaa6e7d 100644 --- 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"); } } }