示例#1
0
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)
示例#2
0
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()
示例#3
0
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()
示例#4
0
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"}
示例#5
0
def test_transform_job_wait(sagemaker_session):
    job = _TransformJob(sagemaker_session, JOB_NAME)
    job.wait()

    assert sagemaker_session.wait_for_transform_job.called_once
示例#6
0
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()
示例#7
0
def test_transform_job_wait(sagemaker_session):
    job = _TransformJob(sagemaker_session, JOB_NAME)
    job.wait()

    assert sagemaker_session.wait_for_transform_job.called_once
示例#8
0
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()