Esempi in Python per _transform_job_status

Esempio n. 1

File: test_session.py Progetto: cheesama/sagemaker-python-sdk

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

Esempio n. 2

File: test_session.py Progetto: cheesama/sagemaker-python-sdk

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'

Esempio n. 3

File: test_session.py Progetto: Jarryd-rk/sagemaker-python-sdk

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'

Esempio n. 4

File: test_session.py Progetto: Jarryd-rk/sagemaker-python-sdk

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

Esempio n. 5

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"