artiq/test/{not,harness}.py: usual CLI handling

artiq/test/harness.py

@@ -11,21 +11,25 @@ emulate the same behavior when invoked under lit.
 
 import sys, os, argparse, importlib
 
-parser = argparse.ArgumentParser(description=__doc__)
-parser.add_argument('-m', metavar='mod', type=str,
-                    help='run library module as a script')
-parser.add_argument('args', type=str, nargs='+',
-                    help='arguments passed to program in sys.argv[1:]')
-args = parser.parse_args(sys.argv[1:])
+def main():
+    parser = argparse.ArgumentParser(description=__doc__)
+    parser.add_argument('-m', metavar='mod', type=str,
+                        help='run library module as a script')
+    parser.add_argument('args', type=str, nargs='+',
+                        help='arguments passed to program in sys.argv[1:]')
+    args = parser.parse_args(sys.argv[1:])
 
-artiq_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
-sys.path.insert(1, artiq_path)
+    artiq_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
+    sys.path.insert(1, artiq_path)
 
-if args.m:
-    sys.argv[1:] = args.args
-    importlib.import_module(args.m).main()
-else:
-    sys.argv[1:] = args.args[1:]
-    with open(args.args[0]) as f:
-        code = compile(f.read(), args.args[0], 'exec')
-        exec(code)
+    if args.m:
+        sys.argv[1:] = args.args
+        importlib.import_module(args.m).main()
+    else:
+        sys.argv[1:] = args.args[1:]
+        with open(args.args[0]) as f:
+            code = compile(f.read(), args.args[0], 'exec')
+            exec(code)
+
+if __name__ == "__main__":
+    main()

artiq/test/not.py

@@ -1,2 +1,7 @@
 import sys, subprocess
-exit(not subprocess.call(sys.argv[1:]))
+
+def main():
+    exit(not subprocess.call(sys.argv[1:]))
+
+if __name__ == "__main__":
+    main()