def test_bo_rescheduling(self, create_build_job):
build = BuildJobFactory()
BuildJobStatus.objects.create(status=JobLifeCycle.SUCCEEDED, job=build)
create_build_job.return_value = build, True, True
with patch('hpsearch.tasks.bo.hp_bo_start.apply_async') as mock_fct:
ExperimentGroupFactory(content=experiment_group_spec_content_bo)
assert mock_fct.call_count == 2
with patch.object(GroupChecks, 'is_checked') as mock_is_check:
with patch('hpsearch.tasks.bo.hp_bo_iterate.apply_async'
) as mock_fct1:
with patch('scheduler.tasks.experiments.'
'experiments_build.apply_async') as mock_fct2:
mock_is_check.return_value = False
ExperimentGroupFactory(
content=experiment_group_spec_content_bo)
assert mock_fct1.call_count == 2
# 2 experiments, but since we are mocking the scheduling function, it's 4 calls,
# every call to start tries to schedule again, but in reality it's just 2 calls
assert mock_fct2.call_count == 4
# Fake
with patch('hpsearch.tasks.bo.hp_bo_start.apply_async') as mock_fct:
experiment_group = ExperimentGroupFactory(
content=experiment_group_spec_content_bo)
assert mock_fct.call_count == 2
assert experiment_group.non_done_experiments.count() == 2
# Mark experiment as done
with patch(
'scheduler.experiment_scheduler.stop_experiment') as _: # noqa
with patch('hpsearch.tasks.bo.hp_bo_start.apply_async'
) as xp_trigger_start:
for xp in experiment_group.experiments.all():
ExperimentStatusFactory(
experiment=xp, status=ExperimentLifeCycle.SUCCEEDED)
assert xp_trigger_start.call_count == experiment_group.experiments.count(
)
with patch('hpsearch.tasks.bo.hp_bo_iterate.apply_async') as mock_fct1:
hp_bo_start(experiment_group.id)
assert mock_fct1.call_count == 1
# Mark experiment as done
with patch(
'scheduler.experiment_scheduler.stop_experiment') as _: # noqa
with patch('hpsearch.tasks.bo.hp_bo_start.apply_async'
) as xp_trigger_start:
for xp in experiment_group.experiments.all():
ExperimentStatusFactory(
experiment=xp, status=ExperimentLifeCycle.SUCCEEDED)
assert xp_trigger_start.call_count == experiment_group.experiments.count(
)
GroupChecks(group=experiment_group.id).clear()
with patch('hpsearch.tasks.bo.hp_bo_create.apply_async') as mock_fct1:
hp_bo_start(experiment_group.id)
assert mock_fct1.call_count == 1
def test_bo_rescheduling(self):
with patch('hpsearch.tasks.bo.hp_bo_start.apply_async') as mock_fct:
ExperimentGroupFactory(content=experiment_group_spec_content_bo)
assert mock_fct.call_count == 1
with patch('hpsearch.tasks.bo.hp_bo_iterate.apply_async') as mock_fct1:
with patch(
'scheduler.tasks.experiments.experiments_build.apply_async'
) as mock_fct2:
ExperimentGroupFactory(
content=experiment_group_spec_content_bo)
assert mock_fct1.call_count == 1
assert mock_fct2.call_count == 2
# Fake
with patch('hpsearch.tasks.bo.hp_bo_start.apply_async') as mock_fct:
experiment_group = ExperimentGroupFactory(
content=experiment_group_spec_content_bo)
assert mock_fct.call_count == 1
assert experiment_group.non_done_experiments.count() == 2
# Mark experiment as done
with patch(
'scheduler.experiment_scheduler.stop_experiment') as _: # noqa
for xp in experiment_group.experiments.all():
ExperimentStatusFactory(experiment=xp,
status=ExperimentLifeCycle.SUCCEEDED)
with patch('hpsearch.tasks.bo.hp_bo_iterate.apply_async') as mock_fct1:
hp_bo_start(experiment_group.id)
assert mock_fct1.call_count == 1
# Mark experiment as done
with patch(
'scheduler.experiment_scheduler.stop_experiment') as _: # noqa
for xp in experiment_group.experiments.all():
ExperimentStatusFactory(experiment=xp,
status=ExperimentLifeCycle.SUCCEEDED)
with patch('hpsearch.tasks.bo.hp_bo_create.apply_async') as mock_fct1:
hp_bo_start(experiment_group.id)
assert mock_fct1.call_count == 1
def test_bo_rescheduling(self):
with patch('hpsearch.tasks.bo.hp_bo_start.apply_async') as mock_fct:
ExperimentGroupFactory(content=experiment_group_spec_content_bo)
assert mock_fct.call_count == 1
with patch('hpsearch.tasks.bo.hp_bo_iterate.apply_async') as mock_fct1:
with patch('scheduler.tasks.experiments.experiments_build.apply_async') as mock_fct2:
ExperimentGroupFactory(
content=experiment_group_spec_content_bo)
assert mock_fct1.call_count == 1
assert mock_fct2.call_count == 2
# Fake
with patch('hpsearch.tasks.bo.hp_bo_start.apply_async') as mock_fct:
experiment_group = ExperimentGroupFactory(
content=experiment_group_spec_content_bo)
assert mock_fct.call_count == 1
assert experiment_group.non_done_experiments.count() == 2
# Mark experiment as done
with patch('scheduler.experiment_scheduler.stop_experiment') as _: # noqa
for xp in experiment_group.experiments.all():
ExperimentStatusFactory(experiment=xp, status=ExperimentLifeCycle.SUCCEEDED)
with patch('hpsearch.tasks.bo.hp_bo_iterate.apply_async') as mock_fct1:
hp_bo_start(experiment_group.id)
assert mock_fct1.call_count == 1
# Mark experiment as done
with patch('scheduler.experiment_scheduler.stop_experiment') as _: # noqa
for xp in experiment_group.experiments.all():
ExperimentStatusFactory(experiment=xp, status=ExperimentLifeCycle.SUCCEEDED)
with patch('hpsearch.tasks.bo.hp_bo_create.apply_async') as mock_fct1:
hp_bo_start(experiment_group.id)
assert mock_fct1.call_count == 1