refactor: move get_real_func to fixtures.py

Author: Glyphack

src/_pytest/_code/code.py

@@ -45,8 +45,8 @@
 from _pytest._io import TerminalWriter
 from _pytest._io.saferepr import safeformat
 from _pytest._io.saferepr import saferepr
-from _pytest.compat import get_real_func
 from _pytest.deprecated import check_ispytest
+from _pytest.fixtures import get_real_func
 from _pytest.pathlib import absolutepath
 from _pytest.pathlib import bestrelpath
 

src/_pytest/compat.py

@@ -4,12 +4,10 @@
 from __future__ import annotations
 
 import enum
-import functools
 import inspect
 from inspect import Parameter
 from inspect import signature
 import os
-from pathlib import Path
 import sys
 from typing import Any
 from typing import Callable
@@ -65,20 +63,6 @@ def is_async_function(func: object) -> bool:
     return iscoroutinefunction(func) or inspect.isasyncgenfunction(func)
 
 
-def getlocation(function, curdir: str | os.PathLike[str] | None = None) -> str:
-    function = get_real_func(function)
-    fn = Path(inspect.getfile(function))
-    lineno = function.__code__.co_firstlineno
-    if curdir is not None:
-        try:
-            relfn = fn.relative_to(curdir)
-        except ValueError:
-            pass
-        else:
-            return "%s:%d" % (relfn, lineno + 1)
-    return "%s:%d" % (fn, lineno + 1)
-
-
 def num_mock_patch_args(function) -> int:
     """Return number of arguments used up by mock arguments (if any)."""
     patchings = getattr(function, "patchings", None)
@@ -209,31 +193,6 @@ def ascii_escaped(val: bytes | str) -> str:
     return ret.translate(_non_printable_ascii_translate_table)
 
 
-def get_real_func(obj):
-    """Get the real function object of the (possibly) wrapped object by
-    :func:`functools.wraps`, or :func:`functools.partial`, or :func:`pytest.fixture`."""
-    from _pytest.fixtures import FixtureFunctionDefinition
-
-    start_obj = obj
-    for _ in range(100):
-        if isinstance(obj, FixtureFunctionDefinition):
-            obj = obj._get_wrapped_function()
-            break
-        new_obj = getattr(obj, "__wrapped__", None)
-        if new_obj is None:
-            break
-        obj = new_obj
-    else:
-        from _pytest._io.saferepr import saferepr
-
-        raise ValueError(
-            f"could not find real function of {saferepr(start_obj)}\nstopped at {saferepr(obj)}"
-        )
-    if isinstance(obj, functools.partial):
-        obj = obj.func
-    return obj
-
-
 def getimfunc(func):
     try:
         return func.__func__

src/_pytest/fixtures.py

@@ -43,10 +43,7 @@
 from _pytest._code.code import TerminalRepr
 from _pytest._io import TerminalWriter
 from _pytest.compat import assert_never
-from _pytest.compat import get_real_func
-from _pytest.compat import getfuncargnames
 from _pytest.compat import getimfunc
-from _pytest.compat import getlocation
 from _pytest.compat import is_generator
 from _pytest.compat import NOTSET
 from _pytest.compat import NotSetType
@@ -58,6 +55,7 @@
 from _pytest.deprecated import check_ispytest
 from _pytest.deprecated import MARKED_FIXTURE
 from _pytest.deprecated import YIELD_FIXTURE
+from _pytest.fixtures import getfuncargnames
 from _pytest.main import Session
 from _pytest.mark import Mark
 from _pytest.mark import ParameterSet
@@ -1914,3 +1912,40 @@ def _showfixtures_main(config: Config, session: Session) -> None:
 def write_docstring(tw: TerminalWriter, doc: str, indent: str = "    ") -> None:
     for line in doc.split("\n"):
         tw.line(indent + line)
+
+
+def get_real_func(obj):
+    """Get the real function object of the (possibly) wrapped object by
+    :func:`functools.wraps`, or :func:`functools.partial`, or :func:`pytest.fixture`."""
+    start_obj = obj
+    for _ in range(100):
+        if isinstance(obj, FixtureFunctionDefinition):
+            obj = obj._get_wrapped_function()
+            break
+        new_obj = getattr(obj, "__wrapped__", None)
+        if new_obj is None:
+            break
+        obj = new_obj
+    else:
+        from _pytest._io.saferepr import saferepr
+
+        raise ValueError(
+            f"could not find real function of {saferepr(start_obj)}\nstopped at {saferepr(obj)}"
+        )
+    if isinstance(obj, functools.partial):
+        obj = obj.func
+    return obj
+
+
+def getlocation(function, curdir: str | os.PathLike[str] | None = None) -> str:
+    function = get_real_func(function)
+    fn = Path(inspect.getfile(function))
+    lineno = function.__code__.co_firstlineno
+    if curdir is not None:
+        try:
+            relfn = fn.relative_to(curdir)
+        except ValueError:
+            pass
+        else:
+            return "%s:%d" % (relfn, lineno + 1)
+    return "%s:%d" % (fn, lineno + 1)

src/_pytest/python.py

@@ -40,7 +40,6 @@
 from _pytest._io.saferepr import saferepr
 from _pytest.compat import ascii_escaped
 from _pytest.compat import get_default_arg_names
-from _pytest.compat import get_real_func
 from _pytest.compat import getimfunc
 from _pytest.compat import is_async_function
 from _pytest.compat import is_generator
@@ -55,6 +54,7 @@
 from _pytest.fixtures import FixtureDef
 from _pytest.fixtures import FixtureRequest
 from _pytest.fixtures import FuncFixtureInfo
+from _pytest.fixtures import get_real_func
 from _pytest.fixtures import get_scope_node
 from _pytest.main import Session
 from _pytest.mark import MARK_GEN

testing/code/test_source.py

@@ -464,7 +464,7 @@ def test_comment_in_statement() -> None:
 
 def test_source_with_decorator() -> None:
     """Test behavior with Source / Code().source with regard to decorators."""
-    from _pytest.compat import get_real_func
+    from _pytest.fixtures import get_real_func
 
     @pytest.mark.foo
     def deco_mark():

testing/test_compat.py

@@ -9,10 +9,10 @@
 from typing import TYPE_CHECKING
 
 from _pytest.compat import assert_never
-from _pytest.compat import get_real_func
 from _pytest.compat import is_generator
 from _pytest.compat import safe_getattr
 from _pytest.compat import safe_isclass
+from _pytest.fixtures import get_real_func
 from _pytest.outcomes import OutcomeException
 from _pytest.pytester import Pytester
 import pytest