init

AlgoriAgent/tests/execute_python_code_test.py

@@ -0,0 +1,90 @@
+# -*- coding: utf-8 -*-
+""" Python code execution test."""
+import unittest
+import sys
+
+from agentscope.service import execute_python_code
+
+
+class ExecutePythonCodeTest(unittest.TestCase):
+    """
+    Python code execution test.
+    """
+
+    def setUp(self) -> None:
+        """Init for ExecutePythonCodeTest."""
+
+        # Basic expression
+        self.arg0 = {"code": "print('Hello World')", "timeout": 5}
+
+        # Raising Exceptions
+        self.arg1 = {
+            "code": "print('Hello World')\nraise ValueError('An intentional "
+            "error')",
+            "timeout": 5,
+        }
+
+        # Using External Libraries
+        self.arg2 = {"code": "import math\nprint(math.sqrt(16))", "timeout": 5}
+
+        # Timeout
+        self.arg3 = {
+            "code": "import time\nprint('Hello World')\ntime.sleep("
+            "10)\nprint('This will not print')",
+            "timeout": 1,
+        }
+
+        # No input code
+        self.arg4 = {"code": "", "timeout": 5}
+
+    def run_test(
+        self,
+        args: dict,
+        expected_output: str,
+        expected_error_substr: str,
+    ) -> None:
+        """A helper function to avoid code repetition"""
+        response = execute_python_code(
+            use_docker=False,
+            **args,
+        )
+        self.assertIn(expected_output, response.content)
+        self.assertIn(expected_error_substr, response.content)
+
+        # Uncomment it when test in local
+        # response = execute_python_code(
+        #     use_docker=True,
+        #     **args,
+        # )
+        # self.assertIn(expected_output, response.content)
+        # self.assertIn(expected_error_substr, response.content)
+
+    def test_basic_expression(self) -> None:
+        """Execute basic expression test."""
+        self.run_test(self.arg0, "Hello World\n", "")
+
+    def test_raising_exceptions(self) -> None:
+        """Execute raising exceptions test."""
+        self.run_test(
+            self.arg1,
+            "Hello World\n",
+            "ValueError: An intentional error\n",
+        )
+
+    def test_using_external_libs(self) -> None:
+        """Execute using external libs test."""
+        self.run_test(self.arg2, "4.0\n", "")
+
+    def test_timeout(self) -> None:
+        """Execute timeout test (NOT available in WinOS.)"""
+        if sys.platform == "win32":
+            return
+        self.run_test(self.arg3, "Hello World\n", "timed out\n")
+
+    def test_no_input_code(self) -> None:
+        """Execute no input code test."""
+        self.run_test(self.arg4, "", "")
+
+
+if __name__ == "__main__":
+    unittest.main()