Add comment to enum iteration test

ngaya-ll · ngaya-ll · commit 191918e879c9 · 2018-02-06T14:08:11.000-08:00
diff --git a/test-data/unit/pythoneval.test b/test-data/unit/pythoneval.test
@@ -1460,15 +1460,19 @@ _testNoCrashOnGenericUnionUnpacking.py:15: error: Revealed type is 'Union[builti
 _testNoCrashOnGenericUnionUnpacking.py:16: error: Revealed type is 'Union[builtins.int*, builtins.str*]'
 
 [case testEnumIteration]
+# Regression test for #2305
+# Tests that:
+# - Mypy allows iteration over enum classes in comprehensions and for-loops
+# - The inferred element type in such iteration contexts is the enum type
 from enum import Enum
 class E(Enum):
     A = 'a'
 (reveal_type(e) for e in E)
 for e in E:
     reveal_type(e)
 [out]
-_testEnumIteration.py:4: error: Revealed type is '_testEnumIteration.E*'
-_testEnumIteration.py:6: error: Revealed type is '_testEnumIteration.E*'
+_testEnumIteration.py:8: error: Revealed type is '_testEnumIteration.E*'
+_testEnumIteration.py:10: error: Revealed type is '_testEnumIteration.E*'
 
 [case testEnumIterable]
 from enum import Enum
