chore: improve reliability of perf trampoline compatibility checks

Author: art049

src/pytest_codspeed/instruments/hooks/__init__.py

@@ -5,11 +5,11 @@
 import warnings
 from typing import TYPE_CHECKING
 
+from pytest_codspeed.utils import SUPPORTS_PERF_TRAMPOLINE
+
 if TYPE_CHECKING:
     from .dist_instrument_hooks import InstrumentHooksPointer, LibType
 
-SUPPORTS_PERF_TRAMPOLINE = sys.version_info >= (3, 12)
-
 
 class InstrumentHooks:
     """Zig library wrapper class providing benchmark measurement functionality."""

src/pytest_codspeed/instruments/valgrind.py

@@ -1,12 +1,12 @@
 from __future__ import annotations
 
-import sys
 import warnings
 from typing import TYPE_CHECKING
 
 from pytest_codspeed import __semver_version__
 from pytest_codspeed.instruments import Instrument
 from pytest_codspeed.instruments.hooks import InstrumentHooks
+from pytest_codspeed.utils import SUPPORTS_PERF_TRAMPOLINE
 
 if TYPE_CHECKING:
     from typing import Any, Callable
@@ -17,8 +17,6 @@
     from pytest_codspeed.instruments import T
     from pytest_codspeed.plugin import BenchmarkMarkerOptions, CodSpeedConfig
 
-SUPPORTS_PERF_TRAMPOLINE = sys.version_info >= (3, 12)
-
 
 class ValgrindInstrument(Instrument):
     instrument = "valgrind"

src/pytest_codspeed/instruments/walltime.py

@@ -16,6 +16,7 @@
 from pytest_codspeed import __semver_version__
 from pytest_codspeed.instruments import Instrument
 from pytest_codspeed.instruments.hooks import InstrumentHooks
+from pytest_codspeed.utils import SUPPORTS_PERF_TRAMPOLINE
 
 if TYPE_CHECKING:
     from typing import Any, Callable
@@ -173,7 +174,13 @@ def __init__(self, config: CodSpeedConfig) -> None:
         self.benchmarks: list[Benchmark] = []
 
     def get_instrument_config_str_and_warns(self) -> tuple[str, list[str]]:
-        return f"mode: walltime, timer_resolution: {TIMER_RESOLUTION_NS:.1f}ns", []
+        config_str = (
+            f"mode: walltime, "
+            f"callgraph: "
+            f"{'enabled' if SUPPORTS_PERF_TRAMPOLINE else 'not supported'}, "
+            f"timer_resolution: {TIMER_RESOLUTION_NS:.1f}ns"
+        )
+        return config_str, []
 
     def measure(
         self,

src/pytest_codspeed/plugin.py

@@ -2,7 +2,6 @@
 
 import functools
 import gc
-import importlib.util
 import json
 import os
 import random
@@ -19,11 +18,11 @@
     CodSpeedConfig,
     PedanticOptions,
 )
-from pytest_codspeed.instruments import (
-    MeasurementMode,
-    get_instrument_from_mode,
-)
+from pytest_codspeed.instruments import MeasurementMode, get_instrument_from_mode
 from pytest_codspeed.utils import (
+    BEFORE_PYTEST_8_1_1,
+    IS_PYTEST_BENCHMARK_INSTALLED,
+    IS_PYTEST_SPEED_INSTALLED,
     get_environment_metadata,
     get_git_relative_uri_and_name,
 )
@@ -37,10 +36,6 @@
 
     T = TypeVar("T")
 
-IS_PYTEST_BENCHMARK_INSTALLED = importlib.util.find_spec("pytest_benchmark") is not None
-IS_PYTEST_SPEED_INSTALLED = importlib.util.find_spec("pytest_speed") is not None
-BEFORE_PYTEST_8_1_1 = pytest.version_tuple < (8, 1, 1)
-
 
 @pytest.hookimpl(trylast=True)
 def pytest_addoption(parser: pytest.Parser):

src/pytest_codspeed/utils.py

@@ -1,10 +1,13 @@
 from __future__ import annotations
 
+import importlib.util
 import os
 import sys
 import sysconfig
 from pathlib import Path
 
+import pytest
+
 from pytest_codspeed import __semver_version__
 
 if sys.version_info < (3, 10):
@@ -13,6 +16,12 @@
     import importlib.metadata as importlib_metadata
 
 
+IS_PYTEST_BENCHMARK_INSTALLED = importlib.util.find_spec("pytest_benchmark") is not None
+IS_PYTEST_SPEED_INSTALLED = importlib.util.find_spec("pytest_speed") is not None
+BEFORE_PYTEST_8_1_1 = pytest.version_tuple < (8, 1, 1)
+SUPPORTS_PERF_TRAMPOLINE = sysconfig.get_config_var("PY_HAVE_PERF_TRAMPOLINE") == 1
+
+
 def get_git_relative_path(abs_path: Path) -> Path:
     """Get the path relative to the git root directory. If the path is not
     inside a git repository, the original path itself is returned.

tests/conftest.py

@@ -10,13 +10,13 @@
 import pytest
 
 from pytest_codspeed.instruments import MeasurementMode
+from pytest_codspeed.utils import IS_PYTEST_BENCHMARK_INSTALLED
 
 if TYPE_CHECKING:
     from _pytest.pytester import RunResult
 
 pytest_plugins = ["pytester"]
 
-IS_PYTEST_BENCHMARK_INSTALLED = importlib.util.find_spec("pytest_benchmark") is not None
 skip_without_pytest_benchmark = pytest.mark.skipif(
     not IS_PYTEST_BENCHMARK_INSTALLED, reason="pytest_benchmark not installed"
 )