Allow passing UninferableBase to safe_infer

pylint-dev · May 1, 2023 · faea711 · faea711
1 parent 4fe752e
commit faea711
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/astroid/helpers.py b/astroid/helpers.py
@@ -156,13 +156,16 @@ def object_issubclass(node, class_or_seq, context: InferenceContext | None = Non
 
 
 def safe_infer(
-    node: nodes.NodeNG | bases.Proxy, context: InferenceContext | None = None
+    node: nodes.NodeNG | bases.Proxy | util.UninferableBase,
+    context: InferenceContext | None = None,
 ) -> InferenceResult | None:
     """Return the inferred value for the given node.
 
     Return None if inference failed or if there is some ambiguity (more than
     one node has been inferred).
     """
+    if isinstance(node, util.UninferableBase):
+        return node
     try:
         inferit = node.infer(context=context)
         value = next(inferit)

diff --git a/tests/test_helpers.py b/tests/test_helpers.py
@@ -258,3 +258,8 @@ class A(type): #@
         builtin_type = self._extract("type")
         self.assertTrue(helpers.is_supertype(builtin_type, cls_a))
         self.assertTrue(helpers.is_subtype(cls_a, builtin_type))
+
+
+def test_uninferable_for_safe_infer() -> None:
+    uninfer = util.Uninferable
+    assert helpers.safe_infer(util.Uninferable) == uninfer