Add unit tests

Author: ddfisher

mypy/test/data/check-flags.test

@@ -0,0 +1,32 @@
+# test cases for --disallow-untyped-defs
+
+[case testUnannotatedFunction]
+# flags: disallow-untyped-defs
+def f(x): pass
+[out]
+main: note: In function "f":
+main:2: error: Function is missing a type annotation
+
+[case testUnannotatedArgument]
+# flags: disallow-untyped-defs
+def f(x) -> int: pass
+[out]
+main: note: In function "f":
+main:2: error: Function is missing a type annotation for one or more arguments
+
+[case testNoArgumentFunction]
+# flags: disallow-untyped-defs
+def f() -> int: pass
+[out]
+
+[case testUnannotatedReturn]
+# flags: disallow-untyped-defs
+def f(x: int): pass
+[out]
+main: note: In function "f":
+main:2: error: Function is missing a return type annotation
+
+[case testLambda]
+# flags: disallow-untyped-defs
+lambda x: x
+[out]

mypy/test/data/lib-stub/builtins.py

@@ -1,8 +1,10 @@
+class Any: pass
+
 class object:
     def __init__(self) -> None: pass
 
 class type:
-    def __init__(self, x) -> None: pass
+    def __init__(self, x: Any) -> None: pass
 
 # These are provided here for convenience.
 class int: pass

mypy/test/testcheck.py

@@ -4,7 +4,7 @@
 import re
 import sys
 
-from typing import Tuple
+from typing import Tuple, List
 
 from mypy import build
 import mypy.myunit  # for mutable globals (ick!)
@@ -53,6 +53,7 @@
     'check-ignore.test',
     'check-type-promotion.test',
     'check-semanal-error.test',
+    'check-flags.test',
 ]
 
 
@@ -69,12 +70,13 @@ def run_test(self, testcase):
         pyversion = testcase_pyversion(testcase.file, testcase.name)
         program_text = '\n'.join(testcase.input)
         module_name, program_name, program_text = self.parse_options(program_text)
+        flags = self.parse_flags(program_text)
         source = BuildSource(program_name, module_name, program_text)
         try:
             build.build(target=build.TYPE_CHECK,
                         sources=[source],
                         pyversion=pyversion,
-                        flags=[build.TEST_BUILTINS],
+                        flags=flags + [build.TEST_BUILTINS],
                         alt_lib_path=test_temp_dir)
         except CompileError as e:
             a = normalize_error_messages(e.messages)
@@ -109,3 +111,10 @@ def parse_options(self, program_text: str) -> Tuple[str, str, str]:
             return m.group(1), path, program_text
         else:
             return '__main__', 'main', program_text
+
+    def parse_flags(self, program_text: str) -> List[str]:
+        m = re.search('# flags: (.*)$', program_text, flags=re.MULTILINE)
+        if m:
+            return m.group(1).split()
+        else:
+            return []