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