update unit tests

laurenyu · laurenyu · commit bb994178cf92 · 2019-12-05T11:36:42.000-08:00
diff --git a/tests/unit/test_processing.py b/tests/unit/test_processing.py
@@ -40,6 +40,7 @@ def sagemaker_session():
         name="upload_data", return_value="mocked_s3_uri_from_upload_data"
     )
     session_mock.download_data = Mock(name="download_data")
+    session_mock.expand_role.return_value = ROLE
     return session_mock
 
 

Original file line number	Diff line number	Diff line change
`@@ -40,6 +40,7 @@ def sagemaker_session():`
`40`	`40`	`name="upload_data", return_value="mocked_s3_uri_from_upload_data"`
`41`	`41`	`)`
`42`	`42`	`session_mock.download_data = Mock(name="download_data")`
	`43`	`+ session_mock.expand_role.return_value = ROLE`
`43`	`44`	`return session_mock`
`44`	`45`
`45`	`46`