Add tests for __init__ modules

python · gvanrossum · Apr 5, 2017 · Jan 20, 2017 · Feb 9, 2017 · Feb 10, 2017
commit 82afe2267e4127098ff62832c16f5f3f5f7f30a4
diff --git a/mypy/test/testfinegrained.py b/mypy/test/testfinegrained.py
@@ -109,6 +109,7 @@ def find_steps() -> List[List[Tuple[str, str]]]:
                 assert num >= 2
                 name = re.sub(r'\.py.*', '', filename)
                 module = '.'.join(dnparts + [name])
+                module = re.sub(r'\.__init__$', '', module)
                 path = os.path.join(dn, filename)
                 steps.setdefault(num, []).append((module, path))
     max_step = max(steps)

diff --git a/test-data/unit/fine-grained.test b/test-data/unit/fine-grained.test
@@ -527,3 +527,33 @@ def g(x: int) -> None: pass
 [out]
 ==
 m/n.py:3: error: Too few arguments for "g"
+
+[case testChangeInPackage__init__]
+import m
+import m.n
+def f() -> None:
+    m.g()
+[file m/__init__.py]
+def g() -> None: pass
+[file m/__init__.py.2]
+def g(x: int) -> None: pass
+[file m/n.py]
+[out]
+==
+main:3: error: Too few arguments for "g"
+
+[case testTriggerTargetInPackage__init__]
+import m
+import m.n
+[file m/__init__.py]
+import a
+def f() -> None:
+    a.g()
+[file a.py]
+def g() -> None: pass
+[file a.py.2]
+def g(x: int) -> None: pass
+[file m/n.py]
+[out]
+==
+m/__init__.py:3: error: Too few arguments for "g"