def testExperimentUpdateTrial(self): save_experiment(self.experiment) trial = self.experiment.trials[0] trial.mark_staged() update_trial(experiment=self.experiment, trial=trial) loaded_experiment = load_experiment(self.experiment.name) self.assertEqual(trial, loaded_experiment.trials[0]) trial._run_metadata = {"foo": "bar"} update_trial(experiment=self.experiment, trial=trial) loaded_experiment = load_experiment(self.experiment.name) self.assertEqual(trial, loaded_experiment.trials[0]) self.experiment.attach_data(get_data(trial_index=trial.index)) update_trial(experiment=self.experiment, trial=trial) loaded_experiment = load_experiment(self.experiment.name) self.assertEqual(self.experiment, loaded_experiment) trial = self.experiment.new_batch_trial( generator_run=get_generator_run()) save_new_trial(experiment=self.experiment, trial=trial) self.experiment.attach_data(get_data(trial_index=trial.index)) update_trial(experiment=self.experiment, trial=trial) loaded_experiment = load_experiment(self.experiment.name) self.assertEqual(self.experiment, loaded_experiment)
def testExperimentNewTrialValidation(self): trial = self.experiment.new_batch_trial() with self.assertRaises(ValueError): # must save experiment first save_new_trial(experiment=self.experiment, trial=trial) save_experiment(self.experiment) with self.assertRaises(ValueError): # can't save new trial twice save_new_trial(experiment=self.experiment, trial=trial)
def testExperimentNewTrial(self): save_experiment(self.experiment) trial = self.experiment.new_batch_trial() save_new_trial(experiment=self.experiment, trial=trial) loaded_experiment = load_experiment(self.experiment.name) self.assertEqual(len(loaded_experiment.trials), 2) self.assertEqual(trial, loaded_experiment.trials[1]) trial = self.experiment.new_batch_trial(generator_run=get_generator_run()) save_new_trial(experiment=self.experiment, trial=trial) loaded_experiment = load_experiment(self.experiment.name) self.assertEqual(len(loaded_experiment.trials), 3) self.assertEqual(trial, loaded_experiment.trials[2])