mirror of
https://github.com/wassname/optuna-dashboard.git
synced 2026-08-20 12:40:54 +08:00
21 lines
887 B
Python
21 lines
887 B
Python
from __future__ import annotations
|
|
|
|
from unittest import TestCase
|
|
|
|
import optuna
|
|
from optuna_dashboard._preference_setting import _SYSTEM_ATTR_FEEDBACK_COMPONENT
|
|
from optuna_dashboard._preference_setting import register_preference_feedback_component
|
|
from optuna_dashboard.preferential._study import PreferentialStudy
|
|
|
|
|
|
class FeedbackSettingTestCase(TestCase):
|
|
def test_widget_to_dict_from_dict(self) -> None:
|
|
study = PreferentialStudy(optuna.create_study())
|
|
register_preference_feedback_component(study, "artifact", "image_key")
|
|
system_attrs = study._study.system_attrs
|
|
feedback_type = system_attrs.get(_SYSTEM_ATTR_FEEDBACK_COMPONENT, {})
|
|
assert "output_type" in feedback_type
|
|
assert feedback_type["output_type"] == "artifact"
|
|
assert "artifact_key" in feedback_type
|
|
assert feedback_type["artifact_key"] == "image_key"
|