From c3e8ec89898d2b245a06c95fd4160fb56f5458af Mon Sep 17 00:00:00 2001 From: dheerajck Date: Sat, 23 Mar 2024 21:07:03 +0530 Subject: [PATCH] Added environment variable PYTEST_VERSION which specifies pytest version, this is set when pytest is running and unset just before pytest is stopped --- src/_pytest/config/__init__.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/_pytest/config/__init__.py b/src/_pytest/config/__init__.py index 7ff27643f..3140e1c0f 100644 --- a/src/_pytest/config/__init__.py +++ b/src/_pytest/config/__init__.py @@ -50,6 +50,7 @@ from .compat import PathAwareHookProxy from .exceptions import PrintHelp as PrintHelp from .exceptions import UsageError as UsageError from .findpaths import determine_setup +from _pytest import __version__ import _pytest._code from _pytest._code import ExceptionInfo from _pytest._code import filter_traceback @@ -152,6 +153,7 @@ def main( :returns: An exit code. """ try: + os.environ["PYTEST_VERSION"] = __version__ try: config = _prepareconfig(args, plugins) except ConftestImportFailure as e: @@ -186,6 +188,8 @@ def main( for msg in e.args: tw.line(f"ERROR: {msg}\n", red=True) return ExitCode.USAGE_ERROR + finally: + os.environ.pop("PYTEST_VERSION", None) def console_main() -> int: