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_jobs(self):
project1 = ProjectFactory()
JobFactory(project=project1)
project1.archive()
project2 = ProjectFactory()
job2 = JobFactory(project=project2)
job2.archive()
assert Job.all.count() == 2
CleaningIntervals.ARCHIVED = -10
delete_archived_jobs()
# Although the other entity is archived it's not deleted because of project1
assert Job.all.count() == 1
def test_delete_build_jobs(self):
project1 = ProjectFactory()
BuildJobFactory(project=project1)
project1.archive()
project2 = ProjectFactory()
job2 = BuildJobFactory(project=project2)
job2.archive()
assert BuildJob.all.count() == 2
conf.set(CLEANING_INTERVALS_ARCHIVES, -10)
delete_archived_build_jobs()
# Although the other entity is archived it's not deleted because of project1
assert BuildJob.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
def test_archive(self):
project = ProjectFactory()
ExperimentGroupFactory(project=project)
ExperimentFactory(project=project)
JobFactory(project=project)
BuildJobFactory(project=project)
NotebookJobFactory(project=project)
TensorboardJobFactory(project=project)
assert project.deleted is False
assert project.experiments.count() == 1
assert project.experiment_groups.count() == 1
assert project.jobs.count() == 1
assert project.build_jobs.count() == 1
assert project.notebook_jobs.count() == 1
assert project.tensorboard_jobs.count() == 1
assert project.all_experiments.count() == 1
assert project.all_experiment_groups.count() == 1
assert project.all_notebook_jobs.count() == 1
assert project.all_tensorboard_jobs.count() == 1
project.archive()
assert project.deleted is True
assert project.experiments.count() == 0
assert project.experiment_groups.count() == 0
assert project.jobs.count() == 0
assert project.build_jobs.count() == 0
assert project.notebook_jobs.count() == 0
assert project.tensorboard_jobs.count() == 0
assert project.all_experiments.count() == 1
assert project.all_experiment_groups.count() == 1
assert project.all_notebook_jobs.count() == 1
assert project.all_tensorboard_jobs.count() == 1
project.unarchive()
assert project.deleted is False
assert project.experiments.count() == 1
assert project.experiment_groups.count() == 1
assert project.jobs.count() == 1
assert project.build_jobs.count() == 1
assert project.notebook_jobs.count() == 1
assert project.tensorboard_jobs.count() == 1
assert project.all_experiments.count() == 1
assert project.all_experiment_groups.count() == 1
assert project.all_notebook_jobs.count() == 1
assert project.all_tensorboard_jobs.count() == 1
def test_delete_experiments(self):
project1 = ProjectFactory()
ExperimentFactory(project=project1)
project1.archive()
project2 = ProjectFactory()
experiment2 = ExperimentFactory(project=project2)
experiment2.archive()
group1 = ExperimentGroupFactory()
experiment3 = ExperimentFactory(experiment_group=group1)
experiment3.archive()
assert Experiment.all.count() == 3
CleaningIntervals.ARCHIVED = -10
delete_archived_experiments()
# Although the other experiment is archived it's not deleted because of project1 and group1
assert Experiment.all.count() == 2
def test_delete_projects(self):
project1 = ProjectFactory()
ExperimentFactory(project=project1)
project1.archive()
project2 = ProjectFactory()
experiment2 = ExperimentFactory(project=project2)
experiment2.archive()
assert Project.all.count() == 2
assert Experiment.all.count() == 2
CleaningIntervals.ARCHIVED = -10
delete_archived_projects()
# Deletes only one project
assert Project.all.count() == 1
# Although the other experiment is archived it's not deleted because of project2
assert Experiment.all.count() == 1
def test_delete_projects(self):
project1 = ProjectFactory()
ExperimentFactory(project=project1)
project1.archive()
project2 = ProjectFactory()
experiment2 = ExperimentFactory(project=project2)
experiment2.archive()
assert Project.all.count() == 2
assert Experiment.all.count() == 2
conf.set(CLEANING_INTERVALS_ARCHIVES, -10)
delete_archived_projects()
# Deletes only one project
assert Project.all.count() == 1
# Although the other experiment is archived it's not deleted because of project2
assert Experiment.all.count() == 1
