def test_train_done_in_progress(sagemaker_session): training_job_desc = {'TrainingJobStatus': 'InProgress'} sagemaker_session.sagemaker_client.describe_training_job = Mock( name='describe_training_job', return_value=training_job_desc) actual_job_desc, training_finished = _train_done(sagemaker_session.sagemaker_client, JOB_NAME, None) assert actual_job_desc['TrainingJobStatus'] == 'InProgress' assert training_finished is False
def test_train_done_completed(sagemaker_session): training_job_desc = {'TrainingJobStatus': 'Completed'} sagemaker_session.sagemaker_client.describe_training_job = Mock( name='describe_training_job', return_value=training_job_desc) actual_job_desc, training_finished = _train_done(sagemaker_session.sagemaker_client, JOB_NAME, None) assert actual_job_desc['TrainingJobStatus'] == 'Completed' assert training_finished is True
def test_train_done_in_progress(sagemaker_session): training_job_desc = {"TrainingJobStatus": "InProgress"} sagemaker_session.sagemaker_client.describe_training_job = Mock( name="describe_training_job", return_value=training_job_desc) actual_job_desc, training_finished = _train_done( sagemaker_session.sagemaker_client, JOB_NAME, None) assert actual_job_desc["TrainingJobStatus"] == "InProgress" assert training_finished is False
def test_train_done_completed(sagemaker_session): training_job_desc = {"TrainingJobStatus": "Completed"} sagemaker_session.sagemaker_client.describe_training_job = Mock( name="describe_training_job", return_value=training_job_desc) actual_job_desc, training_finished = _train_done( sagemaker_session.sagemaker_client, JOB_NAME, None) assert actual_job_desc["TrainingJobStatus"] == "Completed" assert training_finished is True