fix by review

This commit is contained in:
moririn2528
2023-09-12 17:54:42 +09:00
parent fffa586da3
commit 702b35622e
4 changed files with 19 additions and 13 deletions
+2 -2
View File
@@ -11,10 +11,10 @@ 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_type(study, "Artifact", "image_key")
register_preference_feedback_component_type(study, "artifact", "image_key")
system_attrs = study._study.system_attrs
feedback_type = system_attrs.get(_SYSTEM_ATTR_FEEDBACK_COMPONENT, {})
assert "type" in feedback_type
assert feedback_type["type"] == "Artifact"
assert feedback_type["type"] == "artifact"
assert "artifact_key" in feedback_type
assert feedback_type["artifact_key"] == "image_key"