update

baidubce · Dec 12, 2024 · 4b5ef58 · 4b5ef58
1 parent 2287c41
commit 4b5ef58
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 47 deletions.
diff --git a/python/tests/test_base_component.py b/python/tests/test_base_component.py
@@ -14,6 +14,7 @@ def test_valid_output_with_str(self):
         out2 = self.component.create_output(type="code", text="import appbuilder")
         out3 = self.component.create_output(type="urls", text="http://www.baidu.com")
         out4 = self.component.create_output(type="oral_text", text="你是哪个")
+        out5 = self.component.create_output(type="json", text="{'key':'value'}")
         self.assertIsInstance(out1, ComponentOutput)
         self.assertIsInstance(out2, ComponentOutput)
         self.assertIsInstance(out3, ComponentOutput)
@@ -55,15 +56,7 @@ def test_valid_output_type_with_same_key(self):
 
     def test_invalid_output_type_json(self):
         with self.assertRaises(ValueError):
-            output = self.component.create_output(type="json", text="")
-        # with self.assertRaises(AssertionError):
-        #     output = self.component.create_output(type="files", text={})
-        # with self.assertRaises(AssertionError):
-        #     output = self.component.create_output(type="references", text={"info": "text"})
-        # with self.assertRaises(AssertionError):
-        #     output = self.component.create_output(type="image", text={"url": "https://example.com/img"})
-        # with self.assertRaises(AssertionError):
-        #     output = self.component.create_output(type="chart", text={"url": "https://example.com/chart"})
+            output = self.component.create_output(type="test", text="")
 
 
 if __name__ == '__main__':

diff --git a/python/tests/test_core_components.py b/python/tests/test_core_components.py