def test_transform_job_status_none(sagemaker_session): transform_job_desc = {'TransformJobStatus': 'InProgress'} sagemaker_session.sagemaker_client.describe_transform_job = Mock( name='describe_transform_job', return_value=transform_job_desc) result = _transform_job_status(sagemaker_session.sagemaker_client, JOB_NAME) assert result is None
def test_transform_job_status(sagemaker_session): transform_job_desc = {'TransformJobStatus': 'Completed'} sagemaker_session.sagemaker_client.describe_transform_job = Mock( name='describe_transform_job', return_value=transform_job_desc) result = _transform_job_status(sagemaker_session.sagemaker_client, JOB_NAME) assert result['TransformJobStatus'] == 'Completed'
def test_transform_job_status(sagemaker_session): transform_job_desc = {"TransformJobStatus": "Completed"} sagemaker_session.sagemaker_client.describe_transform_job = Mock( name="describe_transform_job", return_value=transform_job_desc) result = _transform_job_status(sagemaker_session.sagemaker_client, JOB_NAME) assert result["TransformJobStatus"] == "Completed"