@pytest.mark.skipif((tests.integ.test_region() in tests.integ.NO_AUTO_ML_REGIONS), reason='AutoML is not supported in the region yet.')
def test_auto_ml_describe_auto_ml_job(sagemaker_session):
    expected_default_input_config = [{'DataSource': {'S3DataSource': {'S3DataType': 'S3Prefix', 'S3Uri': 's3://{}/{}/input/iris_training.csv'.format(sagemaker_session.default_bucket(), PREFIX)}}, 'TargetAttributeName': TARGET_ATTRIBUTE_NAME}]
    expected_default_output_config = {'S3OutputPath': 's3://{}/'.format(sagemaker_session.default_bucket())}
    auto_ml_utils.create_auto_ml_job_if_not_exist(sagemaker_session)
    auto_ml = AutoML(role=ROLE, target_attribute_name=TARGET_ATTRIBUTE_NAME, sagemaker_session=sagemaker_session)
    desc = auto_ml.describe_auto_ml_job(job_name=AUTO_ML_JOB_NAME)
    assert (desc['AutoMLJobName'] == AUTO_ML_JOB_NAME)
    assert (desc['AutoMLJobStatus'] == 'Completed')
    assert isinstance(desc['BestCandidate'], dict)
    assert (desc['InputDataConfig'] == expected_default_input_config)
    assert (desc['AutoMLJobConfig'] == EXPECTED_DEFAULT_JOB_CONFIG)
    assert (desc['OutputDataConfig'] == expected_default_output_config)
@pytest.mark.skipif((tests.integ.test_region() in tests.integ.NO_AUTO_ML_REGIONS), reason='AutoML is not supported in the region yet.')
def test_auto_ml_describe_auto_ml_job(sagemaker_session):
    expected_default_input_config = [{'DataSource': {'S3DataSource': {'S3DataType': 'S3Prefix', 'S3Uri': 's3://{}/{}/input/iris_training.csv'.format(sagemaker_session.default_bucket(), PREFIX)}}, 'TargetAttributeName': TARGET_ATTRIBUTE_NAME}]
    expected_default_output_config = {'S3OutputPath': 's3://{}/'.format(sagemaker_session.default_bucket())}
    auto_ml_utils.create_auto_ml_job_if_not_exist(sagemaker_session)
    auto_ml = AutoML(role=ROLE, target_attribute_name=TARGET_ATTRIBUTE_NAME, sagemaker_session=sagemaker_session)
    desc = auto_ml.describe_auto_ml_job(job_name=AUTO_ML_JOB_NAME)
    assert (desc['AutoMLJobName'] == AUTO_ML_JOB_NAME)
    assert (desc['AutoMLJobStatus'] == 'Completed')
    assert isinstance(desc['BestCandidate'], dict)
    assert (desc['InputDataConfig'] == expected_default_input_config)
    assert (desc['AutoMLJobConfig'] == EXPECTED_DEFAULT_JOB_CONFIG)
    assert (desc['OutputDataConfig'] == expected_default_output_config)