def test_tune_job_status_none(sagemaker_session):
hyperparameter_tuning_job_desc = {'HyperParameterTuningJobStatus': 'InProgress'}
sagemaker_session.sagemaker_client.describe_hyper_parameter_tuning_job = Mock(
name='describe_hyper_parameter_tuning_job', return_value=hyperparameter_tuning_job_desc)
result = _tuning_job_status(sagemaker_session.sagemaker_client, JOB_NAME)
assert result is None
def test_tune_job_status_none(sagemaker_session):
hyperparameter_tuning_job_desc = {'HyperParameterTuningJobStatus': 'InProgress'}
sagemaker_session.sagemaker_client.describe_hyper_parameter_tuning_job = Mock(
name='describe_hyper_parameter_tuning_job', return_value=hyperparameter_tuning_job_desc)
result = _tuning_job_status(sagemaker_session.sagemaker_client, JOB_NAME)
assert result is None
def test_tune_job_status(sagemaker_session):
hyperparameter_tuning_job_desc = {
"HyperParameterTuningJobStatus": "Completed"
}
sagemaker_session.sagemaker_client.describe_hyper_parameter_tuning_job = Mock(
name="describe_hyper_parameter_tuning_job",
return_value=hyperparameter_tuning_job_desc)
result = _tuning_job_status(sagemaker_session.sagemaker_client, JOB_NAME)
assert result["HyperParameterTuningJobStatus"] == "Completed"
