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"