wassname/optuna-dashboard
1
0
Fork 0
mirror of https://github.com/wassname/optuna-dashboard.git synced 2026-09-11 12:30:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix test

Browse Source
This commit is contained in:
moririn2528
2023-08-16 13:20:30 +09:00
parent a9d9621818
commit 4862faa8f0
1 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python_tests/preferential/test_system_attrs.py
+7 -6
View File
@@ -17,12 +17,13 @@ def test_report_and_get_preferences(storage_supplier: Callable[[], StorageSuppli
study.ask()
study.ask()
assert len(get_preferences(study)) == 0
study_id = study._study_id
assert len(get_preferences(study_id, storage)) == 0
better, worse = study.trials[0], study.trials[1]
report_preferences(study, [(better, worse)])
assert len(get_preferences(study)) == 1
report_preferences(study_id, storage, [(better.number, worse.number)])
assert len(get_preferences(study_id, storage)) == 1
actual_better, actual_worse = get_preferences(study)[0]
assert actual_better.number == better.number
assert actual_worse.number == worse.number
actual_better, actual_worse = get_preferences(study_id, storage)[0]
assert actual_better == better.number
assert actual_worse == worse.number