misc: colored output

bin/ac-random-test

@@ -3,6 +3,7 @@ import argparse
 import os
 import shutil
 import subprocess
+import sys
 
 """
 Example of gen-input-program:
@@ -18,9 +19,18 @@ with open("in.txt", "w") as f:
 ```
 """
 
+class bcolors:
+    OKBLUE = '\033[94m'
+    OKGREEN = '\033[92m'
+    FAIL = '\033[91m'
+    ENDC = '\033[0m'
+    BOLD = '\033[1m'
+
+devnull = open(os.devnull, "w")
+
 
 def run_and_decode(args: list[str]) -> str:
-    ret = subprocess.check_output(args, stdin=open("in.txt"))
+    ret = subprocess.check_output(args, stdin=open("in.txt"), stderr=devnull)
     return ret.decode("utf-8").rstrip()
 
 
@@ -31,15 +41,23 @@ def verify(args: argparse.Namespace):
         gen_input_cmd = args.gen_input_cmd
     else:
         gen_input_cmd = [os.path.abspath(args.gen_input_cmd)]
-    subprocess.check_call(gen_input_cmd)
+    subprocess.check_call(gen_input_cmd, stdout=devnull, stderr=devnull)
 
     run_cmd: list[str] = args.run_cmd.split(" ")
     ret_cmd = [s.format(argv0=name) for s in run_cmd]
     ret_lazy_cmd = [s.format(argv0=f"{name}_lazy") for s in run_cmd]
 
     ret = run_and_decode(ret_cmd)
     ret_lazy = run_and_decode(ret_lazy_cmd)
-    assert ret_lazy == ret, f"lazy={ret_lazy}, ans={ret}"
+
+    if ret_lazy == ret:
+        print("Check: " + bcolors.OKGREEN + "passed" + bcolors.ENDC)
+    else:
+        print("Check: " + bcolors.FAIL + "failed" + bcolors.ENDC)
+        print("Found the edge case and stored in in.txt.", end="")
+        print(" (output: expected=" + bcolors.OKGREEN + ret_lazy + bcolors.ENDC, end="")
+        print(", got=" + bcolors.FAIL + ret + bcolors.ENDC + ")")
+        sys.exit(1)
 
 
 if __name__ == "__main__":