def testRunnerValidation(self):
sqa_runner = SQARunner(runner_type=RUNNER_REGISTRY[SyntheticRunner])
with self.assertRaises(ValueError):
with session_scope() as session:
session.add(sqa_runner)
sqa_runner.experiment_id = 0
with session_scope() as session:
session.add(sqa_runner)
with self.assertRaises(ValueError):
sqa_runner.trial_id = 0
with session_scope() as session:
session.add(sqa_runner)
sqa_runner = SQARunner(runner_type=RUNNER_REGISTRY[SyntheticRunner], trial_id=0)
with session_scope() as session:
session.add(sqa_runner)
with self.assertRaises(ValueError):
sqa_runner.experiment_id = 0
with session_scope() as session:
session.add(sqa_runner)
def add_experiment_id(sqa: SQARunner):
sqa.experiment_id = exp_id
