Improve typing.TypedDict inference

1 files changed, 15 insertions, 0 deletions

diff --git a/tests/unittest_brain.py b/tests/unittest_brain.py
index dc3f25d4..51311972 100644
--- a/tests/unittest_brain.py
+++ b/tests/unittest_brain.py
@@ -1188,6 +1188,21 @@ class TypingBrain(unittest.TestCase):
         inferred = next(node.infer())
         self.assertIsInstance(inferred, astroid.Instance)
 
+    def test_typedDict(self):
+        node = builder.extract_node(
+            """
+        from typing import TypedDict
+        class CustomTD(TypedDict):
+            var: int
+        """
+        )
+        assert len(node.bases) == 1
+        inferred_base = next(node.bases[0].infer())
+        self.assertIsInstance(inferred_base, nodes.ClassDef, node.as_string())
+        typing_module = inferred_base.root()
+        assert len(typing_module.locals["TypedDict"]) == 1
+        assert inferred_base == typing_module.locals["TypedDict"][0]
+
 
 class ReBrainTest(unittest.TestCase):
     def test_regex_flags(self):