From 8763111aae725a8ff5ce65fbd105daeaaec3da24 Mon Sep 17 00:00:00 2001 From: Jacob Walls Date: Sun, 14 May 2023 12:48:27 -0400 Subject: [PATCH] Handle ``objects.Super`` in `helpers.object_type()` (#2177) (cherry picked from commit b186f683da0896d2fbed0f2aae3497b29ca93266) --- ChangeLog | 3 +++ astroid/helpers.py | 4 ++-- tests/test_helpers.py | 1 + 3 files changed, 6 insertions(+), 2 deletions(-) diff --git a/ChangeLog b/ChangeLog index 993426bd98..0308eda295 100644 --- a/ChangeLog +++ b/ChangeLog @@ -12,6 +12,9 @@ What's New in astroid 2.15.5? ============================= Release date: TBA +* Handle ``objects.Super`` in ``helpers.object_type()``. + + Refs pylint-dev/pylint#8554 What's New in astroid 2.15.4? diff --git a/astroid/helpers.py b/astroid/helpers.py index 24dba6d7bc..bb19bbf4bd 100644 --- a/astroid/helpers.py +++ b/astroid/helpers.py @@ -8,7 +8,7 @@ from collections.abc import Generator -from astroid import bases, manager, nodes, raw_building, util +from astroid import bases, manager, nodes, objects, raw_building, util from astroid.context import CallContext, InferenceContext from astroid.exceptions import ( AstroidTypeError, @@ -65,7 +65,7 @@ def _object_type( raise InferenceError elif isinstance(inferred, util.UninferableBase): yield inferred - elif isinstance(inferred, (bases.Proxy, nodes.Slice)): + elif isinstance(inferred, (bases.Proxy, nodes.Slice, objects.Super)): yield inferred._proxied else: # pragma: no cover raise AssertionError(f"We don't handle {type(inferred)} currently") diff --git a/tests/test_helpers.py b/tests/test_helpers.py index fe97eb6466..90182a23ee 100644 --- a/tests/test_helpers.py +++ b/tests/test_helpers.py @@ -42,6 +42,7 @@ def test_object_type(self) -> None: ("type", self._extract("type")), ("object", self._extract("type")), ("object()", self._extract("object")), + ("super()", self._extract("super")), ("lambda: None", self._build_custom_builtin("function")), ("len", self._build_custom_builtin("builtin_function_or_method")), ("None", self._build_custom_builtin("NoneType")),