Add lit-based tests for diagnostics.

whitequark · whitequark · commit 98fe15215982 · 2015-06-06T17:15:06.000+03:00
diff --git a/artiq/py2llvm/typing.py b/artiq/py2llvm/typing.py
@@ -358,7 +358,11 @@ def generic_visit(self, node):
 def main():
     import sys, fileinput, os
 
-    inference_mode = True
+    if sys.argv[1] == '+diag':
+        del sys.argv[1]
+        inference_mode = False
+    else:
+        inference_mode = True
 
     engine = diagnostic.Engine(all_errors_are_fatal=True)
     try:
@@ -375,6 +379,8 @@ def main():
         if inference_mode:
             print("\n".join(e.diagnostic.render()), file=sys.stderr)
             exit(1)
+        else:
+            print("\n".join(e.diagnostic.render(only_line=True)))
 
 if __name__ == "__main__":
     main()
diff --git a/lit-test/harness.py b/lit-test/harness.py
@@ -12,8 +12,10 @@
 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:]')
+parser.add_argument('-m', metavar='mod', type=str, required=True,
+                    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__), '..'))
diff --git a/lit-test/py2llvm/typing/error_unify.py b/lit-test/py2llvm/typing/error_unify.py
@@ -0,0 +1,8 @@
+# RUN: %python -m artiq.py2llvm.typing +diag %s >%t
+# RUN: OutputCheck %s --file-to-check=%t
+
+a = 1
+b = []
+
+# CHECK-L: ${LINE:+1}: fatal: cannot unify int(width='a) with list(elt='b)
+a = b
