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