Wrap AttributeInferenceErrors in AstroidTypeErrors for ClassDef.getitem.

pylint-dev · Apr 17, 2017 · b6cf7ab · b6cf7ab
1 parent 5092504
commit b6cf7ab
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 1 deletion.
diff --git a/astroid/scoped_nodes.py b/astroid/scoped_nodes.py
@@ -1520,7 +1520,15 @@ def getitem(self, index, context=None):
 
         This is basically looking up the method in the metaclass and calling it.
         """
-        methods = dunder_lookup.lookup(self, '__getitem__')
+        try:
+            methods = dunder_lookup.lookup(self, '__getitem__')
+        except exceptions.AttributeInferenceError as exc:
+            util.reraise(
+                exceptions.AstroidTypeError(
+                    node=self, error=exc,
+                    context=context
+                )
+            )
 
         method = methods[0]
 

diff --git a/astroid/tests/unittest_inference.py b/astroid/tests/unittest_inference.py
@@ -3470,6 +3470,18 @@ def blurb(self):
         self.assertIsInstance(inferred, nodes.Const)
         self.assertEqual(inferred.value, 25)
 
+    def test_getitem_of_class_raised_type_error(self):
+        # Test that we wrap an AttributeInferenceError
+        # and reraise it as a TypeError in Class.getitem
+        node = extract_node('''
+        def test():
+            yield
+        test()
+        ''')
+        inferred = next(node.infer())
+        with self.assertRaises(exceptions.AstroidTypeError):
+            inferred.getitem(nodes.Const('4'))
+
 
 class GetattrTest(unittest.TestCase):