def test_stop_transform_job(sagemaker_session, transformer): sagemaker_session.stop_transform_job = Mock(name="stop_transform_job") transformer.latest_transform_job = _TransformJob(sagemaker_session, JOB_NAME) transformer.stop_transform_job() sagemaker_session.stop_transform_job.assert_called_once_with(name=JOB_NAME)
def test_start_new(transformer, sagemaker_session): transformer._current_job_name = JOB_NAME job = _TransformJob(sagemaker_session, JOB_NAME) started_job = job.start_new(transformer, DATA, S3_DATA_TYPE, None, None, None, None, None, None) assert started_job.sagemaker_session == sagemaker_session sagemaker_session.transform.assert_called_once()
def test_start_new(transformer, sagemaker_session): transformer._current_job_name = JOB_NAME job = _TransformJob(sagemaker_session, JOB_NAME) started_job = job.start_new(transformer, DATA, S3_DATA_TYPE, None, None, None) assert started_job.sagemaker_session == sagemaker_session sagemaker_session.transform.assert_called_once()
def test_start_new(transformer, sagemaker_session): transformer._current_job_name = JOB_NAME job = _TransformJob(sagemaker_session, JOB_NAME) started_job = job.start_new( transformer, DATA, S3_DATA_TYPE, None, None, None, None, None, None, {"ExperimentName": "exp"}, ) assert started_job.sagemaker_session == sagemaker_session sagemaker_session.transform.assert_called_once() called_args = sagemaker_session.transform.call_args assert called_args[1]["experiment_config"] == {"ExperimentName": "exp"}
def test_transform_job_wait(sagemaker_session): job = _TransformJob(sagemaker_session, JOB_NAME) job.wait() assert sagemaker_session.wait_for_transform_job.called_once
def test_ensure_last_transform_job_exists(transformer, sagemaker_session): transformer.latest_transform_job = _TransformJob(sagemaker_session, 'some-transform-job') transformer._ensure_last_transform_job()