handle functions

Author: ilinum

mypy/partially_defined.py

@@ -318,7 +318,7 @@ def visit_match_stmt(self, o: MatchStmt) -> None:
         self.tracker.end_branch_statement()
 
     def visit_func_def(self, o: FuncDef) -> None:
-        self.tracker.record_definition(o.name)
+        self.process_definition(o.name)
         self.tracker.enter_scope()
         super().visit_func_def(o)
         self.tracker.exit_scope()

test-data/unit/check-partially-defined.test

@@ -142,12 +142,15 @@ def f0(b: bool) -> None:
 
 [case testUseBeforeDefClass]
 # flags: --enable-error-code partially-defined --enable-error-code use-before-def
-def f(x: A):
+def f(x: A):  # No error here.
     pass
-
 y = A()  # E: Name "A" is used before definition
-
 class A: pass
+
+[case testUseBeforeDefFunc]
+# flags: --enable-error-code partially-defined --enable-error-code use-before-def
+foo() # E: Name "foo" is used before definition
+def foo(): pass
 [case testGenerator]
 # flags: --enable-error-code partially-defined
 if int():