Skip to content

Commit 3992399

Browse files
committed
Add more stringent assert statements to test_processing_job_from_processing_arn
1 parent 6d77969 commit 3992399

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

tests/unit/test_processing.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -443,6 +443,16 @@ def test_processing_job_from_processing_arn(sagemaker_session):
443443
processing_job_arn="arn:aws:sagemaker:dummy-region:dummy-account-number:processing-job/dummy-job-name",
444444
)
445445
assert isinstance(processing_job, ProcessingJob)
446+
assert [
447+
processing_input._to_request_dict() for processing_input in processing_job.inputs
448+
] == PROCESSING_JOB_DESCRIPTION["ProcessingInputs"]
449+
assert [
450+
processing_output._to_request_dict() for processing_output in processing_job.outputs
451+
] == PROCESSING_JOB_DESCRIPTION["ProcessingOutputConfig"]["Outputs"]
452+
assert (
453+
processing_job.output_kms_key
454+
== PROCESSING_JOB_DESCRIPTION["ProcessingOutputConfig"]["KmsKeyId"]
455+
)
446456

447457

448458
def _get_script_processor(sagemaker_session):

0 commit comments

Comments
 (0)