[TEST] Fix sagemaker e2e test prediction result (#735)

* include newline in sagemaker assert result * update base on comment
bentoml · May 31, 2020 · dab9616 · dab9616
1 parent 23bad62
commit dab9616
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/e2e_tests/aws_sagemaker/test_sagemaker_update_deployment.py b/e2e_tests/aws_sagemaker/test_sagemaker_update_deployment.py
@@ -38,7 +38,9 @@ def test_sagemaker_update_deployment(basic_bentoservice_v1, basic_bentoservice_v
 
         request_success, prediction_result = send_test_data_to_endpoint(endpoint_name)
         assert request_success, 'Failed to make successful Sagemaker request'
-        assert prediction_result == "cat", 'Sagemaker prediction result mismatch'
+        assert (
+            prediction_result.strip() == '"cat"'
+        ), 'Sagemaker prediction result mismatch'
 
         update_bento_version_deployment_command = [
             'bentoml',
@@ -63,6 +65,8 @@ def test_sagemaker_update_deployment(basic_bentoservice_v1, basic_bentoservice_v
 
         request_success, prediction_result = send_test_data_to_endpoint(endpoint_name)
         assert request_success, 'Failed to make successful Sagemaker request'
-        assert prediction_result == "dog", 'Sagemaker prediction result mismatch'
+        assert (
+            prediction_result.strip() == '"dog"'
+        ), 'Sagemaker prediction result mismatch'
     finally:
         delete_deployment('sagemaker', deployment_name)