Add tests for generic classes, enums, with statements and for statements

python · JukkaL · Sep 27, 2017 · Aug 31, 2017 · Sep 1, 2017 · Sep 11, 2017
commit 649ef32889cc1783a822233abdc2b8abd45cf360
diff --git a/test-data/unit/check-enum.test b/test-data/unit/check-enum.test
@@ -360,6 +360,24 @@ x = y
 [out]
 main:8: error: Incompatible types in assignment (expression has type "__main__.B.E", variable has type "__main__.A.E")
 
+[case testEnumWorkWithForward]
+from enum import Enum
+a: E = E.x
+class E(Enum):
+    x = 1
+    y = 2
+[out]
+
+[case testEnumWorkWithForward2]
+from enum import Enum
+b: F
+F = Enum('F', {'x': 1, 'y': 2})
+
+def fn(x: F) -> None:
+    pass
+fn(b)
+[out]
+
 [case testFunctionalEnum_python2]
 from enum import Enum
 Eu = Enum(u'Eu', u'a b')

diff --git a/test-data/unit/check-namedtuple.test b/test-data/unit/check-namedtuple.test
@@ -434,6 +434,20 @@ def f() -> None:
     A = NamedTuple('A', [('x', int)])
 A  # E: Name 'A' is not defined
 
+[case testNamedTupleForwardAsUpperBound]
+from typing import NamedTuple, TypeVar, Generic
+T = TypeVar('T', bound='M')
+class G(Generic[T]):
+    x: T
+
+yb: G[int]
+yg: G[M]
+z: int = G[M]().x.x
+z = G[M]().x[0]
+
+M = NamedTuple('M', [('x', int)])
+[out]
+
 [case testNamedTupleWithImportCycle]
 import a
 [file a.py]

diff --git a/test-data/unit/check-statements.test b/test-data/unit/check-statements.test
@@ -1565,3 +1565,28 @@ reveal_type(d['weight0']) # E: Revealed type is 'builtins.float*'
 
 [builtins fixtures/floatdict.pyi]
 
+[case testForwardRefsInForStatement]
+from typing import List, NamedTuple
+lst: List[N]
+
+for i in lst: # type: N
+    a: int = i.x
+    b: str = i[0]
+
+N = NamedTuple('N', [('x', int)])
+[builtins fixtures/list.pyi]
+[out]
+
+[case testForwardRefsInWithStatement]
+from typing import ContextManager
+from mypy_extensions import TypedDict
+cm: ContextManager[N]
+
+with cm as g:  # type: N
+    a: str = g['x']
+
+N = TypedDict('N', {'x': int})
+[builtins fixtures/dict.pyi]
+[typing fixtures/typing-full.pyi]
+[out]
+
diff --git a/test-data/unit/check-typeddict.test b/test-data/unit/check-typeddict.test
@@ -1326,6 +1326,22 @@ reveal_type(m['director']['name']) # E: Revealed type is 'builtins.str'
 [builtins fixtures/dict.pyi]
 [out]
 
+[case testTypedDictForwardAsUpperBound]
+from typing import TypeVar, Generic
+from mypy_extensions import TypedDict
+T = TypeVar('T', bound='M')
+class G(Generic[T]):
+    x: T
+
+yb: G[int]
+yg: G[M]
+z: int = G().x['x']
+
+class M(TypedDict):
+    x: int
+[builtins fixtures/dict.pyi]
+[out]
+
 [case testTypedDictWithImportCycleForward]
 import a
 [file a.py]