Tests private cache for dataclass forward-refs

erezsh · Oct 9, 2022 · 533541d · 533541d
1 parent 2f63e27
commit 533541d
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/tests/test_basic.py b/tests/test_basic.py
@@ -651,6 +651,27 @@ class Tree:
         t = Tree('a', [t])
         self.assertRaises(TypeError, Tree, 'a', [1])
 
+    def test_forward_reference_cache(self):
+        @dataclass
+        class A:
+            b: 'B'
+
+        class B:
+            pass
+
+        a = A(B())
+        self.assertRaises(TypeError, A, 1)
+
+        @dataclass
+        class A:
+            b: 'B'
+
+        class B:
+            pass
+
+        a = A(B())
+        self.assertRaises(TypeError, A, 1)
+
     def test_unfrozen(self):
         @dataclass(frozen=False, slots=False)
         class A: