Add test for line and column end for missing function type annotation

mschoettle · mschoettle · commit 5f88e5ddf560 · 2024-03-09T10:57:12.000-05:00
diff --git a/test-data/unit/check-columns.test b/test-data/unit/check-columns.test
@@ -178,6 +178,18 @@ if int():
     def g(x): # E:5: Function is missing a type annotation
         pass
 
+[case testColumnEndFunctionMissingTypeAnnotation]
+# flags: --disallow-untyped-defs --show-error-end
+if int():
+    def f(x: int):
+        pass
+
+    def g(x):
+        pass
+[out]
+main:3:5:3:15: error: Function is missing a return type annotation
+main:6:5:6:10: error: Function is missing a type annotation
+
 [case testColumnNameIsNotDefined]
 ((x)) # E:3: Name "x" is not defined
 
