def test_candidate_estimator_get_steps(sagemaker_session):
auto_ml = AutoML(role=ROLE,
target_attribute_name=TARGET_ATTRIBUTE_NAME,
sagemaker_session=sagemaker_session)
candidates = auto_ml.list_candidates(job_name=AUTO_ML_JOB_NAME)
candidate = candidates[1]
candidate_estimator = CandidateEstimator(candidate, sagemaker_session)
steps = candidate_estimator.get_steps()
assert len(steps) == 3
def test_candidate_estimator_get_steps(sagemaker_session):
candidate_estimator = CandidateEstimator(
CANDIDATE_DICT, sagemaker_session=sagemaker_session)
steps = candidate_estimator.get_steps()
assert len(steps) == 3
