fix: fix predict sample tests for proto-plus==1.14.2 (#250)

dizcology · web-flow · commit b1c9d88646f0 · 2021-03-01T11:24:45.000-08:00
diff --git a/samples/snippets/predict_custom_trained_model_sample_test.py b/samples/snippets/predict_custom_trained_model_sample_test.py
@@ -37,4 +37,4 @@ def test_ucaip_generated_predict_custom_trained_model_sample(capsys):
     )
 
     out, _ = capsys.readouterr()
-    assert "number_value: 1.0" in out
+    assert "1.0" in out
diff --git a/samples/snippets/predict_image_object_detection_sample_test.py b/samples/snippets/predict_image_object_detection_sample_test.py
@@ -31,4 +31,4 @@ def test_ucaip_generated_predict_image_object_detection_sample(capsys):
     )
 
     out, _ = capsys.readouterr()
-    assert 'string_value: "Salad"' in out
+    assert 'Salad' in out
diff --git a/samples/snippets/predict_tabular_classification_sample_test.py b/samples/snippets/predict_tabular_classification_sample_test.py
@@ -35,4 +35,4 @@ def test_ucaip_generated_predict_tabular_classification_sample(capsys):
     )
 
     out, _ = capsys.readouterr()
-    assert 'string_value: "setosa"' in out
+    assert 'setosa' in out

Original file line number	Diff line number	Diff line change
`@@ -37,4 +37,4 @@ def test_ucaip_generated_predict_custom_trained_model_sample(capsys):`
`37`	`37`	`)`
`38`	`38`
`39`	`39`	`out, _ = capsys.readouterr()`
`40`		`- assert "number_value: 1.0" in out`
	`40`	`+ assert "1.0" in out`
Original file line number	Diff line number	Diff line change
`@@ -31,4 +31,4 @@ def test_ucaip_generated_predict_image_object_detection_sample(capsys):`
`31`	`31`	`)`
`32`	`32`
`33`	`33`	`out, _ = capsys.readouterr()`
`34`		`- assert 'string_value: "Salad"' in out`
	`34`	`+ assert 'Salad' in out`
Original file line number	Diff line number	Diff line change
`@@ -35,4 +35,4 @@ def test_ucaip_generated_predict_tabular_classification_sample(capsys):`
`35`	`35`	`)`
`36`	`36`
`37`	`37`	`out, _ = capsys.readouterr()`
`38`		`- assert 'string_value: "setosa"' in out`
	`38`	`+ assert 'setosa' in out`