def test_find_by_trial_and_objective(session: Session) -> None:
trial = TestTrialValueModel._create_model(session)
trial_value = TrialValueModel.find_by_trial_and_objective(trial, 0, session)
assert trial_value is not None
assert 10 == trial_value.value
assert TrialValueModel.find_by_trial_and_objective(trial, 1, session) is None
def test_cascade_delete_on_trial(session: Session) -> None:
trial = TestTrialValueModel._create_model(session)
trial.values.append(TrialValueModel(trial_id=1, objective=1, value=20))
session.commit()
assert 2 == len(TrialValueModel.where_trial_id(trial.trial_id, session))
session.delete(trial)
session.commit()
assert 0 == len(TrialValueModel.where_trial_id(trial.trial_id, session))
def test_where_trial_id(session: Session) -> None:
trial = TestTrialValueModel._create_model(session)
trial_values = TrialValueModel.where_trial_id(trial.trial_id, session)
assert 1 == len(trial_values)
assert 0 == trial_values[0].objective
assert 10 == trial_values[0].value
def _create_model(session: Session) -> TrialModel:
direction = StudyDirectionModel(direction=StudyDirection.MINIMIZE, objective=0)
study = StudyModel(study_id=1, study_name="test-study", directions=[direction])
trial = TrialModel(trial_id=1, study_id=study.study_id, state=TrialState.COMPLETE)
session.add(study)
session.add(trial)
session.add(TrialValueModel(trial_id=trial.trial_id, objective=0, value=10))
session.commit()
return trial
