Upload code_executor.py

code_executor.py

@@ -0,0 +1,25 @@
+
+import io
+import sys
+import contextlib
+import traceback
+
+class CodeExecutor:
+    def __init__(self):
+        pass
+
+    def run_code(self, code: str):
+        stdout = io.StringIO()
+        stderr = io.StringIO()
+        local_vars = {}
+
+        try:
+            with contextlib.redirect_stdout(stdout), contextlib.redirect_stderr(stderr):
+                exec(code, {}, local_vars)
+            return {"output": stdout.getvalue(), "error": stderr.getvalue()}
+        except Exception:
+            return {"output": stdout.getvalue(), "error": traceback.format_exc()}
+
+    def test_function(self, code: str, test_code: str):
+        combined = code + "\n" + test_code
+        return self.run_code(combined)