test subclass of EnumMeta-based class

elazarg · elazarg · commit 3c16eead5930 · 2018-02-26T00:36:37.000+02:00
diff --git a/test-data/unit/check-enum.test b/test-data/unit/check-enum.test
@@ -20,6 +20,17 @@ reveal_type(Medal.bronze)  # E: Revealed type is '__main__.Medal'
 m = Medal.gold
 m = 1  # E: Incompatible types in assignment (expression has type "int", variable has type "Medal")
 
+[case testEnumFromEnumMetaSubclass]
+from enum import EnumMeta
+class Achievement(metaclass=EnumMeta): pass
+class Medal(Achievement):
+    gold = 1
+    silver = "hello"
+    bronze = None
+reveal_type(Medal.bronze)  # E: Revealed type is '__main__.Medal'
+m = Medal.gold
+m = 1  # E: Incompatible types in assignment (expression has type "int", variable has type "Medal")
+
 [case testEnumNameAndValue]
 from enum import Enum
 class Truth(Enum):
