diff --git a/changelog/11666.improvement.rst b/changelog/11666.improvement.rst index b5cdf863b..9cb3a3b7f 100644 --- a/changelog/11666.improvement.rst +++ b/changelog/11666.improvement.rst @@ -2,4 +2,4 @@ Improved the terminal writer to be able to use Terminal256Formatter and Terminal These are newer versions of TerminalFormmater from the pygments library that is currently in use. -These style of these formatters are set by the users through the environmental variable PYTEST_THEME. \ No newline at end of file +These style of these formatters are set by the users through the environmental variable PYTEST_THEME. diff --git a/src/_pytest/_io/terminalwriter.py b/src/_pytest/_io/terminalwriter.py index c499161dd..f21a38059 100644 --- a/src/_pytest/_io/terminalwriter.py +++ b/src/_pytest/_io/terminalwriter.py @@ -219,20 +219,24 @@ class TerminalWriter: else: try: # Import new more rich color formatters from the pygments library - from pygments.formatters.terminal256 import TerminalTrueColorFormatter from pygments.formatters.terminal256 import Terminal256Formatter + from pygments.formatters.terminal256 import TerminalTrueColorFormatter # Use terminal formatters depending on user environment variables - if os.environ.get('COLORTERM','') in ('truecolor', '24bit'): + if os.environ.get("COLORTERM", "") in ("truecolor", "24bit"): # Style determined by user set environment variable, if none then use default style - terminal_formatter = TerminalTrueColorFormatter(style=os.getenv("PYTEST_THEME","default")) - elif '256' in os.environ.get('TERM', ''): - terminal_formatter = Terminal256Formatter(style=os.getenv("PYTEST_THEME","default")) + terminal_formatter = TerminalTrueColorFormatter( + style=os.getenv("PYTEST_THEME", "default") + ) + elif "256" in os.environ.get("TERM", ""): + terminal_formatter = Terminal256Formatter( + style=os.getenv("PYTEST_THEME", "default") + ) else: terminal_formatter = TerminalFormatter( - bg=os.getenv("PYTEST_THEME_MODE", "dark"), - style=os.getenv("PYTEST_THEME"), - ) + bg=os.getenv("PYTEST_THEME_MODE", "dark"), + style=os.getenv("PYTEST_THEME"), + ) highlighted: str = highlight( source, Lexer(),