def test_archive(self):
with patch('hpsearch.tasks.grid.hp_grid_search_start.apply_async'
) as mock_fct:
experiment_group = ExperimentGroupFactory()
assert mock_fct.call_count == 2
assert experiment_group.deleted is False
assert Experiment.objects.filter(
experiment_group=experiment_group).count() == 2
assert ExperimentGroup.objects.count() == 1
assert ExperimentGroup.all.count() == 1
experiment_group.archive()
assert experiment_group.deleted is True
assert ExperimentGroup.objects.count() == 0
assert ExperimentGroup.all.count() == 1
assert Experiment.objects.filter(
experiment_group=experiment_group).count() == 0
assert Experiment.all.filter(
experiment_group=experiment_group).count() == 2
assert experiment_group.experiments.count() == 0
assert experiment_group.all_experiments.count() == 2
experiment_group.unarchive()
assert experiment_group.deleted is False
assert ExperimentGroup.objects.count() == 1
assert ExperimentGroup.all.count() == 1
assert Experiment.objects.filter(
experiment_group=experiment_group).count() == 2
assert Experiment.all.filter(
experiment_group=experiment_group).count() == 2
assert experiment_group.experiments.count() == 2
assert experiment_group.all_experiments.count() == 2
def test_archive(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.grid.hp_grid_search_start.apply_async') as mock_fct:
experiment_group = ExperimentGroupFactory()
assert mock_fct.call_count == 1
assert experiment_group.deleted is False
assert Experiment.objects.filter(experiment_group=experiment_group).count() == 2
assert ExperimentGroup.objects.count() == 1
assert ExperimentGroup.all.count() == 1
experiment_group.archive()
assert experiment_group.deleted is True
assert ExperimentGroup.objects.count() == 0
assert ExperimentGroup.all.count() == 1
assert Experiment.objects.filter(experiment_group=experiment_group).count() == 0
assert Experiment.all.filter(experiment_group=experiment_group).count() == 2
assert experiment_group.experiments.count() == 0
assert experiment_group.all_experiments.count() == 2
experiment_group.restore()
assert experiment_group.deleted is False
assert ExperimentGroup.objects.count() == 1
assert ExperimentGroup.all.count() == 1
assert Experiment.objects.filter(experiment_group=experiment_group).count() == 2
assert Experiment.all.filter(experiment_group=experiment_group).count() == 2
assert experiment_group.experiments.count() == 2
assert experiment_group.all_experiments.count() == 2
def test_delete_experiment_groups(self):
project1 = ProjectFactory()
ExperimentGroupFactory(project=project1)
project1.archive()
project2 = ProjectFactory()
experiment_group2 = ExperimentGroupFactory(project=project2)
experiment_group2.archive()
assert ExperimentGroup.all.count() == 2
CleaningIntervals.ARCHIVED = -10
delete_archived_experiment_groups()
# Although the other entity is archived it's not deleted because of project1
assert ExperimentGroup.all.count() == 1
def test_delete_experiment_groups(self):
project1 = ProjectFactory()
ExperimentGroupFactory(project=project1)
project1.archive()
project2 = ProjectFactory()
experiment_group2 = ExperimentGroupFactory(project=project2)
experiment_group2.archive()
assert ExperimentGroup.all.count() == 2
conf.set(CLEANING_INTERVALS_ARCHIVES, -10)
delete_archived_experiment_groups()
# Although the other entity is archived it's not deleted because of project1
assert ExperimentGroup.all.count() == 1
