Add test for circular types in enum instance attributes.

PiperOrigin-RevId: 407173206
google · Nov 2, 2021 · 03df5c3 · 03df5c3
1 parent 6e33205
commit 03df5c3
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/pytype/tests/test_enums.py b/pytype/tests/test_enums.py
@@ -1130,6 +1130,23 @@ def take_m(m: M):
           return m.x
       """, pythonpath=[d.path])
 
+  def test_instance_attrs_self_referential(self):
+    self.Check("""
+      from dataclasses import dataclass
+      from enum import Enum
+      from typing import Optional
+
+      @dataclass
+      class O:
+        thing: Optional["Thing"] = None
+
+      class Thing(Enum):
+        A = O()
+
+        def __init__(self, o: O):
+          self.b = o.thing
+    """)
+
   def test_enum_bases(self):
     self.CheckWithErrors("""
       import enum