def test_list_candidates_default(sagemaker_session):
auto_ml = AutoML(
role=ROLE, target_attribute_name=TARGET_ATTRIBUTE_NAME, sagemaker_session=sagemaker_session
)
auto_ml.current_job_name = "current_job_name"
auto_ml.list_candidates()
sagemaker_session.list_candidates.assert_called_once()
sagemaker_session.list_candidates.assert_called_with(job_name=auto_ml.current_job_name)
def test_best_candidate_job_no_desc(sagemaker_session):
auto_ml = AutoML(role=ROLE,
target_attribute_name=TARGET_ATTRIBUTE_NAME,
sagemaker_session=sagemaker_session)
auto_ml.current_job_name = JOB_NAME
best_candidate = auto_ml.best_candidate()
sagemaker_session.describe_auto_ml_job.assert_called_once()
sagemaker_session.describe_auto_ml_job.assert_called_with(JOB_NAME)
assert best_candidate == BEST_CANDIDATE
def test_best_candidate_default_job_name(sagemaker_session):
auto_ml = AutoML(role=ROLE,
target_attribute_name=TARGET_ATTRIBUTE_NAME,
sagemaker_session=sagemaker_session)
auto_ml.current_job_name = JOB_NAME
auto_ml._auto_ml_job_desc = AUTO_ML_DESC
best_candidate = auto_ml.best_candidate()
sagemaker_session.describe_auto_ml_job.assert_not_called()
assert best_candidate == BEST_CANDIDATE