import unittest
from io import StringIO, IncrementalNewlineDecoder
from collections import namedtuple
+from traceback import _ANSIColors, _can_colorize
class TestResults(namedtuple('TestResults', 'failed attempted')):
The `run` method is used to process a single DocTest case. It
returns a TestResults instance.
+ >>> save_colorize = traceback._COLORIZE
+ >>> traceback._COLORIZE = False
+
>>> tests = DocTestFinder().find(_TestClass)
>>> runner = DocTestRunner(verbose=False)
>>> tests.sort(key = lambda test: test.name)
can be also customized by subclassing DocTestRunner, and
overriding the methods `report_start`, `report_success`,
`report_unexpected_exception`, and `report_failure`.
+
+ >>> traceback._COLORIZE = save_colorize
"""
# This divider string is used to separate failure messages, and to
# separate sections of the summary.
'Exception raised:\n' + _indent(_exception_traceback(exc_info)))
def _failure_header(self, test, example):
- out = [self.DIVIDER]
+ red, reset = (
+ (_ANSIColors.RED, _ANSIColors.RESET) if _can_colorize() else ("", "")
+ )
+ out = [f"{red}{self.DIVIDER}{reset}"]
if test.filename:
if test.lineno is not None and example.lineno is not None:
lineno = test.lineno + example.lineno + 1
else:
failed.append((name, (failures, tries, skips)))
+ if _can_colorize():
+ bold_green = _ANSIColors.BOLD_GREEN
+ bold_red = _ANSIColors.BOLD_RED
+ green = _ANSIColors.GREEN
+ red = _ANSIColors.RED
+ reset = _ANSIColors.RESET
+ yellow = _ANSIColors.YELLOW
+ else:
+ bold_green = ""
+ bold_red = ""
+ green = ""
+ red = ""
+ reset = ""
+ yellow = ""
+
if verbose:
if notests:
print(f"{_n_items(notests)} had no tests:")
print(f" {name}")
if passed:
- print(f"{_n_items(passed)} passed all tests:")
+ print(f"{green}{_n_items(passed)} passed all tests:{reset}")
for name, count in sorted(passed):
s = "" if count == 1 else "s"
- print(f" {count:3d} test{s} in {name}")
+ print(f" {green}{count:3d} test{s} in {name}{reset}")
if failed:
- print(self.DIVIDER)
+ print(f"{red}{self.DIVIDER}{reset}")
print(f"{_n_items(failed)} had failures:")
for name, (failures, tries, skips) in sorted(failed):
print(f" {failures:3d} of {tries:3d} in {name}")
s = "" if total_tries == 1 else "s"
print(f"{total_tries} test{s} in {_n_items(self._stats)}.")
- and_f = f" and {total_failures} failed" if total_failures else ""
- print(f"{total_tries - total_failures} passed{and_f}.")
+ and_f = (
+ f" and {red}{total_failures} failed{reset}"
+ if total_failures else ""
+ )
+ print(f"{green}{total_tries - total_failures} passed{reset}{and_f}.")
if total_failures:
s = "" if total_failures == 1 else "s"
- msg = f"***Test Failed*** {total_failures} failure{s}"
+ msg = f"{bold_red}***Test Failed*** {total_failures} failure{s}{reset}"
if total_skips:
s = "" if total_skips == 1 else "s"
- msg = f"{msg} and {total_skips} skipped test{s}"
+ msg = f"{msg} and {yellow}{total_skips} skipped test{s}{reset}"
print(f"{msg}.")
elif verbose:
- print("Test passed.")
+ print(f"{bold_green}Test passed.{reset}")
return TestResults(total_failures, total_tries, skipped=total_skips)
d[name] = (failures, tries, skips)
-def _n_items(items: list) -> str:
+def _n_items(items: list | dict) -> str:
"""
Helper to pluralise the number of items in a list.
"""
class OutputChecker:
"""
- A class used to check the whether the actual output from a doctest
+ A class used to check whether the actual output from a doctest
example matches the expected output. `OutputChecker` defines two
methods: `check_output`, which compares a given pair of outputs,
and returns true if they match; and `output_difference`, which
"Error", "TestFailed", "TestDidNotRun", "ResourceDenied",
# io
"record_original_stdout", "get_original_stdout", "captured_stdout",
- "captured_stdin", "captured_stderr",
+ "captured_stdin", "captured_stderr", "captured_output",
# unittest
"is_resource_enabled", "requires", "requires_freebsd_version",
"requires_gil_enabled", "requires_linux_version", "requires_mac_ver",
import tempfile
import types
import contextlib
+import traceback
def doctest_skip_if(condition):
>>> tests = finder.find(sample_func)
>>> print(tests) # doctest: +ELLIPSIS
- [<DocTest sample_func from test_doctest.py:37 (1 example)>]
+ [<DocTest sample_func from test_doctest.py:38 (1 example)>]
The exact name depends on how test_doctest was invoked, so allow for
leading path components.
DocTestRunner is used to run DocTest test cases, and to accumulate
statistics. Here's a simple DocTest case we can use:
+ >>> save_colorize = traceback._COLORIZE
+ >>> traceback._COLORIZE = False
+
>>> def f(x):
... '''
... >>> x = 12
6
ok
TestResults(failed=1, attempted=3)
+
+ >>> traceback._COLORIZE = save_colorize
"""
def verbose_flag(): r"""
The `verbose` flag makes the test runner generate more detailed
lines between the first line and the type/value may be omitted or
replaced with any other string:
+ >>> save_colorize = traceback._COLORIZE
+ >>> traceback._COLORIZE = False
+
>>> def f(x):
... '''
... >>> x = 12
...
ZeroDivisionError: integer division or modulo by zero
TestResults(failed=1, attempted=1)
+
+ >>> traceback._COLORIZE = save_colorize
"""
def displayhook(): r"""
Test that changing sys.displayhook doesn't matter for doctest.
The DONT_ACCEPT_TRUE_FOR_1 flag disables matches between True/False
and 1/0:
+ >>> save_colorize = traceback._COLORIZE
+ >>> traceback._COLORIZE = False
+
>>> def f(x):
... '>>> True\n1\n'
Clean up.
>>> del doctest.OPTIONFLAGS_BY_NAME[unlikely]
+ >>> traceback._COLORIZE = save_colorize
"""
single example. To turn an option on for an example, follow that
example with a comment of the form ``# doctest: +OPTION``:
+ >>> save_colorize = traceback._COLORIZE
+ >>> traceback._COLORIZE = False
+
>>> def f(x): r'''
... >>> print(list(range(10))) # should fail: no ellipsis
... [0, 1, ..., 9]
>>> test = doctest.DocTestParser().get_doctest(s, {}, 's', 's.py', 0)
Traceback (most recent call last):
ValueError: line 0 of the doctest for s has an option directive on a line with no example: '# doctest: +ELLIPSIS'
+
+ >>> traceback._COLORIZE = save_colorize
"""
def test_testsource(): r"""
with a version that restores stdout. This is necessary for you to
see debugger output.
+ >>> save_colorize = traceback._COLORIZE
+ >>> traceback._COLORIZE = False
+
>>> doc = '''
... >>> x = 42
... >>> raise Exception('clé')
... finally:
... sys.stdin = real_stdin
--Return--
- > <doctest test.test_doctest.test_doctest.test_pdb_set_trace[7]>(3)calls_set_trace()->None
+ > <doctest test.test_doctest.test_doctest.test_pdb_set_trace[9]>(3)calls_set_trace()->None
-> import pdb; pdb.set_trace()
(Pdb) print(y)
2
Got:
9
TestResults(failed=1, attempted=3)
+
+ >>> traceback._COLORIZE = save_colorize
"""
def test_pdb_set_trace_nested():
called with the name of a file, which is taken to be relative to the
calling module. The return value is (#failures, #tests).
-We don't want `-v` in sys.argv for these tests.
+We don't want color or `-v` in sys.argv for these tests.
+
+ >>> save_colorize = traceback._COLORIZE
+ >>> traceback._COLORIZE = False
>>> save_argv = sys.argv
>>> if '-v' in sys.argv:
TestResults(failed=0, attempted=2)
>>> doctest.master = None # Reset master.
>>> sys.argv = save_argv
+ >>> traceback._COLORIZE = save_colorize
"""
class TestImporter(importlib.abc.MetaPathFinder, importlib.abc.ResourceLoader):
def test_unicode(): """
Check doctest with a non-ascii filename:
+ >>> save_colorize = traceback._COLORIZE
+ >>> traceback._COLORIZE = False
+
>>> doc = '''
... >>> raise Exception('clé')
... '''
raise Exception('clé')
Exception: clé
TestResults(failed=1, attempted=1)
+
+ >>> traceback._COLORIZE = save_colorize
"""
+
@doctest_skip_if(not support.has_subprocess_support)
def test_CLI(): r"""
The doctest module can be used to run doctests against an arbitrary file.
def test_exception_with_note(note):
"""
+ >>> save_colorize = traceback._COLORIZE
+ >>> traceback._COLORIZE = False
+
>>> test_exception_with_note('Note')
Traceback (most recent call last):
...
ValueError: message
note
TestResults(failed=1, attempted=...)
+
+ >>> traceback._COLORIZE = save_colorize
"""
exc = ValueError('Text')
exc.add_note(note)
def test_syntax_error_with_incorrect_expected_note():
"""
+ >>> save_colorize = traceback._COLORIZE
+ >>> traceback._COLORIZE = False
+
>>> def f(x):
... r'''
... >>> exc = SyntaxError("error", ("x.py", 23, None, "bad syntax"))
note1
note2
TestResults(failed=1, attempted=...)
+
+ >>> traceback._COLORIZE = save_colorize
"""
BOLD_RED = '\x1b[1;31m'
MAGENTA = '\x1b[35m'
BOLD_MAGENTA = '\x1b[1;35m'
+ GREEN = "\x1b[32m"
+ BOLD_GREEN = "\x1b[1;32m"
GREY = '\x1b[90m'
RESET = '\x1b[0m'
+ YELLOW = "\x1b[33m"
+
class StackSummary(list):
"""A list of FrameSummary objects, representing a stack of frames."""
--- /dev/null
+Add colour to doctest output. Patch by Hugo van Kemenade.
projects / thirdparty / Python / cpython.git / commitdiff
