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
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
