1 files changed, 24 insertions, 0 deletions
diff --git a/llvm/utils/lit/tests/Inputs/shtest-timeout/quick_then_slow.py b/llvm/utils/lit/tests/Inputs/shtest-timeout/quick_then_slow.py
new file mode 100644
index 00000000000..b81fbe5a8bf
--- /dev/null
+++ b/llvm/utils/lit/tests/Inputs/shtest-timeout/quick_then_slow.py
@@ -0,0 +1,24 @@
+# RUN: %{python} %s quick
+# RUN: %{python} %s slow
+from __future__ import print_function
+
+import time
+import sys
+
+if len(sys.argv) != 2:
+    print("Wrong number of args")
+    sys.exit(1)
+
+mode =  sys.argv[1]
+
+if mode == 'slow':
+    print("Running in slow mode")
+    sys.stdout.flush() # Make sure the print gets flushed so it appears in lit output.
+    time.sleep(6)
+    sys.exit(0)
+elif mode == 'quick':
+    print("Running in quick mode")
+    sys.exit(0)
+else:
+    print("Unrecognised mode {}".format(mode))
+    sys.exit(1)