Add GO

execute.py

@@ -42,6 +42,8 @@ def check_correctness(check_program, timeout, task_id, completion_id, language):
         p = multiprocessing.Process(target=unsafe_execute, args=(check_program, result, timeout))
     elif language == "cpp":
         p = multiprocessing.Process(target=unsafe_execute_cpp, args=(check_program, result, timeout))
+    elif language == "go":
+        p = multiprocessing.Process(target=unsafe_execute_go, args=(check_program, result, timeout))
     elif language == "java":
         p = multiprocessing.Process(target=unsafe_execute_java, args=(check_program, result, timeout))
     elif language == "javascript":
@@ -136,6 +138,33 @@ def unsafe_execute_cpp(check_program, result, timeout):
             except subprocess.TimeoutExpired as e:
                 result.append("timed out")            
 
+
+def unsafe_execute_go(check_program, result, timeout):
+
+    with create_tempdir():
+        open(f"main_test.go", 'w').write(check_program)
+
+        try:
+            exec_result = subprocess.run(["go", "test", f"-timeout={timeout}s", "main_test.go"], timeout=timeout, capture_output=True)
+
+            if exec_result.returncode == 0:
+                result.append("passed")
+            else:
+                if exec_result.stderr:
+                    try:
+                        err = exec_result.stderr.decode()
+                    except:
+                        err = exec_result.stderr
+                else:
+                    try:
+                        err = exec_result.stdout.decode()
+                    except:
+                        err = exec_result.stdout
+                result.append(f"failed: {err}")
+        except subprocess.TimeoutExpired as e:
+            result.append("timed out")
+
+
 def unsafe_execute_java(check_program, result, timeout):
 
     with create_tempdir():
@@ -178,7 +207,6 @@ def unsafe_execute_js(check_program, result, timeout):
 
         # Run program.
         try:
-            exec_globals = {}
             exec_result = subprocess.run(["node", "test.js"], timeout=timeout, capture_output=True)
             if exec_result.stderr.decode():
                 err = exec_result.stderr.decode()