Thanks to visit codestin.com
Credit goes to github.com

Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Check get_colors using the same output stream
  • Loading branch information
hugovk committed Jan 13, 2025
commit d0ce62201d6dd1670ae0de977c5fa141aa622bb5
7 changes: 5 additions & 2 deletions Lib/_colorize.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,8 +26,11 @@ class ANSIColors:
setattr(NoColors, attr, "")


def get_colors(colorize: bool = False) -> ANSIColors:
if colorize or can_colorize():
def get_colors(colorize: bool = False, *, file=None) -> ANSIColors:
if file is None:
file = sys.stdout

Comment thread
hugovk marked this conversation as resolved.
Outdated
if colorize or can_colorize(file=file):
return ANSIColors()
else:
return NoColors
Expand Down
4 changes: 2 additions & 2 deletions Lib/test/libregrtest/single.py
Original file line number Diff line number Diff line change
Expand Up @@ -162,7 +162,7 @@ def test_func():
def _runtest_env_changed_exc(result: TestResult, runtests: RunTests,
display_failure: bool = True) -> None:
# Handle exceptions, detect environment changes.
ansi = get_colors()
ansi = get_colors(file=sys.stderr)
Comment thread
hugovk marked this conversation as resolved.
Outdated
red, reset, yellow = ansi.RED, ansi.RESET, ansi.YELLOW

# Reset the environment_altered flag to detect if a test altered
Expand Down Expand Up @@ -303,7 +303,7 @@ def run_single_test(test_name: TestName, runtests: RunTests) -> TestResult:
If runtests.use_junit, xml_data is a list containing each generated
testsuite element.
"""
ansi = get_colors()
ansi = get_colors(file=sys.stderr)
red, reset, yellow = ansi.BOLD_RED, ansi.RESET, ansi.YELLOW

start_time = time.perf_counter()
Expand Down
2 changes: 1 addition & 1 deletion Lib/test/support/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -2839,7 +2839,7 @@ def no_color():
from .os_helper import EnvironmentVarGuard

with (
swap_attr(_colorize, "can_colorize", lambda: False),
swap_attr(_colorize, "can_colorize", lambda file=None: False),
EnvironmentVarGuard() as env,
):
for var in {"FORCE_COLOR", "NO_COLOR", "PYTHON_COLORS"}:
Expand Down
6 changes: 4 additions & 2 deletions Lib/unittest/runner.py
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,8 @@ def __init__(self, stream, descriptions, verbosity, *, durations=None):
self.showAll = verbosity > 1
self.dots = verbosity == 1
self.descriptions = descriptions
self._ansi = get_colors()
file = sys.stderr if stream == "<stderr>" else None
Comment thread
hugovk marked this conversation as resolved.
Outdated
self._ansi = get_colors(file=file)
self._newline = True
self.durations = durations

Expand Down Expand Up @@ -286,7 +287,8 @@ def run(self, test):
expected_fails, unexpected_successes, skipped = results

infos = []
ansi = get_colors()
file = sys.stderr if self.stream == "<stderr>" else None
Comment thread
hugovk marked this conversation as resolved.
Outdated
ansi = get_colors(file=file)
bold_red = ansi.BOLD_RED
green = ansi.GREEN
red = ansi.RED
Expand Down