fix by pytest error

This commit is contained in:
moririn2528
2023-09-15 17:51:05 +09:00
parent 71ada285e7
commit 9093f845fa
2 changed files with 3 additions and 3 deletions
+1 -1
View File
@@ -211,7 +211,7 @@ class APITestCase(TestCase):
self.assertEqual(status, 200)
study_detail = json.loads(body)
assert study_detail["feedback_component_type"]["type"] == "artifact"
assert study_detail["feedback_component_type"]["output_type"] == "artifact"
assert study_detail["feedback_component_type"]["artifact_key"] == "image"
def test_skip_trial(self) -> None:
+2 -2
View File
@@ -14,7 +14,7 @@ class FeedbackSettingTestCase(TestCase):
register_preference_feedback_component(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 "output_type" in feedback_type
assert feedback_type["output_type"] == "artifact"
assert "artifact_key" in feedback_type
assert feedback_type["artifact_key"] == "image_key"