teststubtest: add a test for the previous commit

Also covers some of the MRO logic in verify_typeinfo
python · May 25, 2020 · c2279e3 · c2279e3
1 parent 4d30a84
commit c2279e3
Showing 1 changed file with 32 additions and 0 deletions.
diff --git a/mypy/test/teststubtest.py b/mypy/test/teststubtest.py
@@ -588,6 +588,38 @@ def __mangle_bad(self, text): pass
             error="X.__mangle_bad"
         )
 
+    @collect_cases
+    def test_mro(self) -> Iterator[Case]:
+        yield Case(
+            stub="""
+            class A:
+                def foo(self, x: int) -> None: ...
+            class B(A):
+                pass
+            class C(A):
+                pass
+            """,
+            runtime="""
+            class A:
+                def foo(self, x: int) -> None: ...
+            class B(A):
+                def foo(self, x: int) -> None: ...
+            class C(A):
+                def foo(self, y: int) -> None: ...
+            """,
+            error="C.foo"
+        )
+        yield Case(
+            stub="""
+            class X: ...
+            """,
+            runtime="""
+            class X:
+                def __init__(self, x): pass
+            """,
+            error="X.__init__"
+        )
+
 
 def remove_color_code(s: str) -> str:
     return re.sub("\\x1b.*?m", "", s)  # this works!