automl: fix batch test due to error message change (GoogleCloudPlatform#3057)

nnegrey · web-flow · commit 1718111c545e · 2020-03-06T14:07:55.000-08:00
fix: GoogleCloudPlatform#3052
diff --git a/automl/cloud-client/batch_predict_test.py b/automl/cloud-client/batch_predict_test.py
@@ -37,11 +37,11 @@ def test_batch_predict(capsys):
         )
         out, _ = capsys.readouterr()
         assert (
-            "The model is either not found or not supported for prediction yet"
+            "does not exist"
             in out
         )
     except Exception as e:
         assert (
-            "The model is either not found or not supported for prediction yet"
+            "does not exist"
             in e.message
         )

Original file line number	Diff line number	Diff line change
`@@ -37,11 +37,11 @@ def test_batch_predict(capsys):`
`37`	`37`	`)`
`38`	`38`	`out, _ = capsys.readouterr()`
`39`	`39`	`assert (`
`40`		`- "The model is either not found or not supported for prediction yet"`
	`40`	`+ "does not exist"`
`41`	`41`	`in out`
`42`	`42`	`)`
`43`	`43`	`except Exception as e:`
`44`	`44`	`assert (`
`45`		`- "The model is either not found or not supported for prediction yet"`
	`45`	`+ "does not exist"`
`46`	`46`	`in e.message`
`47`	`47`	`)`