1 files changed, 16 insertions, 0 deletions
diff --git a/lit/Quit/expect_exit_code.py b/lit/Quit/expect_exit_code.py
new file mode 100755
index 000000000000..f4a7590b7d52
--- /dev/null
+++ b/lit/Quit/expect_exit_code.py
@@ -0,0 +1,16 @@
+#!/usr/bin/env python2
+
+import subprocess
+import sys
+
+args = sys.argv
+
+expected_exit_code = args[1]
+
+args = args[2:]
+print("Running " + (" ".join(args)))
+real_exit_code = subprocess.call(args)
+
+if str(real_exit_code) != expected_exit_code:
+  print("Got exit code %d but expected %s" % (real_exit_code, expected_exit_code))
+  exit(1)