mirror of https://github.com/python/cpython
3555 lines
138 KiB
Python
3555 lines
138 KiB
Python
"""Test cases for traceback module"""
|
||
|
||
from collections import namedtuple
|
||
from io import StringIO
|
||
import linecache
|
||
import sys
|
||
import types
|
||
import inspect
|
||
import builtins
|
||
import unittest
|
||
import re
|
||
import tempfile
|
||
import random
|
||
import string
|
||
from test import support
|
||
import shutil
|
||
from test.support import (Error, captured_output, cpython_only, ALWAYS_EQ,
|
||
requires_debug_ranges, has_no_debug_ranges,
|
||
requires_subprocess)
|
||
from test.support.os_helper import TESTFN, unlink
|
||
from test.support.script_helper import assert_python_ok, assert_python_failure
|
||
from test.support.import_helper import forget
|
||
|
||
import json
|
||
import textwrap
|
||
import traceback
|
||
from functools import partial
|
||
from pathlib import Path
|
||
|
||
MODULE_PREFIX = f'{__name__}.' if __name__ == '__main__' else ''
|
||
|
||
test_code = namedtuple('code', ['co_filename', 'co_name'])
|
||
test_code.co_positions = lambda _: iter([(6, 6, 0, 0)])
|
||
test_frame = namedtuple('frame', ['f_code', 'f_globals', 'f_locals'])
|
||
test_tb = namedtuple('tb', ['tb_frame', 'tb_lineno', 'tb_next', 'tb_lasti'])
|
||
|
||
|
||
LEVENSHTEIN_DATA_FILE = Path(__file__).parent / 'levenshtein_examples.json'
|
||
|
||
|
||
class TracebackCases(unittest.TestCase):
|
||
# For now, a very minimal set of tests. I want to be sure that
|
||
# formatting of SyntaxErrors works based on changes for 2.1.
|
||
|
||
def get_exception_format(self, func, exc):
|
||
try:
|
||
func()
|
||
except exc as value:
|
||
return traceback.format_exception_only(exc, value)
|
||
else:
|
||
raise ValueError("call did not raise exception")
|
||
|
||
def syntax_error_with_caret(self):
|
||
compile("def fact(x):\n\treturn x!\n", "?", "exec")
|
||
|
||
def syntax_error_with_caret_2(self):
|
||
compile("1 +\n", "?", "exec")
|
||
|
||
def syntax_error_with_caret_range(self):
|
||
compile("f(x, y for y in range(30), z)", "?", "exec")
|
||
|
||
def syntax_error_bad_indentation(self):
|
||
compile("def spam():\n print(1)\n print(2)", "?", "exec")
|
||
|
||
def syntax_error_with_caret_non_ascii(self):
|
||
compile('Python = "\u1e54\xfd\u0163\u0125\xf2\xf1" +', "?", "exec")
|
||
|
||
def syntax_error_bad_indentation2(self):
|
||
compile(" print(2)", "?", "exec")
|
||
|
||
def tokenizer_error_with_caret_range(self):
|
||
compile("blech ( ", "?", "exec")
|
||
|
||
def test_caret(self):
|
||
err = self.get_exception_format(self.syntax_error_with_caret,
|
||
SyntaxError)
|
||
self.assertEqual(len(err), 4)
|
||
self.assertTrue(err[1].strip() == "return x!")
|
||
self.assertIn("^", err[2]) # third line has caret
|
||
self.assertEqual(err[1].find("!"), err[2].find("^")) # in the right place
|
||
self.assertEqual(err[2].count("^"), 1)
|
||
|
||
err = self.get_exception_format(self.syntax_error_with_caret_2,
|
||
SyntaxError)
|
||
self.assertIn("^", err[2]) # third line has caret
|
||
self.assertEqual(err[2].count('\n'), 1) # and no additional newline
|
||
self.assertEqual(err[1].find("+") + 1, err[2].find("^")) # in the right place
|
||
self.assertEqual(err[2].count("^"), 1)
|
||
|
||
err = self.get_exception_format(self.syntax_error_with_caret_non_ascii,
|
||
SyntaxError)
|
||
self.assertIn("^", err[2]) # third line has caret
|
||
self.assertEqual(err[2].count('\n'), 1) # and no additional newline
|
||
self.assertEqual(err[1].find("+") + 1, err[2].find("^")) # in the right place
|
||
self.assertEqual(err[2].count("^"), 1)
|
||
|
||
err = self.get_exception_format(self.syntax_error_with_caret_range,
|
||
SyntaxError)
|
||
self.assertIn("^", err[2]) # third line has caret
|
||
self.assertEqual(err[2].count('\n'), 1) # and no additional newline
|
||
self.assertEqual(err[1].find("y"), err[2].find("^")) # in the right place
|
||
self.assertEqual(err[2].count("^"), len("y for y in range(30)"))
|
||
|
||
err = self.get_exception_format(self.tokenizer_error_with_caret_range,
|
||
SyntaxError)
|
||
self.assertIn("^", err[2]) # third line has caret
|
||
self.assertEqual(err[2].count('\n'), 1) # and no additional newline
|
||
self.assertEqual(err[1].find("("), err[2].find("^")) # in the right place
|
||
self.assertEqual(err[2].count("^"), 1)
|
||
|
||
def test_nocaret(self):
|
||
exc = SyntaxError("error", ("x.py", 23, None, "bad syntax"))
|
||
err = traceback.format_exception_only(SyntaxError, exc)
|
||
self.assertEqual(len(err), 3)
|
||
self.assertEqual(err[1].strip(), "bad syntax")
|
||
|
||
def test_no_caret_with_no_debug_ranges_flag(self):
|
||
# Make sure that if `-X no_debug_ranges` is used, there are no carets
|
||
# in the traceback.
|
||
try:
|
||
with open(TESTFN, 'w') as f:
|
||
f.write("x = 1 / 0\n")
|
||
|
||
_, _, stderr = assert_python_failure(
|
||
'-X', 'no_debug_ranges', TESTFN)
|
||
|
||
lines = stderr.splitlines()
|
||
self.assertEqual(len(lines), 4)
|
||
self.assertEqual(lines[0], b'Traceback (most recent call last):')
|
||
self.assertIn(b'line 1, in <module>', lines[1])
|
||
self.assertEqual(lines[2], b' x = 1 / 0')
|
||
self.assertEqual(lines[3], b'ZeroDivisionError: division by zero')
|
||
finally:
|
||
unlink(TESTFN)
|
||
|
||
def test_no_caret_with_no_debug_ranges_flag_python_traceback(self):
|
||
code = textwrap.dedent("""
|
||
import traceback
|
||
try:
|
||
x = 1 / 0
|
||
except:
|
||
traceback.print_exc()
|
||
""")
|
||
try:
|
||
with open(TESTFN, 'w') as f:
|
||
f.write(code)
|
||
|
||
_, _, stderr = assert_python_ok(
|
||
'-X', 'no_debug_ranges', TESTFN)
|
||
|
||
lines = stderr.splitlines()
|
||
self.assertEqual(len(lines), 4)
|
||
self.assertEqual(lines[0], b'Traceback (most recent call last):')
|
||
self.assertIn(b'line 4, in <module>', lines[1])
|
||
self.assertEqual(lines[2], b' x = 1 / 0')
|
||
self.assertEqual(lines[3], b'ZeroDivisionError: division by zero')
|
||
finally:
|
||
unlink(TESTFN)
|
||
|
||
def test_recursion_error_during_traceback(self):
|
||
code = textwrap.dedent("""
|
||
import sys
|
||
from weakref import ref
|
||
|
||
sys.setrecursionlimit(15)
|
||
|
||
def f():
|
||
ref(lambda: 0, [])
|
||
f()
|
||
|
||
try:
|
||
f()
|
||
except RecursionError:
|
||
pass
|
||
""")
|
||
try:
|
||
with open(TESTFN, 'w') as f:
|
||
f.write(code)
|
||
|
||
rc, _, _ = assert_python_ok(TESTFN)
|
||
self.assertEqual(rc, 0)
|
||
finally:
|
||
unlink(TESTFN)
|
||
|
||
def test_bad_indentation(self):
|
||
err = self.get_exception_format(self.syntax_error_bad_indentation,
|
||
IndentationError)
|
||
self.assertEqual(len(err), 4)
|
||
self.assertEqual(err[1].strip(), "print(2)")
|
||
self.assertIn("^", err[2])
|
||
self.assertEqual(err[1].find(")") + 1, err[2].find("^"))
|
||
|
||
# No caret for "unexpected indent"
|
||
err = self.get_exception_format(self.syntax_error_bad_indentation2,
|
||
IndentationError)
|
||
self.assertEqual(len(err), 3)
|
||
self.assertEqual(err[1].strip(), "print(2)")
|
||
|
||
def test_base_exception(self):
|
||
# Test that exceptions derived from BaseException are formatted right
|
||
e = KeyboardInterrupt()
|
||
lst = traceback.format_exception_only(e.__class__, e)
|
||
self.assertEqual(lst, ['KeyboardInterrupt\n'])
|
||
|
||
def test_format_exception_only_bad__str__(self):
|
||
class X(Exception):
|
||
def __str__(self):
|
||
1/0
|
||
err = traceback.format_exception_only(X, X())
|
||
self.assertEqual(len(err), 1)
|
||
str_value = '<exception str() failed>'
|
||
if X.__module__ in ('__main__', 'builtins'):
|
||
str_name = X.__qualname__
|
||
else:
|
||
str_name = '.'.join([X.__module__, X.__qualname__])
|
||
self.assertEqual(err[0], "%s: %s\n" % (str_name, str_value))
|
||
|
||
@requires_subprocess()
|
||
def test_encoded_file(self):
|
||
# Test that tracebacks are correctly printed for encoded source files:
|
||
# - correct line number (Issue2384)
|
||
# - respect file encoding (Issue3975)
|
||
import sys, subprocess
|
||
|
||
# The spawned subprocess has its stdout redirected to a PIPE, and its
|
||
# encoding may be different from the current interpreter, on Windows
|
||
# at least.
|
||
process = subprocess.Popen([sys.executable, "-c",
|
||
"import sys; print(sys.stdout.encoding)"],
|
||
stdout=subprocess.PIPE,
|
||
stderr=subprocess.STDOUT)
|
||
stdout, stderr = process.communicate()
|
||
output_encoding = str(stdout, 'ascii').splitlines()[0]
|
||
|
||
def do_test(firstlines, message, charset, lineno):
|
||
# Raise the message in a subprocess, and catch the output
|
||
try:
|
||
with open(TESTFN, "w", encoding=charset) as output:
|
||
output.write("""{0}if 1:
|
||
import traceback;
|
||
raise RuntimeError('{1}')
|
||
""".format(firstlines, message))
|
||
|
||
process = subprocess.Popen([sys.executable, TESTFN],
|
||
stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
|
||
stdout, stderr = process.communicate()
|
||
stdout = stdout.decode(output_encoding).splitlines()
|
||
finally:
|
||
unlink(TESTFN)
|
||
|
||
# The source lines are encoded with the 'backslashreplace' handler
|
||
encoded_message = message.encode(output_encoding,
|
||
'backslashreplace')
|
||
# and we just decoded them with the output_encoding.
|
||
message_ascii = encoded_message.decode(output_encoding)
|
||
|
||
err_line = "raise RuntimeError('{0}')".format(message_ascii)
|
||
err_msg = "RuntimeError: {0}".format(message_ascii)
|
||
|
||
self.assertIn(("line %s" % lineno), stdout[1],
|
||
"Invalid line number: {0!r} instead of {1}".format(
|
||
stdout[1], lineno))
|
||
self.assertTrue(stdout[2].endswith(err_line),
|
||
"Invalid traceback line: {0!r} instead of {1!r}".format(
|
||
stdout[2], err_line))
|
||
actual_err_msg = stdout[3]
|
||
self.assertTrue(actual_err_msg == err_msg,
|
||
"Invalid error message: {0!r} instead of {1!r}".format(
|
||
actual_err_msg, err_msg))
|
||
|
||
do_test("", "foo", "ascii", 3)
|
||
for charset in ("ascii", "iso-8859-1", "utf-8", "GBK"):
|
||
if charset == "ascii":
|
||
text = "foo"
|
||
elif charset == "GBK":
|
||
text = "\u4E02\u5100"
|
||
else:
|
||
text = "h\xe9 ho"
|
||
do_test("# coding: {0}\n".format(charset),
|
||
text, charset, 4)
|
||
do_test("#!shebang\n# coding: {0}\n".format(charset),
|
||
text, charset, 5)
|
||
do_test(" \t\f\n# coding: {0}\n".format(charset),
|
||
text, charset, 5)
|
||
# Issue #18960: coding spec should have no effect
|
||
do_test("x=0\n# coding: GBK\n", "h\xe9 ho", 'utf-8', 5)
|
||
|
||
def test_print_traceback_at_exit(self):
|
||
# Issue #22599: Ensure that it is possible to use the traceback module
|
||
# to display an exception at Python exit
|
||
code = textwrap.dedent("""
|
||
import sys
|
||
import traceback
|
||
|
||
class PrintExceptionAtExit(object):
|
||
def __init__(self):
|
||
try:
|
||
x = 1 / 0
|
||
except Exception as e:
|
||
self.exc = e
|
||
# self.exc.__traceback__ contains frames:
|
||
# explicitly clear the reference to self in the current
|
||
# frame to break a reference cycle
|
||
self = None
|
||
|
||
def __del__(self):
|
||
traceback.print_exception(self.exc)
|
||
|
||
# Keep a reference in the module namespace to call the destructor
|
||
# when the module is unloaded
|
||
obj = PrintExceptionAtExit()
|
||
""")
|
||
rc, stdout, stderr = assert_python_ok('-c', code)
|
||
expected = [b'Traceback (most recent call last):',
|
||
b' File "<string>", line 8, in __init__',
|
||
b'ZeroDivisionError: division by zero']
|
||
self.assertEqual(stderr.splitlines(), expected)
|
||
|
||
def test_print_exception(self):
|
||
output = StringIO()
|
||
traceback.print_exception(
|
||
Exception, Exception("projector"), None, file=output
|
||
)
|
||
self.assertEqual(output.getvalue(), "Exception: projector\n")
|
||
|
||
def test_print_exception_exc(self):
|
||
output = StringIO()
|
||
traceback.print_exception(Exception("projector"), file=output)
|
||
self.assertEqual(output.getvalue(), "Exception: projector\n")
|
||
|
||
def test_format_exception_exc(self):
|
||
e = Exception("projector")
|
||
output = traceback.format_exception(e)
|
||
self.assertEqual(output, ["Exception: projector\n"])
|
||
with self.assertRaisesRegex(ValueError, 'Both or neither'):
|
||
traceback.format_exception(e.__class__, e)
|
||
with self.assertRaisesRegex(ValueError, 'Both or neither'):
|
||
traceback.format_exception(e.__class__, tb=e.__traceback__)
|
||
with self.assertRaisesRegex(TypeError, 'positional-only'):
|
||
traceback.format_exception(exc=e)
|
||
|
||
def test_format_exception_only_exc(self):
|
||
output = traceback.format_exception_only(Exception("projector"))
|
||
self.assertEqual(output, ["Exception: projector\n"])
|
||
|
||
def test_exception_is_None(self):
|
||
NONE_EXC_STRING = 'NoneType: None\n'
|
||
excfile = StringIO()
|
||
traceback.print_exception(None, file=excfile)
|
||
self.assertEqual(excfile.getvalue(), NONE_EXC_STRING)
|
||
|
||
excfile = StringIO()
|
||
traceback.print_exception(None, None, None, file=excfile)
|
||
self.assertEqual(excfile.getvalue(), NONE_EXC_STRING)
|
||
|
||
excfile = StringIO()
|
||
traceback.print_exc(None, file=excfile)
|
||
self.assertEqual(excfile.getvalue(), NONE_EXC_STRING)
|
||
|
||
self.assertEqual(traceback.format_exc(None), NONE_EXC_STRING)
|
||
self.assertEqual(traceback.format_exception(None), [NONE_EXC_STRING])
|
||
self.assertEqual(
|
||
traceback.format_exception(None, None, None), [NONE_EXC_STRING])
|
||
self.assertEqual(
|
||
traceback.format_exception_only(None), [NONE_EXC_STRING])
|
||
self.assertEqual(
|
||
traceback.format_exception_only(None, None), [NONE_EXC_STRING])
|
||
|
||
def test_signatures(self):
|
||
self.assertEqual(
|
||
str(inspect.signature(traceback.print_exception)),
|
||
('(exc, /, value=<implicit>, tb=<implicit>, '
|
||
'limit=None, file=None, chain=True)'))
|
||
|
||
self.assertEqual(
|
||
str(inspect.signature(traceback.format_exception)),
|
||
('(exc, /, value=<implicit>, tb=<implicit>, limit=None, '
|
||
'chain=True)'))
|
||
|
||
self.assertEqual(
|
||
str(inspect.signature(traceback.format_exception_only)),
|
||
'(exc, /, value=<implicit>)')
|
||
|
||
|
||
class PurePythonExceptionFormattingMixin:
|
||
def get_exception(self, callable, slice_start=0, slice_end=-1):
|
||
try:
|
||
callable()
|
||
self.fail("No exception thrown.")
|
||
except:
|
||
return traceback.format_exc().splitlines()[slice_start:slice_end]
|
||
|
||
callable_line = get_exception.__code__.co_firstlineno + 2
|
||
|
||
|
||
class CAPIExceptionFormattingMixin:
|
||
LEGACY = 0
|
||
|
||
def get_exception(self, callable, slice_start=0, slice_end=-1):
|
||
from _testcapi import exception_print
|
||
try:
|
||
callable()
|
||
self.fail("No exception thrown.")
|
||
except Exception as e:
|
||
with captured_output("stderr") as tbstderr:
|
||
exception_print(e, self.LEGACY)
|
||
return tbstderr.getvalue().splitlines()[slice_start:slice_end]
|
||
|
||
callable_line = get_exception.__code__.co_firstlineno + 3
|
||
|
||
class CAPIExceptionFormattingLegacyMixin(CAPIExceptionFormattingMixin):
|
||
LEGACY = 1
|
||
|
||
@requires_debug_ranges()
|
||
class TracebackErrorLocationCaretTestBase:
|
||
"""
|
||
Tests for printing code error expressions as part of PEP 657
|
||
"""
|
||
def test_basic_caret(self):
|
||
# NOTE: In caret tests, "if True:" is used as a way to force indicator
|
||
# display, since the raising expression spans only part of the line.
|
||
def f():
|
||
if True: raise ValueError("basic caret tests")
|
||
|
||
lineno_f = f.__code__.co_firstlineno
|
||
expected_f = (
|
||
'Traceback (most recent call last):\n'
|
||
f' File "{__file__}", line {self.callable_line}, in get_exception\n'
|
||
' callable()\n'
|
||
f' File "{__file__}", line {lineno_f+1}, in f\n'
|
||
' if True: raise ValueError("basic caret tests")\n'
|
||
' ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^\n'
|
||
)
|
||
result_lines = self.get_exception(f)
|
||
self.assertEqual(result_lines, expected_f.splitlines())
|
||
|
||
def test_line_with_unicode(self):
|
||
# Make sure that even if a line contains multi-byte unicode characters
|
||
# the correct carets are printed.
|
||
def f_with_unicode():
|
||
if True: raise ValueError("Ĥellö Wörld")
|
||
|
||
lineno_f = f_with_unicode.__code__.co_firstlineno
|
||
expected_f = (
|
||
'Traceback (most recent call last):\n'
|
||
f' File "{__file__}", line {self.callable_line}, in get_exception\n'
|
||
' callable()\n'
|
||
f' File "{__file__}", line {lineno_f+1}, in f_with_unicode\n'
|
||
' if True: raise ValueError("Ĥellö Wörld")\n'
|
||
' ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^\n'
|
||
)
|
||
result_lines = self.get_exception(f_with_unicode)
|
||
self.assertEqual(result_lines, expected_f.splitlines())
|
||
|
||
def test_caret_in_type_annotation(self):
|
||
def f_with_type():
|
||
def foo(a: THIS_DOES_NOT_EXIST ) -> int:
|
||
return 0
|
||
|
||
lineno_f = f_with_type.__code__.co_firstlineno
|
||
expected_f = (
|
||
'Traceback (most recent call last):\n'
|
||
f' File "{__file__}", line {self.callable_line}, in get_exception\n'
|
||
' callable()\n'
|
||
f' File "{__file__}", line {lineno_f+1}, in f_with_type\n'
|
||
' def foo(a: THIS_DOES_NOT_EXIST ) -> int:\n'
|
||
' ^^^^^^^^^^^^^^^^^^^\n'
|
||
)
|
||
result_lines = self.get_exception(f_with_type)
|
||
self.assertEqual(result_lines, expected_f.splitlines())
|
||
|
||
def test_caret_multiline_expression(self):
|
||
# Make sure no carets are printed for expressions spanning multiple
|
||
# lines.
|
||
def f_with_multiline():
|
||
if True: raise ValueError(
|
||
"error over multiple lines"
|
||
)
|
||
|
||
lineno_f = f_with_multiline.__code__.co_firstlineno
|
||
expected_f = (
|
||
'Traceback (most recent call last):\n'
|
||
f' File "{__file__}", line {self.callable_line}, in get_exception\n'
|
||
' callable()\n'
|
||
f' File "{__file__}", line {lineno_f+1}, in f_with_multiline\n'
|
||
' if True: raise ValueError(\n'
|
||
' ^^^^^^^^^^^^^^^^^'
|
||
)
|
||
result_lines = self.get_exception(f_with_multiline)
|
||
self.assertEqual(result_lines, expected_f.splitlines())
|
||
|
||
def test_caret_multiline_expression_syntax_error(self):
|
||
# Make sure an expression spanning multiple lines that has
|
||
# a syntax error is correctly marked with carets.
|
||
code = textwrap.dedent("""
|
||
def foo(*args, **kwargs):
|
||
pass
|
||
|
||
a, b, c = 1, 2, 3
|
||
|
||
foo(a, z
|
||
for z in
|
||
range(10), b, c)
|
||
""")
|
||
|
||
def f_with_multiline():
|
||
# Need to defer the compilation until in self.get_exception(..)
|
||
return compile(code, "?", "exec")
|
||
|
||
lineno_f = f_with_multiline.__code__.co_firstlineno
|
||
|
||
expected_f = (
|
||
'Traceback (most recent call last):\n'
|
||
f' File "{__file__}", line {self.callable_line}, in get_exception\n'
|
||
' callable()\n'
|
||
f' File "{__file__}", line {lineno_f+2}, in f_with_multiline\n'
|
||
' return compile(code, "?", "exec")\n'
|
||
' ^^^^^^^^^^^^^^^^^^^^^^^^^^\n'
|
||
' File "?", line 7\n'
|
||
' foo(a, z\n'
|
||
' ^'
|
||
)
|
||
|
||
result_lines = self.get_exception(f_with_multiline)
|
||
self.assertEqual(result_lines, expected_f.splitlines())
|
||
|
||
def test_caret_multiline_expression_bin_op(self):
|
||
# Make sure no carets are printed for expressions spanning multiple
|
||
# lines.
|
||
def f_with_multiline():
|
||
return (
|
||
2 + 1 /
|
||
0
|
||
)
|
||
|
||
lineno_f = f_with_multiline.__code__.co_firstlineno
|
||
expected_f = (
|
||
'Traceback (most recent call last):\n'
|
||
f' File "{__file__}", line {self.callable_line}, in get_exception\n'
|
||
' callable()\n'
|
||
f' File "{__file__}", line {lineno_f+2}, in f_with_multiline\n'
|
||
' 2 + 1 /\n'
|
||
' ^^^'
|
||
)
|
||
result_lines = self.get_exception(f_with_multiline)
|
||
self.assertEqual(result_lines, expected_f.splitlines())
|
||
|
||
def test_caret_for_binary_operators(self):
|
||
def f_with_binary_operator():
|
||
divisor = 20
|
||
return 10 + divisor / 0 + 30
|
||
|
||
lineno_f = f_with_binary_operator.__code__.co_firstlineno
|
||
expected_error = (
|
||
'Traceback (most recent call last):\n'
|
||
f' File "{__file__}", line {self.callable_line}, in get_exception\n'
|
||
' callable()\n'
|
||
f' File "{__file__}", line {lineno_f+2}, in f_with_binary_operator\n'
|
||
' return 10 + divisor / 0 + 30\n'
|
||
' ~~~~~~~~^~~\n'
|
||
)
|
||
result_lines = self.get_exception(f_with_binary_operator)
|
||
self.assertEqual(result_lines, expected_error.splitlines())
|
||
|
||
def test_caret_for_binary_operators_with_unicode(self):
|
||
def f_with_binary_operator():
|
||
áóí = 20
|
||
return 10 + áóí / 0 + 30
|
||
|
||
lineno_f = f_with_binary_operator.__code__.co_firstlineno
|
||
expected_error = (
|
||
'Traceback (most recent call last):\n'
|
||
f' File "{__file__}", line {self.callable_line}, in get_exception\n'
|
||
' callable()\n'
|
||
f' File "{__file__}", line {lineno_f+2}, in f_with_binary_operator\n'
|
||
' return 10 + áóí / 0 + 30\n'
|
||
' ~~~~^~~\n'
|
||
)
|
||
result_lines = self.get_exception(f_with_binary_operator)
|
||
self.assertEqual(result_lines, expected_error.splitlines())
|
||
|
||
def test_caret_for_binary_operators_two_char(self):
|
||
def f_with_binary_operator():
|
||
divisor = 20
|
||
return 10 + divisor // 0 + 30
|
||
|
||
lineno_f = f_with_binary_operator.__code__.co_firstlineno
|
||
expected_error = (
|
||
'Traceback (most recent call last):\n'
|
||
f' File "{__file__}", line {self.callable_line}, in get_exception\n'
|
||
' callable()\n'
|
||
f' File "{__file__}", line {lineno_f+2}, in f_with_binary_operator\n'
|
||
' return 10 + divisor // 0 + 30\n'
|
||
' ~~~~~~~~^^~~\n'
|
||
)
|
||
result_lines = self.get_exception(f_with_binary_operator)
|
||
self.assertEqual(result_lines, expected_error.splitlines())
|
||
|
||
def test_caret_for_subscript(self):
|
||
def f_with_subscript():
|
||
some_dict = {'x': {'y': None}}
|
||
return some_dict['x']['y']['z']
|
||
|
||
lineno_f = f_with_subscript.__code__.co_firstlineno
|
||
expected_error = (
|
||
'Traceback (most recent call last):\n'
|
||
f' File "{__file__}", line {self.callable_line}, in get_exception\n'
|
||
' callable()\n'
|
||
f' File "{__file__}", line {lineno_f+2}, in f_with_subscript\n'
|
||
" return some_dict['x']['y']['z']\n"
|
||
' ~~~~~~~~~~~~~~~~~~~^^^^^\n'
|
||
)
|
||
result_lines = self.get_exception(f_with_subscript)
|
||
self.assertEqual(result_lines, expected_error.splitlines())
|
||
|
||
def test_caret_for_subscript_unicode(self):
|
||
def f_with_subscript():
|
||
some_dict = {'ó': {'á': {'í': {'theta': 1}}}}
|
||
return some_dict['ó']['á']['í']['beta']
|
||
|
||
lineno_f = f_with_subscript.__code__.co_firstlineno
|
||
expected_error = (
|
||
'Traceback (most recent call last):\n'
|
||
f' File "{__file__}", line {self.callable_line}, in get_exception\n'
|
||
' callable()\n'
|
||
f' File "{__file__}", line {lineno_f+2}, in f_with_subscript\n'
|
||
" return some_dict['ó']['á']['í']['beta']\n"
|
||
' ~~~~~~~~~~~~~~~~~~~~~~~~^^^^^^^^\n'
|
||
)
|
||
result_lines = self.get_exception(f_with_subscript)
|
||
self.assertEqual(result_lines, expected_error.splitlines())
|
||
|
||
def test_traceback_specialization_with_syntax_error(self):
|
||
bytecode = compile("1 / 0 / 1 / 2\n", TESTFN, "exec")
|
||
|
||
with open(TESTFN, "w") as file:
|
||
# make the file's contents invalid
|
||
file.write("1 $ 0 / 1 / 2\n")
|
||
self.addCleanup(unlink, TESTFN)
|
||
|
||
func = partial(exec, bytecode)
|
||
result_lines = self.get_exception(func)
|
||
|
||
lineno_f = bytecode.co_firstlineno
|
||
expected_error = (
|
||
'Traceback (most recent call last):\n'
|
||
f' File "{__file__}", line {self.callable_line}, in get_exception\n'
|
||
' callable()\n'
|
||
f' File "{TESTFN}", line {lineno_f}, in <module>\n'
|
||
" 1 $ 0 / 1 / 2\n"
|
||
' ^^^^^\n'
|
||
)
|
||
self.assertEqual(result_lines, expected_error.splitlines())
|
||
|
||
def test_traceback_very_long_line(self):
|
||
source = "if True: " + "a" * 256
|
||
bytecode = compile(source, TESTFN, "exec")
|
||
|
||
with open(TESTFN, "w") as file:
|
||
file.write(source)
|
||
self.addCleanup(unlink, TESTFN)
|
||
|
||
func = partial(exec, bytecode)
|
||
result_lines = self.get_exception(func)
|
||
|
||
lineno_f = bytecode.co_firstlineno
|
||
expected_error = (
|
||
'Traceback (most recent call last):\n'
|
||
f' File "{__file__}", line {self.callable_line}, in get_exception\n'
|
||
' callable()\n'
|
||
f' File "{TESTFN}", line {lineno_f}, in <module>\n'
|
||
f' {source}\n'
|
||
f' {" "*len("if True: ") + "^"*256}\n'
|
||
)
|
||
self.assertEqual(result_lines, expected_error.splitlines())
|
||
|
||
def test_secondary_caret_not_elided(self):
|
||
# Always show a line's indicators if they include the secondary character.
|
||
def f_with_subscript():
|
||
some_dict = {'x': {'y': None}}
|
||
some_dict['x']['y']['z']
|
||
|
||
lineno_f = f_with_subscript.__code__.co_firstlineno
|
||
expected_error = (
|
||
'Traceback (most recent call last):\n'
|
||
f' File "{__file__}", line {self.callable_line}, in get_exception\n'
|
||
' callable()\n'
|
||
f' File "{__file__}", line {lineno_f+2}, in f_with_subscript\n'
|
||
" some_dict['x']['y']['z']\n"
|
||
' ~~~~~~~~~~~~~~~~~~~^^^^^\n'
|
||
)
|
||
result_lines = self.get_exception(f_with_subscript)
|
||
self.assertEqual(result_lines, expected_error.splitlines())
|
||
|
||
def test_caret_exception_group(self):
|
||
# Notably, this covers whether indicators handle margin strings correctly.
|
||
# (Exception groups use margin strings to display vertical indicators.)
|
||
# The implementation must account for both "indent" and "margin" offsets.
|
||
|
||
def exc():
|
||
if True: raise ExceptionGroup("eg", [ValueError(1), TypeError(2)])
|
||
|
||
expected_error = (
|
||
f' + Exception Group Traceback (most recent call last):\n'
|
||
f' | File "{__file__}", line {self.callable_line}, in get_exception\n'
|
||
f' | callable()\n'
|
||
f' | File "{__file__}", line {exc.__code__.co_firstlineno + 1}, in exc\n'
|
||
f' | if True: raise ExceptionGroup("eg", [ValueError(1), TypeError(2)])\n'
|
||
f' | ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^\n'
|
||
f' | ExceptionGroup: eg (2 sub-exceptions)\n'
|
||
f' +-+---------------- 1 ----------------\n'
|
||
f' | ValueError: 1\n'
|
||
f' +---------------- 2 ----------------\n'
|
||
f' | TypeError: 2\n')
|
||
|
||
result_lines = self.get_exception(exc)
|
||
self.assertEqual(result_lines, expected_error.splitlines())
|
||
|
||
def assertSpecialized(self, func, expected_specialization):
|
||
result_lines = self.get_exception(func)
|
||
specialization_line = result_lines[-1]
|
||
self.assertEqual(specialization_line.lstrip(), expected_specialization)
|
||
|
||
def test_specialization_variations(self):
|
||
self.assertSpecialized(lambda: 1/0,
|
||
"~^~")
|
||
self.assertSpecialized(lambda: 1/0/3,
|
||
"~^~")
|
||
self.assertSpecialized(lambda: 1 / 0,
|
||
"~~^~~")
|
||
self.assertSpecialized(lambda: 1 / 0 / 3,
|
||
"~~^~~")
|
||
self.assertSpecialized(lambda: 1/ 0,
|
||
"~^~~")
|
||
self.assertSpecialized(lambda: 1/ 0/3,
|
||
"~^~~")
|
||
self.assertSpecialized(lambda: 1 / 0,
|
||
"~~~~~^~~~")
|
||
self.assertSpecialized(lambda: 1 / 0 / 5,
|
||
"~~~~~^~~~")
|
||
self.assertSpecialized(lambda: 1 /0,
|
||
"~~^~")
|
||
self.assertSpecialized(lambda: 1//0,
|
||
"~^^~")
|
||
self.assertSpecialized(lambda: 1//0//4,
|
||
"~^^~")
|
||
self.assertSpecialized(lambda: 1 // 0,
|
||
"~~^^~~")
|
||
self.assertSpecialized(lambda: 1 // 0 // 4,
|
||
"~~^^~~")
|
||
self.assertSpecialized(lambda: 1 //0,
|
||
"~~^^~")
|
||
self.assertSpecialized(lambda: 1// 0,
|
||
"~^^~~")
|
||
|
||
def test_decorator_application_lineno_correct(self):
|
||
def dec_error(func):
|
||
raise TypeError
|
||
def dec_fine(func):
|
||
return func
|
||
def applydecs():
|
||
@dec_error
|
||
@dec_fine
|
||
def g(): pass
|
||
result_lines = self.get_exception(applydecs)
|
||
lineno_applydescs = applydecs.__code__.co_firstlineno
|
||
lineno_dec_error = dec_error.__code__.co_firstlineno
|
||
expected_error = (
|
||
'Traceback (most recent call last):\n'
|
||
f' File "{__file__}", line {self.callable_line}, in get_exception\n'
|
||
' callable()\n'
|
||
f' File "{__file__}", line {lineno_applydescs + 1}, in applydecs\n'
|
||
' @dec_error\n'
|
||
' ^^^^^^^^^\n'
|
||
f' File "{__file__}", line {lineno_dec_error + 1}, in dec_error\n'
|
||
' raise TypeError\n'
|
||
)
|
||
self.assertEqual(result_lines, expected_error.splitlines())
|
||
|
||
def applydecs_class():
|
||
@dec_error
|
||
@dec_fine
|
||
class A: pass
|
||
result_lines = self.get_exception(applydecs_class)
|
||
lineno_applydescs_class = applydecs_class.__code__.co_firstlineno
|
||
expected_error = (
|
||
'Traceback (most recent call last):\n'
|
||
f' File "{__file__}", line {self.callable_line}, in get_exception\n'
|
||
' callable()\n'
|
||
f' File "{__file__}", line {lineno_applydescs_class + 1}, in applydecs_class\n'
|
||
' @dec_error\n'
|
||
' ^^^^^^^^^\n'
|
||
f' File "{__file__}", line {lineno_dec_error + 1}, in dec_error\n'
|
||
' raise TypeError\n'
|
||
)
|
||
self.assertEqual(result_lines, expected_error.splitlines())
|
||
|
||
def test_multiline_method_call_a(self):
|
||
def f():
|
||
(None
|
||
.method
|
||
)()
|
||
actual = self.get_exception(f)
|
||
expected = [
|
||
"Traceback (most recent call last):",
|
||
f" File \"{__file__}\", line {self.callable_line}, in get_exception",
|
||
" callable()",
|
||
f" File \"{__file__}\", line {f.__code__.co_firstlineno + 2}, in f",
|
||
" .method",
|
||
" ^^^^^^",
|
||
]
|
||
self.assertEqual(actual, expected)
|
||
|
||
def test_multiline_method_call_b(self):
|
||
def f():
|
||
(None.
|
||
method
|
||
)()
|
||
actual = self.get_exception(f)
|
||
expected = [
|
||
"Traceback (most recent call last):",
|
||
f" File \"{__file__}\", line {self.callable_line}, in get_exception",
|
||
" callable()",
|
||
f" File \"{__file__}\", line {f.__code__.co_firstlineno + 2}, in f",
|
||
" method",
|
||
]
|
||
self.assertEqual(actual, expected)
|
||
|
||
def test_multiline_method_call_c(self):
|
||
def f():
|
||
(None
|
||
. method
|
||
)()
|
||
actual = self.get_exception(f)
|
||
expected = [
|
||
"Traceback (most recent call last):",
|
||
f" File \"{__file__}\", line {self.callable_line}, in get_exception",
|
||
" callable()",
|
||
f" File \"{__file__}\", line {f.__code__.co_firstlineno + 2}, in f",
|
||
" . method",
|
||
" ^^^^^^",
|
||
]
|
||
self.assertEqual(actual, expected)
|
||
|
||
def test_wide_characters_unicode_with_problematic_byte_offset(self):
|
||
def f():
|
||
ｗｉｄｔｈ
|
||
|
||
actual = self.get_exception(f)
|
||
expected = [
|
||
"Traceback (most recent call last):",
|
||
f" File \"{__file__}\", line {self.callable_line}, in get_exception",
|
||
" callable()",
|
||
f" File \"{__file__}\", line {f.__code__.co_firstlineno + 1}, in f",
|
||
" ｗｉｄｔｈ",
|
||
]
|
||
self.assertEqual(actual, expected)
|
||
|
||
|
||
def test_byte_offset_with_wide_characters_middle(self):
|
||
def f():
|
||
ｗｉｄｔｈ = 1
|
||
raise ValueError(ｗｉｄｔｈ)
|
||
|
||
actual = self.get_exception(f)
|
||
expected = [
|
||
"Traceback (most recent call last):",
|
||
f" File \"{__file__}\", line {self.callable_line}, in get_exception",
|
||
" callable()",
|
||
f" File \"{__file__}\", line {f.__code__.co_firstlineno + 2}, in f",
|
||
" raise ValueError(ｗｉｄｔｈ)",
|
||
]
|
||
self.assertEqual(actual, expected)
|
||
|
||
def test_byte_offset_multiline(self):
|
||
def f():
|
||
ｗｗｗ = 1
|
||
ｔｈ = 0
|
||
|
||
print(1, ｗｗｗ(
|
||
ｔｈ))
|
||
|
||
actual = self.get_exception(f)
|
||
expected = [
|
||
"Traceback (most recent call last):",
|
||
f" File \"{__file__}\", line {self.callable_line}, in get_exception",
|
||
" callable()",
|
||
f" File \"{__file__}\", line {f.__code__.co_firstlineno + 4}, in f",
|
||
" print(1, ｗｗｗ(",
|
||
" ^^^^",
|
||
]
|
||
self.assertEqual(actual, expected)
|
||
|
||
|
||
|
||
@requires_debug_ranges()
|
||
class PurePythonTracebackErrorCaretTests(
|
||
PurePythonExceptionFormattingMixin,
|
||
TracebackErrorLocationCaretTestBase,
|
||
unittest.TestCase,
|
||
):
|
||
"""
|
||
Same set of tests as above using the pure Python implementation of
|
||
traceback printing in traceback.py.
|
||
"""
|
||
|
||
|
||
@cpython_only
|
||
@requires_debug_ranges()
|
||
class CPythonTracebackErrorCaretTests(
|
||
CAPIExceptionFormattingMixin,
|
||
TracebackErrorLocationCaretTestBase,
|
||
unittest.TestCase,
|
||
):
|
||
"""
|
||
Same set of tests as above but with Python's internal traceback printing.
|
||
"""
|
||
|
||
@cpython_only
|
||
@requires_debug_ranges()
|
||
class CPythonTracebackErrorCaretTests(
|
||
CAPIExceptionFormattingLegacyMixin,
|
||
TracebackErrorLocationCaretTestBase,
|
||
unittest.TestCase,
|
||
):
|
||
"""
|
||
Same set of tests as above but with Python's legacy internal traceback printing.
|
||
"""
|
||
|
||
class TracebackFormatTests(unittest.TestCase):
|
||
|
||
def some_exception(self):
|
||
raise KeyError('blah')
|
||
|
||
@cpython_only
|
||
def check_traceback_format(self, cleanup_func=None):
|
||
from _testcapi import traceback_print
|
||
try:
|
||
self.some_exception()
|
||
except KeyError as e:
|
||
tb = e.__traceback__
|
||
if cleanup_func is not None:
|
||
# Clear the inner frames, not this one
|
||
cleanup_func(tb.tb_next)
|
||
traceback_fmt = 'Traceback (most recent call last):\n' + \
|
||
''.join(traceback.format_tb(tb))
|
||
file_ = StringIO()
|
||
traceback_print(tb, file_)
|
||
python_fmt = file_.getvalue()
|
||
# Call all _tb and _exc functions
|
||
with captured_output("stderr") as tbstderr:
|
||
traceback.print_tb(tb)
|
||
tbfile = StringIO()
|
||
traceback.print_tb(tb, file=tbfile)
|
||
with captured_output("stderr") as excstderr:
|
||
traceback.print_exc()
|
||
excfmt = traceback.format_exc()
|
||
excfile = StringIO()
|
||
traceback.print_exc(file=excfile)
|
||
else:
|
||
raise Error("unable to create test traceback string")
|
||
|
||
# Make sure that Python and the traceback module format the same thing
|
||
self.assertEqual(traceback_fmt, python_fmt)
|
||
# Now verify the _tb func output
|
||
self.assertEqual(tbstderr.getvalue(), tbfile.getvalue())
|
||
# Now verify the _exc func output
|
||
self.assertEqual(excstderr.getvalue(), excfile.getvalue())
|
||
self.assertEqual(excfmt, excfile.getvalue())
|
||
|
||
# Make sure that the traceback is properly indented.
|
||
tb_lines = python_fmt.splitlines()
|
||
banner = tb_lines[0]
|
||
self.assertEqual(len(tb_lines), 5)
|
||
location, source_line = tb_lines[-2], tb_lines[-1]
|
||
self.assertTrue(banner.startswith('Traceback'))
|
||
self.assertTrue(location.startswith(' File'))
|
||
self.assertTrue(source_line.startswith(' raise'))
|
||
|
||
def test_traceback_format(self):
|
||
self.check_traceback_format()
|
||
|
||
def test_traceback_format_with_cleared_frames(self):
|
||
# Check that traceback formatting also works with a clear()ed frame
|
||
def cleanup_tb(tb):
|
||
tb.tb_frame.clear()
|
||
self.check_traceback_format(cleanup_tb)
|
||
|
||
def test_stack_format(self):
|
||
# Verify _stack functions. Note we have to use _getframe(1) to
|
||
# compare them without this frame appearing in the output
|
||
with captured_output("stderr") as ststderr:
|
||
traceback.print_stack(sys._getframe(1))
|
||
stfile = StringIO()
|
||
traceback.print_stack(sys._getframe(1), file=stfile)
|
||
self.assertEqual(ststderr.getvalue(), stfile.getvalue())
|
||
|
||
stfmt = traceback.format_stack(sys._getframe(1))
|
||
|
||
self.assertEqual(ststderr.getvalue(), "".join(stfmt))
|
||
|
||
def test_print_stack(self):
|
||
def prn():
|
||
traceback.print_stack()
|
||
with captured_output("stderr") as stderr:
|
||
prn()
|
||
lineno = prn.__code__.co_firstlineno
|
||
self.assertEqual(stderr.getvalue().splitlines()[-4:], [
|
||
' File "%s", line %d, in test_print_stack' % (__file__, lineno+3),
|
||
' prn()',
|
||
' File "%s", line %d, in prn' % (__file__, lineno+1),
|
||
' traceback.print_stack()',
|
||
])
|
||
|
||
# issue 26823 - Shrink recursive tracebacks
|
||
def _check_recursive_traceback_display(self, render_exc):
|
||
# Always show full diffs when this test fails
|
||
# Note that rearranging things may require adjusting
|
||
# the relative line numbers in the expected tracebacks
|
||
self.maxDiff = None
|
||
|
||
# Check hitting the recursion limit
|
||
def f():
|
||
f()
|
||
|
||
with captured_output("stderr") as stderr_f:
|
||
try:
|
||
f()
|
||
except RecursionError:
|
||
render_exc()
|
||
else:
|
||
self.fail("no recursion occurred")
|
||
|
||
lineno_f = f.__code__.co_firstlineno
|
||
result_f = (
|
||
'Traceback (most recent call last):\n'
|
||
f' File "{__file__}", line {lineno_f+5}, in _check_recursive_traceback_display\n'
|
||
' f()\n'
|
||
f' File "{__file__}", line {lineno_f+1}, in f\n'
|
||
' f()\n'
|
||
f' File "{__file__}", line {lineno_f+1}, in f\n'
|
||
' f()\n'
|
||
f' File "{__file__}", line {lineno_f+1}, in f\n'
|
||
' f()\n'
|
||
# XXX: The following line changes depending on whether the tests
|
||
# are run through the interactive interpreter or with -m
|
||
# It also varies depending on the platform (stack size)
|
||
# Fortunately, we don't care about exactness here, so we use regex
|
||
r' \[Previous line repeated (\d+) more times\]' '\n'
|
||
'RecursionError: maximum recursion depth exceeded\n'
|
||
)
|
||
|
||
expected = result_f.splitlines()
|
||
actual = stderr_f.getvalue().splitlines()
|
||
|
||
# Check the output text matches expectations
|
||
# 2nd last line contains the repetition count
|
||
self.assertEqual(actual[:-2], expected[:-2])
|
||
self.assertRegex(actual[-2], expected[-2])
|
||
# last line can have additional text appended
|
||
self.assertIn(expected[-1], actual[-1])
|
||
|
||
# Check the recursion count is roughly as expected
|
||
rec_limit = sys.getrecursionlimit()
|
||
self.assertIn(int(re.search(r"\d+", actual[-2]).group()), range(rec_limit-60, rec_limit))
|
||
|
||
# Check a known (limited) number of recursive invocations
|
||
def g(count=10):
|
||
if count:
|
||
return g(count-1)
|
||
raise ValueError
|
||
|
||
with captured_output("stderr") as stderr_g:
|
||
try:
|
||
g()
|
||
except ValueError:
|
||
render_exc()
|
||
else:
|
||
self.fail("no value error was raised")
|
||
|
||
lineno_g = g.__code__.co_firstlineno
|
||
result_g = (
|
||
f' File "{__file__}", line {lineno_g+2}, in g\n'
|
||
' return g(count-1)\n'
|
||
' ^^^^^^^^^^\n'
|
||
f' File "{__file__}", line {lineno_g+2}, in g\n'
|
||
' return g(count-1)\n'
|
||
' ^^^^^^^^^^\n'
|
||
f' File "{__file__}", line {lineno_g+2}, in g\n'
|
||
' return g(count-1)\n'
|
||
' ^^^^^^^^^^\n'
|
||
' [Previous line repeated 7 more times]\n'
|
||
f' File "{__file__}", line {lineno_g+3}, in g\n'
|
||
' raise ValueError\n'
|
||
'ValueError\n'
|
||
)
|
||
tb_line = (
|
||
'Traceback (most recent call last):\n'
|
||
f' File "{__file__}", line {lineno_g+7}, in _check_recursive_traceback_display\n'
|
||
' g()\n'
|
||
)
|
||
expected = (tb_line + result_g).splitlines()
|
||
actual = stderr_g.getvalue().splitlines()
|
||
self.assertEqual(actual, expected)
|
||
|
||
# Check 2 different repetitive sections
|
||
def h(count=10):
|
||
if count:
|
||
return h(count-1)
|
||
g()
|
||
|
||
with captured_output("stderr") as stderr_h:
|
||
try:
|
||
h()
|
||
except ValueError:
|
||
render_exc()
|
||
else:
|
||
self.fail("no value error was raised")
|
||
|
||
lineno_h = h.__code__.co_firstlineno
|
||
result_h = (
|
||
'Traceback (most recent call last):\n'
|
||
f' File "{__file__}", line {lineno_h+7}, in _check_recursive_traceback_display\n'
|
||
' h()\n'
|
||
f' File "{__file__}", line {lineno_h+2}, in h\n'
|
||
' return h(count-1)\n'
|
||
' ^^^^^^^^^^\n'
|
||
f' File "{__file__}", line {lineno_h+2}, in h\n'
|
||
' return h(count-1)\n'
|
||
' ^^^^^^^^^^\n'
|
||
f' File "{__file__}", line {lineno_h+2}, in h\n'
|
||
' return h(count-1)\n'
|
||
' ^^^^^^^^^^\n'
|
||
' [Previous line repeated 7 more times]\n'
|
||
f' File "{__file__}", line {lineno_h+3}, in h\n'
|
||
' g()\n'
|
||
)
|
||
expected = (result_h + result_g).splitlines()
|
||
actual = stderr_h.getvalue().splitlines()
|
||
self.assertEqual(actual, expected)
|
||
|
||
# Check the boundary conditions. First, test just below the cutoff.
|
||
with captured_output("stderr") as stderr_g:
|
||
try:
|
||
g(traceback._RECURSIVE_CUTOFF)
|
||
except ValueError:
|
||
render_exc()
|
||
else:
|
||
self.fail("no error raised")
|
||
result_g = (
|
||
f' File "{__file__}", line {lineno_g+2}, in g\n'
|
||
' return g(count-1)\n'
|
||
' ^^^^^^^^^^\n'
|
||
f' File "{__file__}", line {lineno_g+2}, in g\n'
|
||
' return g(count-1)\n'
|
||
' ^^^^^^^^^^\n'
|
||
f' File "{__file__}", line {lineno_g+2}, in g\n'
|
||
' return g(count-1)\n'
|
||
' ^^^^^^^^^^\n'
|
||
f' File "{__file__}", line {lineno_g+3}, in g\n'
|
||
' raise ValueError\n'
|
||
'ValueError\n'
|
||
)
|
||
tb_line = (
|
||
'Traceback (most recent call last):\n'
|
||
f' File "{__file__}", line {lineno_g+77}, in _check_recursive_traceback_display\n'
|
||
' g(traceback._RECURSIVE_CUTOFF)\n'
|
||
)
|
||
expected = (tb_line + result_g).splitlines()
|
||
actual = stderr_g.getvalue().splitlines()
|
||
self.assertEqual(actual, expected)
|
||
|
||
# Second, test just above the cutoff.
|
||
with captured_output("stderr") as stderr_g:
|
||
try:
|
||
g(traceback._RECURSIVE_CUTOFF + 1)
|
||
except ValueError:
|
||
render_exc()
|
||
else:
|
||
self.fail("no error raised")
|
||
result_g = (
|
||
f' File "{__file__}", line {lineno_g+2}, in g\n'
|
||
' return g(count-1)\n'
|
||
' ^^^^^^^^^^\n'
|
||
f' File "{__file__}", line {lineno_g+2}, in g\n'
|
||
' return g(count-1)\n'
|
||
' ^^^^^^^^^^\n'
|
||
f' File "{__file__}", line {lineno_g+2}, in g\n'
|
||
' return g(count-1)\n'
|
||
' ^^^^^^^^^^\n'
|
||
' [Previous line repeated 1 more time]\n'
|
||
f' File "{__file__}", line {lineno_g+3}, in g\n'
|
||
' raise ValueError\n'
|
||
'ValueError\n'
|
||
)
|
||
tb_line = (
|
||
'Traceback (most recent call last):\n'
|
||
f' File "{__file__}", line {lineno_g+108}, in _check_recursive_traceback_display\n'
|
||
' g(traceback._RECURSIVE_CUTOFF + 1)\n'
|
||
)
|
||
expected = (tb_line + result_g).splitlines()
|
||
actual = stderr_g.getvalue().splitlines()
|
||
self.assertEqual(actual, expected)
|
||
|
||
@requires_debug_ranges()
|
||
def test_recursive_traceback_python(self):
|
||
self._check_recursive_traceback_display(traceback.print_exc)
|
||
|
||
@cpython_only
|
||
@requires_debug_ranges()
|
||
def test_recursive_traceback_cpython_internal(self):
|
||
from _testcapi import exception_print
|
||
def render_exc():
|
||
exception_print(sys.exception())
|
||
self._check_recursive_traceback_display(render_exc)
|
||
|
||
def test_format_stack(self):
|
||
def fmt():
|
||
return traceback.format_stack()
|
||
result = fmt()
|
||
lineno = fmt.__code__.co_firstlineno
|
||
self.assertEqual(result[-2:], [
|
||
' File "%s", line %d, in test_format_stack\n'
|
||
' result = fmt()\n' % (__file__, lineno+2),
|
||
' File "%s", line %d, in fmt\n'
|
||
' return traceback.format_stack()\n' % (__file__, lineno+1),
|
||
])
|
||
|
||
@cpython_only
|
||
def test_unhashable(self):
|
||
from _testcapi import exception_print
|
||
|
||
class UnhashableException(Exception):
|
||
def __eq__(self, other):
|
||
return True
|
||
|
||
ex1 = UnhashableException('ex1')
|
||
ex2 = UnhashableException('ex2')
|
||
try:
|
||
raise ex2 from ex1
|
||
except UnhashableException:
|
||
try:
|
||
raise ex1
|
||
except UnhashableException as e:
|
||
exc_val = e
|
||
|
||
with captured_output("stderr") as stderr_f:
|
||
exception_print(exc_val)
|
||
|
||
tb = stderr_f.getvalue().strip().splitlines()
|
||
self.assertEqual(11, len(tb))
|
||
self.assertEqual(context_message.strip(), tb[5])
|
||
self.assertIn('UnhashableException: ex2', tb[3])
|
||
self.assertIn('UnhashableException: ex1', tb[10])
|
||
|
||
def deep_eg(self):
|
||
e = TypeError(1)
|
||
for i in range(2000):
|
||
e = ExceptionGroup('eg', [e])
|
||
return e
|
||
|
||
@cpython_only
|
||
def test_exception_group_deep_recursion_capi(self):
|
||
from _testcapi import exception_print
|
||
LIMIT = 75
|
||
eg = self.deep_eg()
|
||
with captured_output("stderr") as stderr_f:
|
||
with support.infinite_recursion(max_depth=LIMIT):
|
||
exception_print(eg)
|
||
output = stderr_f.getvalue()
|
||
self.assertIn('ExceptionGroup', output)
|
||
self.assertLessEqual(output.count('ExceptionGroup'), LIMIT)
|
||
|
||
def test_exception_group_deep_recursion_traceback(self):
|
||
LIMIT = 75
|
||
eg = self.deep_eg()
|
||
with captured_output("stderr") as stderr_f:
|
||
with support.infinite_recursion(max_depth=LIMIT):
|
||
traceback.print_exception(type(eg), eg, eg.__traceback__)
|
||
output = stderr_f.getvalue()
|
||
self.assertIn('ExceptionGroup', output)
|
||
self.assertLessEqual(output.count('ExceptionGroup'), LIMIT)
|
||
|
||
@cpython_only
|
||
def test_print_exception_bad_type_capi(self):
|
||
from _testcapi import exception_print
|
||
with captured_output("stderr") as stderr:
|
||
exception_print(42)
|
||
self.assertEqual(
|
||
stderr.getvalue(),
|
||
('TypeError: print_exception(): '
|
||
'Exception expected for value, int found\n')
|
||
)
|
||
|
||
def test_print_exception_bad_type_python(self):
|
||
msg = "Exception expected for value, int found"
|
||
with self.assertRaisesRegex(TypeError, msg):
|
||
traceback.print_exception(42)
|
||
|
||
|
||
cause_message = (
|
||
"\nThe above exception was the direct cause "
|
||
"of the following exception:\n\n")
|
||
|
||
context_message = (
|
||
"\nDuring handling of the above exception, "
|
||
"another exception occurred:\n\n")
|
||
|
||
boundaries = re.compile(
|
||
'(%s|%s)' % (re.escape(cause_message), re.escape(context_message)))
|
||
|
||
class BaseExceptionReportingTests:
|
||
|
||
def get_exception(self, exception_or_callable):
|
||
if isinstance(exception_or_callable, BaseException):
|
||
return exception_or_callable
|
||
try:
|
||
exception_or_callable()
|
||
except Exception as e:
|
||
return e
|
||
|
||
callable_line = get_exception.__code__.co_firstlineno + 4
|
||
|
||
def zero_div(self):
|
||
1/0 # In zero_div
|
||
|
||
def check_zero_div(self, msg):
|
||
lines = msg.splitlines()
|
||
if has_no_debug_ranges():
|
||
self.assertTrue(lines[-3].startswith(' File'))
|
||
self.assertIn('1/0 # In zero_div', lines[-2])
|
||
else:
|
||
self.assertTrue(lines[-4].startswith(' File'))
|
||
self.assertIn('1/0 # In zero_div', lines[-3])
|
||
self.assertTrue(lines[-1].startswith('ZeroDivisionError'), lines[-1])
|
||
|
||
def test_simple(self):
|
||
try:
|
||
1/0 # Marker
|
||
except ZeroDivisionError as _:
|
||
e = _
|
||
lines = self.get_report(e).splitlines()
|
||
if has_no_debug_ranges():
|
||
self.assertEqual(len(lines), 4)
|
||
self.assertTrue(lines[3].startswith('ZeroDivisionError'))
|
||
else:
|
||
self.assertEqual(len(lines), 5)
|
||
self.assertTrue(lines[4].startswith('ZeroDivisionError'))
|
||
self.assertTrue(lines[0].startswith('Traceback'))
|
||
self.assertTrue(lines[1].startswith(' File'))
|
||
self.assertIn('1/0 # Marker', lines[2])
|
||
|
||
def test_cause(self):
|
||
def inner_raise():
|
||
try:
|
||
self.zero_div()
|
||
except ZeroDivisionError as e:
|
||
raise KeyError from e
|
||
def outer_raise():
|
||
inner_raise() # Marker
|
||
blocks = boundaries.split(self.get_report(outer_raise))
|
||
self.assertEqual(len(blocks), 3)
|
||
self.assertEqual(blocks[1], cause_message)
|
||
self.check_zero_div(blocks[0])
|
||
self.assertIn('inner_raise() # Marker', blocks[2])
|
||
|
||
def test_context(self):
|
||
def inner_raise():
|
||
try:
|
||
self.zero_div()
|
||
except ZeroDivisionError:
|
||
raise KeyError
|
||
def outer_raise():
|
||
inner_raise() # Marker
|
||
blocks = boundaries.split(self.get_report(outer_raise))
|
||
self.assertEqual(len(blocks), 3)
|
||
self.assertEqual(blocks[1], context_message)
|
||
self.check_zero_div(blocks[0])
|
||
self.assertIn('inner_raise() # Marker', blocks[2])
|
||
|
||
def test_context_suppression(self):
|
||
try:
|
||
try:
|
||
raise Exception
|
||
except:
|
||
raise ZeroDivisionError from None
|
||
except ZeroDivisionError as _:
|
||
e = _
|
||
lines = self.get_report(e).splitlines()
|
||
self.assertEqual(len(lines), 4)
|
||
self.assertTrue(lines[3].startswith('ZeroDivisionError'))
|
||
self.assertTrue(lines[0].startswith('Traceback'))
|
||
self.assertTrue(lines[1].startswith(' File'))
|
||
self.assertIn('ZeroDivisionError from None', lines[2])
|
||
|
||
def test_cause_and_context(self):
|
||
# When both a cause and a context are set, only the cause should be
|
||
# displayed and the context should be muted.
|
||
def inner_raise():
|
||
try:
|
||
self.zero_div()
|
||
except ZeroDivisionError as _e:
|
||
e = _e
|
||
try:
|
||
xyzzy
|
||
except NameError:
|
||
raise KeyError from e
|
||
def outer_raise():
|
||
inner_raise() # Marker
|
||
blocks = boundaries.split(self.get_report(outer_raise))
|
||
self.assertEqual(len(blocks), 3)
|
||
self.assertEqual(blocks[1], cause_message)
|
||
self.check_zero_div(blocks[0])
|
||
self.assertIn('inner_raise() # Marker', blocks[2])
|
||
|
||
def test_cause_recursive(self):
|
||
def inner_raise():
|
||
try:
|
||
try:
|
||
self.zero_div()
|
||
except ZeroDivisionError as e:
|
||
z = e
|
||
raise KeyError from e
|
||
except KeyError as e:
|
||
raise z from e
|
||
def outer_raise():
|
||
inner_raise() # Marker
|
||
blocks = boundaries.split(self.get_report(outer_raise))
|
||
self.assertEqual(len(blocks), 3)
|
||
self.assertEqual(blocks[1], cause_message)
|
||
# The first block is the KeyError raised from the ZeroDivisionError
|
||
self.assertIn('raise KeyError from e', blocks[0])
|
||
self.assertNotIn('1/0', blocks[0])
|
||
# The second block (apart from the boundary) is the ZeroDivisionError
|
||
# re-raised from the KeyError
|
||
self.assertIn('inner_raise() # Marker', blocks[2])
|
||
self.check_zero_div(blocks[2])
|
||
|
||
def test_syntax_error_offset_at_eol(self):
|
||
# See #10186.
|
||
def e():
|
||
raise SyntaxError('', ('', 0, 5, 'hello'))
|
||
msg = self.get_report(e).splitlines()
|
||
self.assertEqual(msg[-2], " ^")
|
||
def e():
|
||
exec("x = 5 | 4 |")
|
||
msg = self.get_report(e).splitlines()
|
||
self.assertEqual(msg[-2], ' ^')
|
||
|
||
def test_syntax_error_no_lineno(self):
|
||
# See #34463.
|
||
|
||
# Without filename
|
||
e = SyntaxError('bad syntax')
|
||
msg = self.get_report(e).splitlines()
|
||
self.assertEqual(msg,
|
||
['SyntaxError: bad syntax'])
|
||
e.lineno = 100
|
||
msg = self.get_report(e).splitlines()
|
||
self.assertEqual(msg,
|
||
[' File "<string>", line 100', 'SyntaxError: bad syntax'])
|
||
|
||
# With filename
|
||
e = SyntaxError('bad syntax')
|
||
e.filename = 'myfile.py'
|
||
|
||
msg = self.get_report(e).splitlines()
|
||
self.assertEqual(msg,
|
||
['SyntaxError: bad syntax (myfile.py)'])
|
||
e.lineno = 100
|
||
msg = self.get_report(e).splitlines()
|
||
self.assertEqual(msg,
|
||
[' File "myfile.py", line 100', 'SyntaxError: bad syntax'])
|
||
|
||
def test_message_none(self):
|
||
# A message that looks like "None" should not be treated specially
|
||
err = self.get_report(Exception(None))
|
||
self.assertIn('Exception: None\n', err)
|
||
err = self.get_report(Exception('None'))
|
||
self.assertIn('Exception: None\n', err)
|
||
err = self.get_report(Exception())
|
||
self.assertIn('Exception\n', err)
|
||
err = self.get_report(Exception(''))
|
||
self.assertIn('Exception\n', err)
|
||
|
||
def test_syntax_error_various_offsets(self):
|
||
for offset in range(-5, 10):
|
||
for add in [0, 2]:
|
||
text = " "*add + "text%d" % offset
|
||
expected = [' File "file.py", line 1']
|
||
if offset < 1:
|
||
expected.append(" %s" % text.lstrip())
|
||
elif offset <= 6:
|
||
expected.append(" %s" % text.lstrip())
|
||
expected.append(" %s^" % (" "*(offset-1)))
|
||
else:
|
||
expected.append(" %s" % text.lstrip())
|
||
expected.append(" %s^" % (" "*5))
|
||
expected.append("SyntaxError: msg")
|
||
expected.append("")
|
||
err = self.get_report(SyntaxError("msg", ("file.py", 1, offset+add, text)))
|
||
exp = "\n".join(expected)
|
||
self.assertEqual(exp, err)
|
||
|
||
def test_exception_with_note(self):
|
||
e = ValueError(123)
|
||
vanilla = self.get_report(e)
|
||
|
||
e.add_note('My Note')
|
||
self.assertEqual(self.get_report(e), vanilla + 'My Note\n')
|
||
|
||
del e.__notes__
|
||
e.add_note('')
|
||
self.assertEqual(self.get_report(e), vanilla + '\n')
|
||
|
||
del e.__notes__
|
||
e.add_note('Your Note')
|
||
self.assertEqual(self.get_report(e), vanilla + 'Your Note\n')
|
||
|
||
del e.__notes__
|
||
self.assertEqual(self.get_report(e), vanilla)
|
||
|
||
def test_exception_with_invalid_notes(self):
|
||
e = ValueError(123)
|
||
vanilla = self.get_report(e)
|
||
|
||
# non-sequence __notes__
|
||
class BadThing:
|
||
def __str__(self):
|
||
return 'bad str'
|
||
|
||
def __repr__(self):
|
||
return 'bad repr'
|
||
|
||
# unprintable, non-sequence __notes__
|
||
class Unprintable:
|
||
def __repr__(self):
|
||
raise ValueError('bad value')
|
||
|
||
e.__notes__ = BadThing()
|
||
notes_repr = 'bad repr'
|
||
self.assertEqual(self.get_report(e), vanilla + notes_repr + '\n')
|
||
|
||
e.__notes__ = Unprintable()
|
||
err_msg = '<__notes__ repr() failed>'
|
||
self.assertEqual(self.get_report(e), vanilla + err_msg + '\n')
|
||
|
||
# non-string item in the __notes__ sequence
|
||
e.__notes__ = [BadThing(), 'Final Note']
|
||
bad_note = 'bad str'
|
||
self.assertEqual(self.get_report(e), vanilla + bad_note + '\nFinal Note\n')
|
||
|
||
# unprintable, non-string item in the __notes__ sequence
|
||
e.__notes__ = [Unprintable(), 'Final Note']
|
||
err_msg = '<note str() failed>'
|
||
self.assertEqual(self.get_report(e), vanilla + err_msg + '\nFinal Note\n')
|
||
|
||
e.__notes__ = "please do not explode me"
|
||
err_msg = "'please do not explode me'"
|
||
self.assertEqual(self.get_report(e), vanilla + err_msg + '\n')
|
||
|
||
e.__notes__ = b"please do not show me as numbers"
|
||
err_msg = "b'please do not show me as numbers'"
|
||
self.assertEqual(self.get_report(e), vanilla + err_msg + '\n')
|
||
|
||
def test_exception_with_note_with_multiple_notes(self):
|
||
e = ValueError(42)
|
||
vanilla = self.get_report(e)
|
||
|
||
e.add_note('Note 1')
|
||
e.add_note('Note 2')
|
||
e.add_note('Note 3')
|
||
|
||
self.assertEqual(
|
||
self.get_report(e),
|
||
vanilla + 'Note 1\n' + 'Note 2\n' + 'Note 3\n')
|
||
|
||
del e.__notes__
|
||
e.add_note('Note 4')
|
||
del e.__notes__
|
||
e.add_note('Note 5')
|
||
e.add_note('Note 6')
|
||
|
||
self.assertEqual(
|
||
self.get_report(e),
|
||
vanilla + 'Note 5\n' + 'Note 6\n')
|
||
|
||
def test_exception_qualname(self):
|
||
class A:
|
||
class B:
|
||
class X(Exception):
|
||
def __str__(self):
|
||
return "I am X"
|
||
|
||
err = self.get_report(A.B.X())
|
||
str_value = 'I am X'
|
||
str_name = '.'.join([A.B.X.__module__, A.B.X.__qualname__])
|
||
exp = "%s: %s\n" % (str_name, str_value)
|
||
self.assertEqual(exp, MODULE_PREFIX + err)
|
||
|
||
def test_exception_modulename(self):
|
||
class X(Exception):
|
||
def __str__(self):
|
||
return "I am X"
|
||
|
||
for modulename in '__main__', 'builtins', 'some_module':
|
||
X.__module__ = modulename
|
||
with self.subTest(modulename=modulename):
|
||
err = self.get_report(X())
|
||
str_value = 'I am X'
|
||
if modulename in ['builtins', '__main__']:
|
||
str_name = X.__qualname__
|
||
else:
|
||
str_name = '.'.join([X.__module__, X.__qualname__])
|
||
exp = "%s: %s\n" % (str_name, str_value)
|
||
self.assertEqual(exp, err)
|
||
|
||
def test_exception_angle_bracketed_filename(self):
|
||
src = textwrap.dedent("""
|
||
try:
|
||
raise ValueError(42)
|
||
except Exception as e:
|
||
exc = e
|
||
""")
|
||
|
||
code = compile(src, "<does not exist>", "exec")
|
||
g, l = {}, {}
|
||
exec(code, g, l)
|
||
err = self.get_report(l['exc'])
|
||
exp = ' File "<does not exist>", line 3, in <module>\nValueError: 42\n'
|
||
self.assertIn(exp, err)
|
||
|
||
def test_exception_modulename_not_unicode(self):
|
||
class X(Exception):
|
||
def __str__(self):
|
||
return "I am X"
|
||
|
||
X.__module__ = 42
|
||
|
||
err = self.get_report(X())
|
||
exp = f'<unknown>.{X.__qualname__}: I am X\n'
|
||
self.assertEqual(exp, err)
|
||
|
||
def test_exception_bad__str__(self):
|
||
class X(Exception):
|
||
def __str__(self):
|
||
1/0
|
||
err = self.get_report(X())
|
||
str_value = '<exception str() failed>'
|
||
str_name = '.'.join([X.__module__, X.__qualname__])
|
||
self.assertEqual(MODULE_PREFIX + err, f"{str_name}: {str_value}\n")
|
||
|
||
|
||
# #### Exception Groups ####
|
||
|
||
def test_exception_group_basic(self):
|
||
def exc():
|
||
raise ExceptionGroup("eg", [ValueError(1), TypeError(2)])
|
||
|
||
expected = (
|
||
f' + Exception Group Traceback (most recent call last):\n'
|
||
f' | File "{__file__}", line {self.callable_line}, in get_exception\n'
|
||
f' | exception_or_callable()\n'
|
||
f' | File "{__file__}", line {exc.__code__.co_firstlineno + 1}, in exc\n'
|
||
f' | raise ExceptionGroup("eg", [ValueError(1), TypeError(2)])\n'
|
||
f' | ExceptionGroup: eg (2 sub-exceptions)\n'
|
||
f' +-+---------------- 1 ----------------\n'
|
||
f' | ValueError: 1\n'
|
||
f' +---------------- 2 ----------------\n'
|
||
f' | TypeError: 2\n'
|
||
f' +------------------------------------\n')
|
||
|
||
report = self.get_report(exc)
|
||
self.assertEqual(report, expected)
|
||
|
||
def test_exception_group_cause(self):
|
||
def exc():
|
||
EG = ExceptionGroup
|
||
try:
|
||
raise EG("eg1", [ValueError(1), TypeError(2)])
|
||
except Exception as e:
|
||
raise EG("eg2", [ValueError(3), TypeError(4)]) from e
|
||
|
||
expected = (f' + Exception Group Traceback (most recent call last):\n'
|
||
f' | File "{__file__}", line {exc.__code__.co_firstlineno + 3}, in exc\n'
|
||
f' | raise EG("eg1", [ValueError(1), TypeError(2)])\n'
|
||
f' | ExceptionGroup: eg1 (2 sub-exceptions)\n'
|
||
f' +-+---------------- 1 ----------------\n'
|
||
f' | ValueError: 1\n'
|
||
f' +---------------- 2 ----------------\n'
|
||
f' | TypeError: 2\n'
|
||
f' +------------------------------------\n'
|
||
f'\n'
|
||
f'The above exception was the direct cause of the following exception:\n'
|
||
f'\n'
|
||
f' + Exception Group Traceback (most recent call last):\n'
|
||
f' | File "{__file__}", line {self.callable_line}, in get_exception\n'
|
||
f' | exception_or_callable()\n'
|
||
f' | File "{__file__}", line {exc.__code__.co_firstlineno + 5}, in exc\n'
|
||
f' | raise EG("eg2", [ValueError(3), TypeError(4)]) from e\n'
|
||
f' | ExceptionGroup: eg2 (2 sub-exceptions)\n'
|
||
f' +-+---------------- 1 ----------------\n'
|
||
f' | ValueError: 3\n'
|
||
f' +---------------- 2 ----------------\n'
|
||
f' | TypeError: 4\n'
|
||
f' +------------------------------------\n')
|
||
|
||
report = self.get_report(exc)
|
||
self.assertEqual(report, expected)
|
||
|
||
def test_exception_group_context_with_context(self):
|
||
def exc():
|
||
EG = ExceptionGroup
|
||
try:
|
||
try:
|
||
raise EG("eg1", [ValueError(1), TypeError(2)])
|
||
except:
|
||
raise EG("eg2", [ValueError(3), TypeError(4)])
|
||
except:
|
||
raise ImportError(5)
|
||
|
||
expected = (
|
||
f' + Exception Group Traceback (most recent call last):\n'
|
||
f' | File "{__file__}", line {exc.__code__.co_firstlineno + 4}, in exc\n'
|
||
f' | raise EG("eg1", [ValueError(1), TypeError(2)])\n'
|
||
f' | ExceptionGroup: eg1 (2 sub-exceptions)\n'
|
||
f' +-+---------------- 1 ----------------\n'
|
||
f' | ValueError: 1\n'
|
||
f' +---------------- 2 ----------------\n'
|
||
f' | TypeError: 2\n'
|
||
f' +------------------------------------\n'
|
||
f'\n'
|
||
f'During handling of the above exception, another exception occurred:\n'
|
||
f'\n'
|
||
f' + Exception Group Traceback (most recent call last):\n'
|
||
f' | File "{__file__}", line {exc.__code__.co_firstlineno + 6}, in exc\n'
|
||
f' | raise EG("eg2", [ValueError(3), TypeError(4)])\n'
|
||
f' | ExceptionGroup: eg2 (2 sub-exceptions)\n'
|
||
f' +-+---------------- 1 ----------------\n'
|
||
f' | ValueError: 3\n'
|
||
f' +---------------- 2 ----------------\n'
|
||
f' | TypeError: 4\n'
|
||
f' +------------------------------------\n'
|
||
f'\n'
|
||
f'During handling of the above exception, another exception occurred:\n'
|
||
f'\n'
|
||
f'Traceback (most recent call last):\n'
|
||
f' File "{__file__}", line {self.callable_line}, in get_exception\n'
|
||
f' exception_or_callable()\n'
|
||
f' File "{__file__}", line {exc.__code__.co_firstlineno + 8}, in exc\n'
|
||
f' raise ImportError(5)\n'
|
||
f'ImportError: 5\n')
|
||
|
||
report = self.get_report(exc)
|
||
self.assertEqual(report, expected)
|
||
|
||
def test_exception_group_nested(self):
|
||
def exc():
|
||
EG = ExceptionGroup
|
||
VE = ValueError
|
||
TE = TypeError
|
||
try:
|
||
try:
|
||
raise EG("nested", [TE(2), TE(3)])
|
||
except Exception as e:
|
||
exc = e
|
||
raise EG("eg", [VE(1), exc, VE(4)])
|
||
except:
|
||
raise EG("top", [VE(5)])
|
||
|
||
expected = (f' + Exception Group Traceback (most recent call last):\n'
|
||
f' | File "{__file__}", line {exc.__code__.co_firstlineno + 9}, in exc\n'
|
||
f' | raise EG("eg", [VE(1), exc, VE(4)])\n'
|
||
f' | ExceptionGroup: eg (3 sub-exceptions)\n'
|
||
f' +-+---------------- 1 ----------------\n'
|
||
f' | ValueError: 1\n'
|
||
f' +---------------- 2 ----------------\n'
|
||
f' | Exception Group Traceback (most recent call last):\n'
|
||
f' | File "{__file__}", line {exc.__code__.co_firstlineno + 6}, in exc\n'
|
||
f' | raise EG("nested", [TE(2), TE(3)])\n'
|
||
f' | ExceptionGroup: nested (2 sub-exceptions)\n'
|
||
f' +-+---------------- 1 ----------------\n'
|
||
f' | TypeError: 2\n'
|
||
f' +---------------- 2 ----------------\n'
|
||
f' | TypeError: 3\n'
|
||
f' +------------------------------------\n'
|
||
f' +---------------- 3 ----------------\n'
|
||
f' | ValueError: 4\n'
|
||
f' +------------------------------------\n'
|
||
f'\n'
|
||
f'During handling of the above exception, another exception occurred:\n'
|
||
f'\n'
|
||
f' + Exception Group Traceback (most recent call last):\n'
|
||
f' | File "{__file__}", line {self.callable_line}, in get_exception\n'
|
||
f' | exception_or_callable()\n'
|
||
f' | File "{__file__}", line {exc.__code__.co_firstlineno + 11}, in exc\n'
|
||
f' | raise EG("top", [VE(5)])\n'
|
||
f' | ExceptionGroup: top (1 sub-exception)\n'
|
||
f' +-+---------------- 1 ----------------\n'
|
||
f' | ValueError: 5\n'
|
||
f' +------------------------------------\n')
|
||
|
||
report = self.get_report(exc)
|
||
self.assertEqual(report, expected)
|
||
|
||
def test_exception_group_width_limit(self):
|
||
excs = []
|
||
for i in range(1000):
|
||
excs.append(ValueError(i))
|
||
eg = ExceptionGroup('eg', excs)
|
||
|
||
expected = (' | ExceptionGroup: eg (1000 sub-exceptions)\n'
|
||
' +-+---------------- 1 ----------------\n'
|
||
' | ValueError: 0\n'
|
||
' +---------------- 2 ----------------\n'
|
||
' | ValueError: 1\n'
|
||
' +---------------- 3 ----------------\n'
|
||
' | ValueError: 2\n'
|
||
' +---------------- 4 ----------------\n'
|
||
' | ValueError: 3\n'
|
||
' +---------------- 5 ----------------\n'
|
||
' | ValueError: 4\n'
|
||
' +---------------- 6 ----------------\n'
|
||
' | ValueError: 5\n'
|
||
' +---------------- 7 ----------------\n'
|
||
' | ValueError: 6\n'
|
||
' +---------------- 8 ----------------\n'
|
||
' | ValueError: 7\n'
|
||
' +---------------- 9 ----------------\n'
|
||
' | ValueError: 8\n'
|
||
' +---------------- 10 ----------------\n'
|
||
' | ValueError: 9\n'
|
||
' +---------------- 11 ----------------\n'
|
||
' | ValueError: 10\n'
|
||
' +---------------- 12 ----------------\n'
|
||
' | ValueError: 11\n'
|
||
' +---------------- 13 ----------------\n'
|
||
' | ValueError: 12\n'
|
||
' +---------------- 14 ----------------\n'
|
||
' | ValueError: 13\n'
|
||
' +---------------- 15 ----------------\n'
|
||
' | ValueError: 14\n'
|
||
' +---------------- ... ----------------\n'
|
||
' | and 985 more exceptions\n'
|
||
' +------------------------------------\n')
|
||
|
||
report = self.get_report(eg)
|
||
self.assertEqual(report, expected)
|
||
|
||
def test_exception_group_depth_limit(self):
|
||
exc = TypeError('bad type')
|
||
for i in range(1000):
|
||
exc = ExceptionGroup(
|
||
f'eg{i}',
|
||
[ValueError(i), exc, ValueError(-i)])
|
||
|
||
expected = (' | ExceptionGroup: eg999 (3 sub-exceptions)\n'
|
||
' +-+---------------- 1 ----------------\n'
|
||
' | ValueError: 999\n'
|
||
' +---------------- 2 ----------------\n'
|
||
' | ExceptionGroup: eg998 (3 sub-exceptions)\n'
|
||
' +-+---------------- 1 ----------------\n'
|
||
' | ValueError: 998\n'
|
||
' +---------------- 2 ----------------\n'
|
||
' | ExceptionGroup: eg997 (3 sub-exceptions)\n'
|
||
' +-+---------------- 1 ----------------\n'
|
||
' | ValueError: 997\n'
|
||
' +---------------- 2 ----------------\n'
|
||
' | ExceptionGroup: eg996 (3 sub-exceptions)\n'
|
||
' +-+---------------- 1 ----------------\n'
|
||
' | ValueError: 996\n'
|
||
' +---------------- 2 ----------------\n'
|
||
' | ExceptionGroup: eg995 (3 sub-exceptions)\n'
|
||
' +-+---------------- 1 ----------------\n'
|
||
' | ValueError: 995\n'
|
||
' +---------------- 2 ----------------\n'
|
||
' | ExceptionGroup: eg994 (3 sub-exceptions)\n'
|
||
' +-+---------------- 1 ----------------\n'
|
||
' | ValueError: 994\n'
|
||
' +---------------- 2 ----------------\n'
|
||
' | ExceptionGroup: eg993 (3 sub-exceptions)\n'
|
||
' +-+---------------- 1 ----------------\n'
|
||
' | ValueError: 993\n'
|
||
' +---------------- 2 ----------------\n'
|
||
' | ExceptionGroup: eg992 (3 sub-exceptions)\n'
|
||
' +-+---------------- 1 ----------------\n'
|
||
' | ValueError: 992\n'
|
||
' +---------------- 2 ----------------\n'
|
||
' | ExceptionGroup: eg991 (3 sub-exceptions)\n'
|
||
' +-+---------------- 1 ----------------\n'
|
||
' | ValueError: 991\n'
|
||
' +---------------- 2 ----------------\n'
|
||
' | ExceptionGroup: eg990 (3 sub-exceptions)\n'
|
||
' +-+---------------- 1 ----------------\n'
|
||
' | ValueError: 990\n'
|
||
' +---------------- 2 ----------------\n'
|
||
' | ... (max_group_depth is 10)\n'
|
||
' +---------------- 3 ----------------\n'
|
||
' | ValueError: -990\n'
|
||
' +------------------------------------\n'
|
||
' +---------------- 3 ----------------\n'
|
||
' | ValueError: -991\n'
|
||
' +------------------------------------\n'
|
||
' +---------------- 3 ----------------\n'
|
||
' | ValueError: -992\n'
|
||
' +------------------------------------\n'
|
||
' +---------------- 3 ----------------\n'
|
||
' | ValueError: -993\n'
|
||
' +------------------------------------\n'
|
||
' +---------------- 3 ----------------\n'
|
||
' | ValueError: -994\n'
|
||
' +------------------------------------\n'
|
||
' +---------------- 3 ----------------\n'
|
||
' | ValueError: -995\n'
|
||
' +------------------------------------\n'
|
||
' +---------------- 3 ----------------\n'
|
||
' | ValueError: -996\n'
|
||
' +------------------------------------\n'
|
||
' +---------------- 3 ----------------\n'
|
||
' | ValueError: -997\n'
|
||
' +------------------------------------\n'
|
||
' +---------------- 3 ----------------\n'
|
||
' | ValueError: -998\n'
|
||
' +------------------------------------\n'
|
||
' +---------------- 3 ----------------\n'
|
||
' | ValueError: -999\n'
|
||
' +------------------------------------\n')
|
||
|
||
report = self.get_report(exc)
|
||
self.assertEqual(report, expected)
|
||
|
||
def test_exception_group_with_notes(self):
|
||
def exc():
|
||
try:
|
||
excs = []
|
||
for msg in ['bad value', 'terrible value']:
|
||
try:
|
||
raise ValueError(msg)
|
||
except ValueError as e:
|
||
e.add_note(f'the {msg}')
|
||
excs.append(e)
|
||
raise ExceptionGroup("nested", excs)
|
||
except ExceptionGroup as e:
|
||
e.add_note(('>> Multi line note\n'
|
||
'>> Because I am such\n'
|
||
'>> an important exception.\n'
|
||
'>> empty lines work too\n'
|
||
'\n'
|
||
'(that was an empty line)'))
|
||
raise
|
||
|
||
expected = (f' + Exception Group Traceback (most recent call last):\n'
|
||
f' | File "{__file__}", line {self.callable_line}, in get_exception\n'
|
||
f' | exception_or_callable()\n'
|
||
f' | File "{__file__}", line {exc.__code__.co_firstlineno + 9}, in exc\n'
|
||
f' | raise ExceptionGroup("nested", excs)\n'
|
||
f' | ExceptionGroup: nested (2 sub-exceptions)\n'
|
||
f' | >> Multi line note\n'
|
||
f' | >> Because I am such\n'
|
||
f' | >> an important exception.\n'
|
||
f' | >> empty lines work too\n'
|
||
f' | \n'
|
||
f' | (that was an empty line)\n'
|
||
f' +-+---------------- 1 ----------------\n'
|
||
f' | Traceback (most recent call last):\n'
|
||
f' | File "{__file__}", line {exc.__code__.co_firstlineno + 5}, in exc\n'
|
||
f' | raise ValueError(msg)\n'
|
||
f' | ValueError: bad value\n'
|
||
f' | the bad value\n'
|
||
f' +---------------- 2 ----------------\n'
|
||
f' | Traceback (most recent call last):\n'
|
||
f' | File "{__file__}", line {exc.__code__.co_firstlineno + 5}, in exc\n'
|
||
f' | raise ValueError(msg)\n'
|
||
f' | ValueError: terrible value\n'
|
||
f' | the terrible value\n'
|
||
f' +------------------------------------\n')
|
||
|
||
report = self.get_report(exc)
|
||
self.assertEqual(report, expected)
|
||
|
||
def test_exception_group_with_multiple_notes(self):
|
||
def exc():
|
||
try:
|
||
excs = []
|
||
for msg in ['bad value', 'terrible value']:
|
||
try:
|
||
raise ValueError(msg)
|
||
except ValueError as e:
|
||
e.add_note(f'the {msg}')
|
||
e.add_note(f'Goodbye {msg}')
|
||
excs.append(e)
|
||
raise ExceptionGroup("nested", excs)
|
||
except ExceptionGroup as e:
|
||
e.add_note(('>> Multi line note\n'
|
||
'>> Because I am such\n'
|
||
'>> an important exception.\n'
|
||
'>> empty lines work too\n'
|
||
'\n'
|
||
'(that was an empty line)'))
|
||
e.add_note('Goodbye!')
|
||
raise
|
||
|
||
expected = (f' + Exception Group Traceback (most recent call last):\n'
|
||
f' | File "{__file__}", line {self.callable_line}, in get_exception\n'
|
||
f' | exception_or_callable()\n'
|
||
f' | File "{__file__}", line {exc.__code__.co_firstlineno + 10}, in exc\n'
|
||
f' | raise ExceptionGroup("nested", excs)\n'
|
||
f' | ExceptionGroup: nested (2 sub-exceptions)\n'
|
||
f' | >> Multi line note\n'
|
||
f' | >> Because I am such\n'
|
||
f' | >> an important exception.\n'
|
||
f' | >> empty lines work too\n'
|
||
f' | \n'
|
||
f' | (that was an empty line)\n'
|
||
f' | Goodbye!\n'
|
||
f' +-+---------------- 1 ----------------\n'
|
||
f' | Traceback (most recent call last):\n'
|
||
f' | File "{__file__}", line {exc.__code__.co_firstlineno + 5}, in exc\n'
|
||
f' | raise ValueError(msg)\n'
|
||
f' | ValueError: bad value\n'
|
||
f' | the bad value\n'
|
||
f' | Goodbye bad value\n'
|
||
f' +---------------- 2 ----------------\n'
|
||
f' | Traceback (most recent call last):\n'
|
||
f' | File "{__file__}", line {exc.__code__.co_firstlineno + 5}, in exc\n'
|
||
f' | raise ValueError(msg)\n'
|
||
f' | ValueError: terrible value\n'
|
||
f' | the terrible value\n'
|
||
f' | Goodbye terrible value\n'
|
||
f' +------------------------------------\n')
|
||
|
||
report = self.get_report(exc)
|
||
self.assertEqual(report, expected)
|
||
|
||
def test_KeyboardInterrupt_at_first_line_of_frame(self):
|
||
# see GH-93249
|
||
def f():
|
||
return sys._getframe()
|
||
|
||
tb_next = None
|
||
frame = f()
|
||
lasti = 0
|
||
lineno = f.__code__.co_firstlineno
|
||
tb = types.TracebackType(tb_next, frame, lasti, lineno)
|
||
|
||
exc = KeyboardInterrupt()
|
||
exc.__traceback__ = tb
|
||
|
||
expected = (f'Traceback (most recent call last):\n'
|
||
f' File "{__file__}", line {lineno}, in f\n'
|
||
f' def f():\n'
|
||
f'\n'
|
||
f'KeyboardInterrupt\n')
|
||
|
||
report = self.get_report(exc)
|
||
# remove trailing writespace:
|
||
report = '\n'.join([l.rstrip() for l in report.split('\n')])
|
||
self.assertEqual(report, expected)
|
||
|
||
|
||
class PyExcReportingTests(BaseExceptionReportingTests, unittest.TestCase):
|
||
#
|
||
# This checks reporting through the 'traceback' module, with both
|
||
# format_exception() and print_exception().
|
||
#
|
||
|
||
def get_report(self, e):
|
||
e = self.get_exception(e)
|
||
s = ''.join(
|
||
traceback.format_exception(type(e), e, e.__traceback__))
|
||
with captured_output("stderr") as sio:
|
||
traceback.print_exception(type(e), e, e.__traceback__)
|
||
self.assertEqual(sio.getvalue(), s)
|
||
return s
|
||
|
||
|
||
class CExcReportingTests(BaseExceptionReportingTests, unittest.TestCase):
|
||
#
|
||
# This checks built-in reporting by the interpreter.
|
||
#
|
||
|
||
@cpython_only
|
||
def get_report(self, e):
|
||
from _testcapi import exception_print
|
||
e = self.get_exception(e)
|
||
with captured_output("stderr") as s:
|
||
exception_print(e)
|
||
return s.getvalue()
|
||
|
||
|
||
class LimitTests(unittest.TestCase):
|
||
|
||
''' Tests for limit argument.
|
||
It's enough to test extact_tb, extract_stack and format_exception '''
|
||
|
||
def last_raises1(self):
|
||
raise Exception('Last raised')
|
||
|
||
def last_raises2(self):
|
||
self.last_raises1()
|
||
|
||
def last_raises3(self):
|
||
self.last_raises2()
|
||
|
||
def last_raises4(self):
|
||
self.last_raises3()
|
||
|
||
def last_raises5(self):
|
||
self.last_raises4()
|
||
|
||
def last_returns_frame1(self):
|
||
return sys._getframe()
|
||
|
||
def last_returns_frame2(self):
|
||
return self.last_returns_frame1()
|
||
|
||
def last_returns_frame3(self):
|
||
return self.last_returns_frame2()
|
||
|
||
def last_returns_frame4(self):
|
||
return self.last_returns_frame3()
|
||
|
||
def last_returns_frame5(self):
|
||
return self.last_returns_frame4()
|
||
|
||
def test_extract_stack(self):
|
||
frame = self.last_returns_frame5()
|
||
def extract(**kwargs):
|
||
return traceback.extract_stack(frame, **kwargs)
|
||
def assertEqualExcept(actual, expected, ignore):
|
||
self.assertEqual(actual[:ignore], expected[:ignore])
|
||
self.assertEqual(actual[ignore+1:], expected[ignore+1:])
|
||
self.assertEqual(len(actual), len(expected))
|
||
|
||
with support.swap_attr(sys, 'tracebacklimit', 1000):
|
||
nolim = extract()
|
||
self.assertGreater(len(nolim), 5)
|
||
self.assertEqual(extract(limit=2), nolim[-2:])
|
||
assertEqualExcept(extract(limit=100), nolim[-100:], -5-1)
|
||
self.assertEqual(extract(limit=-2), nolim[:2])
|
||
assertEqualExcept(extract(limit=-100), nolim[:100], len(nolim)-5-1)
|
||
self.assertEqual(extract(limit=0), [])
|
||
del sys.tracebacklimit
|
||
assertEqualExcept(extract(), nolim, -5-1)
|
||
sys.tracebacklimit = 2
|
||
self.assertEqual(extract(), nolim[-2:])
|
||
self.assertEqual(extract(limit=3), nolim[-3:])
|
||
self.assertEqual(extract(limit=-3), nolim[:3])
|
||
sys.tracebacklimit = 0
|
||
self.assertEqual(extract(), [])
|
||
sys.tracebacklimit = -1
|
||
self.assertEqual(extract(), [])
|
||
|
||
def test_extract_tb(self):
|
||
try:
|
||
self.last_raises5()
|
||
except Exception as e:
|
||
tb = e.__traceback__
|
||
def extract(**kwargs):
|
||
return traceback.extract_tb(tb, **kwargs)
|
||
|
||
with support.swap_attr(sys, 'tracebacklimit', 1000):
|
||
nolim = extract()
|
||
self.assertEqual(len(nolim), 5+1)
|
||
self.assertEqual(extract(limit=2), nolim[:2])
|
||
self.assertEqual(extract(limit=10), nolim)
|
||
self.assertEqual(extract(limit=-2), nolim[-2:])
|
||
self.assertEqual(extract(limit=-10), nolim)
|
||
self.assertEqual(extract(limit=0), [])
|
||
del sys.tracebacklimit
|
||
self.assertEqual(extract(), nolim)
|
||
sys.tracebacklimit = 2
|
||
self.assertEqual(extract(), nolim[:2])
|
||
self.assertEqual(extract(limit=3), nolim[:3])
|
||
self.assertEqual(extract(limit=-3), nolim[-3:])
|
||
sys.tracebacklimit = 0
|
||
self.assertEqual(extract(), [])
|
||
sys.tracebacklimit = -1
|
||
self.assertEqual(extract(), [])
|
||
|
||
def test_format_exception(self):
|
||
try:
|
||
self.last_raises5()
|
||
except Exception as e:
|
||
exc = e
|
||
# [1:-1] to exclude "Traceback (...)" header and
|
||
# exception type and value
|
||
def extract(**kwargs):
|
||
return traceback.format_exception(exc, **kwargs)[1:-1]
|
||
|
||
with support.swap_attr(sys, 'tracebacklimit', 1000):
|
||
nolim = extract()
|
||
self.assertEqual(len(nolim), 5+1)
|
||
self.assertEqual(extract(limit=2), nolim[:2])
|
||
self.assertEqual(extract(limit=10), nolim)
|
||
self.assertEqual(extract(limit=-2), nolim[-2:])
|
||
self.assertEqual(extract(limit=-10), nolim)
|
||
self.assertEqual(extract(limit=0), [])
|
||
del sys.tracebacklimit
|
||
self.assertEqual(extract(), nolim)
|
||
sys.tracebacklimit = 2
|
||
self.assertEqual(extract(), nolim[:2])
|
||
self.assertEqual(extract(limit=3), nolim[:3])
|
||
self.assertEqual(extract(limit=-3), nolim[-3:])
|
||
sys.tracebacklimit = 0
|
||
self.assertEqual(extract(), [])
|
||
sys.tracebacklimit = -1
|
||
self.assertEqual(extract(), [])
|
||
|
||
|
||
class MiscTracebackCases(unittest.TestCase):
|
||
#
|
||
# Check non-printing functions in traceback module
|
||
#
|
||
|
||
def test_clear(self):
|
||
def outer():
|
||
middle()
|
||
def middle():
|
||
inner()
|
||
def inner():
|
||
i = 1
|
||
1/0
|
||
|
||
try:
|
||
outer()
|
||
except BaseException as e:
|
||
tb = e.__traceback__
|
||
|
||
# Initial assertion: there's one local in the inner frame.
|
||
inner_frame = tb.tb_next.tb_next.tb_next.tb_frame
|
||
self.assertEqual(len(inner_frame.f_locals), 1)
|
||
|
||
# Clear traceback frames
|
||
traceback.clear_frames(tb)
|
||
|
||
# Local variable dict should now be empty.
|
||
self.assertEqual(len(inner_frame.f_locals), 0)
|
||
|
||
def test_extract_stack(self):
|
||
def extract():
|
||
return traceback.extract_stack()
|
||
result = extract()
|
||
lineno = extract.__code__.co_firstlineno
|
||
self.assertEqual(result[-2:], [
|
||
(__file__, lineno+2, 'test_extract_stack', 'result = extract()'),
|
||
(__file__, lineno+1, 'extract', 'return traceback.extract_stack()'),
|
||
])
|
||
self.assertEqual(len(result[0]), 4)
|
||
|
||
|
||
class TestFrame(unittest.TestCase):
|
||
|
||
def test_basics(self):
|
||
linecache.clearcache()
|
||
linecache.lazycache("f", globals())
|
||
f = traceback.FrameSummary("f", 1, "dummy")
|
||
self.assertEqual(f,
|
||
("f", 1, "dummy", '"""Test cases for traceback module"""'))
|
||
self.assertEqual(tuple(f),
|
||
("f", 1, "dummy", '"""Test cases for traceback module"""'))
|
||
self.assertEqual(f, traceback.FrameSummary("f", 1, "dummy"))
|
||
self.assertEqual(f, tuple(f))
|
||
# Since tuple.__eq__ doesn't support FrameSummary, the equality
|
||
# operator fallbacks to FrameSummary.__eq__.
|
||
self.assertEqual(tuple(f), f)
|
||
self.assertIsNone(f.locals)
|
||
self.assertNotEqual(f, object())
|
||
self.assertEqual(f, ALWAYS_EQ)
|
||
|
||
def test_lazy_lines(self):
|
||
linecache.clearcache()
|
||
f = traceback.FrameSummary("f", 1, "dummy", lookup_line=False)
|
||
self.assertEqual(None, f._line)
|
||
linecache.lazycache("f", globals())
|
||
self.assertEqual(
|
||
'"""Test cases for traceback module"""',
|
||
f.line)
|
||
|
||
def test_no_line(self):
|
||
f = traceback.FrameSummary("f", None, "dummy")
|
||
self.assertEqual(f.line, None)
|
||
|
||
def test_explicit_line(self):
|
||
f = traceback.FrameSummary("f", 1, "dummy", line="line")
|
||
self.assertEqual("line", f.line)
|
||
|
||
def test_len(self):
|
||
f = traceback.FrameSummary("f", 1, "dummy", line="line")
|
||
self.assertEqual(len(f), 4)
|
||
|
||
|
||
class TestStack(unittest.TestCase):
|
||
|
||
def test_walk_stack(self):
|
||
def deeper():
|
||
return list(traceback.walk_stack(None))
|
||
s1 = list(traceback.walk_stack(None))
|
||
s2 = deeper()
|
||
self.assertEqual(len(s2) - len(s1), 1)
|
||
self.assertEqual(s2[1:], s1)
|
||
|
||
def test_walk_tb(self):
|
||
try:
|
||
1/0
|
||
except Exception as e:
|
||
tb = e.__traceback__
|
||
s = list(traceback.walk_tb(tb))
|
||
self.assertEqual(len(s), 1)
|
||
|
||
def test_extract_stack(self):
|
||
s = traceback.StackSummary.extract(traceback.walk_stack(None))
|
||
self.assertIsInstance(s, traceback.StackSummary)
|
||
|
||
def test_extract_stack_limit(self):
|
||
s = traceback.StackSummary.extract(traceback.walk_stack(None), limit=5)
|
||
self.assertEqual(len(s), 5)
|
||
|
||
def test_extract_stack_lookup_lines(self):
|
||
linecache.clearcache()
|
||
linecache.updatecache('/foo.py', globals())
|
||
c = test_code('/foo.py', 'method')
|
||
f = test_frame(c, None, None)
|
||
s = traceback.StackSummary.extract(iter([(f, 6)]), lookup_lines=True)
|
||
linecache.clearcache()
|
||
self.assertEqual(s[0].line, "import sys")
|
||
|
||
def test_extract_stackup_deferred_lookup_lines(self):
|
||
linecache.clearcache()
|
||
c = test_code('/foo.py', 'method')
|
||
f = test_frame(c, None, None)
|
||
s = traceback.StackSummary.extract(iter([(f, 6)]), lookup_lines=False)
|
||
self.assertEqual({}, linecache.cache)
|
||
linecache.updatecache('/foo.py', globals())
|
||
self.assertEqual(s[0].line, "import sys")
|
||
|
||
def test_from_list(self):
|
||
s = traceback.StackSummary.from_list([('foo.py', 1, 'fred', 'line')])
|
||
self.assertEqual(
|
||
[' File "foo.py", line 1, in fred\n line\n'],
|
||
s.format())
|
||
|
||
def test_from_list_edited_stack(self):
|
||
s = traceback.StackSummary.from_list([('foo.py', 1, 'fred', 'line')])
|
||
s[0] = ('foo.py', 2, 'fred', 'line')
|
||
s2 = traceback.StackSummary.from_list(s)
|
||
self.assertEqual(
|
||
[' File "foo.py", line 2, in fred\n line\n'],
|
||
s2.format())
|
||
|
||
def test_format_smoke(self):
|
||
# For detailed tests see the format_list tests, which consume the same
|
||
# code.
|
||
s = traceback.StackSummary.from_list([('foo.py', 1, 'fred', 'line')])
|
||
self.assertEqual(
|
||
[' File "foo.py", line 1, in fred\n line\n'],
|
||
s.format())
|
||
|
||
def test_locals(self):
|
||
linecache.updatecache('/foo.py', globals())
|
||
c = test_code('/foo.py', 'method')
|
||
f = test_frame(c, globals(), {'something': 1})
|
||
s = traceback.StackSummary.extract(iter([(f, 6)]), capture_locals=True)
|
||
self.assertEqual(s[0].locals, {'something': '1'})
|
||
|
||
def test_no_locals(self):
|
||
linecache.updatecache('/foo.py', globals())
|
||
c = test_code('/foo.py', 'method')
|
||
f = test_frame(c, globals(), {'something': 1})
|
||
s = traceback.StackSummary.extract(iter([(f, 6)]))
|
||
self.assertEqual(s[0].locals, None)
|
||
|
||
def test_format_locals(self):
|
||
def some_inner(k, v):
|
||
a = 1
|
||
b = 2
|
||
return traceback.StackSummary.extract(
|
||
traceback.walk_stack(None), capture_locals=True, limit=1)
|
||
s = some_inner(3, 4)
|
||
self.assertEqual(
|
||
[' File "%s", line %d, in some_inner\n'
|
||
' return traceback.StackSummary.extract(\n'
|
||
' a = 1\n'
|
||
' b = 2\n'
|
||
' k = 3\n'
|
||
' v = 4\n' % (__file__, some_inner.__code__.co_firstlineno + 3)
|
||
], s.format())
|
||
|
||
def test_custom_format_frame(self):
|
||
class CustomStackSummary(traceback.StackSummary):
|
||
def format_frame_summary(self, frame_summary):
|
||
return f'{frame_summary.filename}:{frame_summary.lineno}'
|
||
|
||
def some_inner():
|
||
return CustomStackSummary.extract(
|
||
traceback.walk_stack(None), limit=1)
|
||
|
||
s = some_inner()
|
||
self.assertEqual(
|
||
s.format(),
|
||
[f'{__file__}:{some_inner.__code__.co_firstlineno + 1}'])
|
||
|
||
def test_dropping_frames(self):
|
||
def f():
|
||
1/0
|
||
|
||
def g():
|
||
try:
|
||
f()
|
||
except Exception as e:
|
||
return e.__traceback__
|
||
|
||
tb = g()
|
||
|
||
class Skip_G(traceback.StackSummary):
|
||
def format_frame_summary(self, frame_summary):
|
||
if frame_summary.name == 'g':
|
||
return None
|
||
return super().format_frame_summary(frame_summary)
|
||
|
||
stack = Skip_G.extract(
|
||
traceback.walk_tb(tb)).format()
|
||
|
||
self.assertEqual(len(stack), 1)
|
||
lno = f.__code__.co_firstlineno + 1
|
||
self.assertEqual(
|
||
stack[0],
|
||
f' File "{__file__}", line {lno}, in f\n 1/0\n'
|
||
)
|
||
|
||
class Unrepresentable:
|
||
def __repr__(self) -> str:
|
||
raise Exception("Unrepresentable")
|
||
|
||
class TestTracebackException(unittest.TestCase):
|
||
|
||
def test_smoke(self):
|
||
try:
|
||
1/0
|
||
except Exception as e:
|
||
exc_obj = e
|
||
exc = traceback.TracebackException.from_exception(e)
|
||
expected_stack = traceback.StackSummary.extract(
|
||
traceback.walk_tb(e.__traceback__))
|
||
self.assertEqual(None, exc.__cause__)
|
||
self.assertEqual(None, exc.__context__)
|
||
self.assertEqual(False, exc.__suppress_context__)
|
||
self.assertEqual(expected_stack, exc.stack)
|
||
self.assertEqual(type(exc_obj), exc.exc_type)
|
||
self.assertEqual(str(exc_obj), str(exc))
|
||
|
||
def test_from_exception(self):
|
||
# Check all the parameters are accepted.
|
||
def foo():
|
||
1/0
|
||
try:
|
||
foo()
|
||
except Exception as e:
|
||
exc_obj = e
|
||
tb = e.__traceback__
|
||
self.expected_stack = traceback.StackSummary.extract(
|
||
traceback.walk_tb(tb), limit=1, lookup_lines=False,
|
||
capture_locals=True)
|
||
self.exc = traceback.TracebackException.from_exception(
|
||
e, limit=1, lookup_lines=False, capture_locals=True)
|
||
expected_stack = self.expected_stack
|
||
exc = self.exc
|
||
self.assertEqual(None, exc.__cause__)
|
||
self.assertEqual(None, exc.__context__)
|
||
self.assertEqual(False, exc.__suppress_context__)
|
||
self.assertEqual(expected_stack, exc.stack)
|
||
self.assertEqual(type(exc_obj), exc.exc_type)
|
||
self.assertEqual(str(exc_obj), str(exc))
|
||
|
||
def test_cause(self):
|
||
try:
|
||
try:
|
||
1/0
|
||
finally:
|
||
exc = sys.exception()
|
||
exc_context = traceback.TracebackException.from_exception(exc)
|
||
cause = Exception("cause")
|
||
raise Exception("uh oh") from cause
|
||
except Exception as e:
|
||
exc_obj = e
|
||
exc = traceback.TracebackException.from_exception(e)
|
||
expected_stack = traceback.StackSummary.extract(
|
||
traceback.walk_tb(e.__traceback__))
|
||
exc_cause = traceback.TracebackException(Exception, cause, None)
|
||
self.assertEqual(exc_cause, exc.__cause__)
|
||
self.assertEqual(exc_context, exc.__context__)
|
||
self.assertEqual(True, exc.__suppress_context__)
|
||
self.assertEqual(expected_stack, exc.stack)
|
||
self.assertEqual(type(exc_obj), exc.exc_type)
|
||
self.assertEqual(str(exc_obj), str(exc))
|
||
|
||
def test_context(self):
|
||
try:
|
||
try:
|
||
1/0
|
||
finally:
|
||
exc = sys.exception()
|
||
exc_context = traceback.TracebackException.from_exception(exc)
|
||
raise Exception("uh oh")
|
||
except Exception as e:
|
||
exc_obj = e
|
||
exc = traceback.TracebackException.from_exception(e)
|
||
expected_stack = traceback.StackSummary.extract(
|
||
traceback.walk_tb(e.__traceback__))
|
||
self.assertEqual(None, exc.__cause__)
|
||
self.assertEqual(exc_context, exc.__context__)
|
||
self.assertEqual(False, exc.__suppress_context__)
|
||
self.assertEqual(expected_stack, exc.stack)
|
||
self.assertEqual(type(exc_obj), exc.exc_type)
|
||
self.assertEqual(str(exc_obj), str(exc))
|
||
|
||
def test_long_context_chain(self):
|
||
def f():
|
||
try:
|
||
1/0
|
||
except:
|
||
f()
|
||
|
||
try:
|
||
f()
|
||
except RecursionError as e:
|
||
exc_obj = e
|
||
else:
|
||
self.fail("Exception not raised")
|
||
|
||
te = traceback.TracebackException.from_exception(exc_obj)
|
||
res = list(te.format())
|
||
|
||
# many ZeroDiv errors followed by the RecursionError
|
||
self.assertGreater(len(res), sys.getrecursionlimit())
|
||
self.assertGreater(
|
||
len([l for l in res if 'ZeroDivisionError:' in l]),
|
||
sys.getrecursionlimit() * 0.5)
|
||
self.assertIn(
|
||
"RecursionError: maximum recursion depth exceeded", res[-1])
|
||
|
||
def test_compact_with_cause(self):
|
||
try:
|
||
try:
|
||
1/0
|
||
finally:
|
||
cause = Exception("cause")
|
||
raise Exception("uh oh") from cause
|
||
except Exception as e:
|
||
exc_obj = e
|
||
exc = traceback.TracebackException.from_exception(exc_obj, compact=True)
|
||
expected_stack = traceback.StackSummary.extract(
|
||
traceback.walk_tb(exc_obj.__traceback__))
|
||
exc_cause = traceback.TracebackException(Exception, cause, None)
|
||
self.assertEqual(exc_cause, exc.__cause__)
|
||
self.assertEqual(None, exc.__context__)
|
||
self.assertEqual(True, exc.__suppress_context__)
|
||
self.assertEqual(expected_stack, exc.stack)
|
||
self.assertEqual(type(exc_obj), exc.exc_type)
|
||
self.assertEqual(str(exc_obj), str(exc))
|
||
|
||
def test_compact_no_cause(self):
|
||
try:
|
||
try:
|
||
1/0
|
||
finally:
|
||
exc = sys.exception()
|
||
exc_context = traceback.TracebackException.from_exception(exc)
|
||
raise Exception("uh oh")
|
||
except Exception as e:
|
||
exc_obj = e
|
||
exc = traceback.TracebackException.from_exception(e, compact=True)
|
||
expected_stack = traceback.StackSummary.extract(
|
||
traceback.walk_tb(exc_obj.__traceback__))
|
||
self.assertEqual(None, exc.__cause__)
|
||
self.assertEqual(exc_context, exc.__context__)
|
||
self.assertEqual(False, exc.__suppress_context__)
|
||
self.assertEqual(expected_stack, exc.stack)
|
||
self.assertEqual(type(exc_obj), exc.exc_type)
|
||
self.assertEqual(str(exc_obj), str(exc))
|
||
|
||
def test_no_refs_to_exception_and_traceback_objects(self):
|
||
try:
|
||
1/0
|
||
except Exception as e:
|
||
exc_obj = e
|
||
|
||
refcnt1 = sys.getrefcount(exc_obj)
|
||
refcnt2 = sys.getrefcount(exc_obj.__traceback__)
|
||
exc = traceback.TracebackException.from_exception(exc_obj)
|
||
self.assertEqual(sys.getrefcount(exc_obj), refcnt1)
|
||
self.assertEqual(sys.getrefcount(exc_obj.__traceback__), refcnt2)
|
||
|
||
def test_comparison_basic(self):
|
||
try:
|
||
1/0
|
||
except Exception as e:
|
||
exc_obj = e
|
||
exc = traceback.TracebackException.from_exception(exc_obj)
|
||
exc2 = traceback.TracebackException.from_exception(exc_obj)
|
||
self.assertIsNot(exc, exc2)
|
||
self.assertEqual(exc, exc2)
|
||
self.assertNotEqual(exc, object())
|
||
self.assertEqual(exc, ALWAYS_EQ)
|
||
|
||
def test_comparison_params_variations(self):
|
||
def raise_exc():
|
||
try:
|
||
raise ValueError('bad value')
|
||
except:
|
||
raise
|
||
|
||
def raise_with_locals():
|
||
x, y = 1, 2
|
||
raise_exc()
|
||
|
||
try:
|
||
raise_with_locals()
|
||
except Exception as e:
|
||
exc_obj = e
|
||
|
||
exc = traceback.TracebackException.from_exception(exc_obj)
|
||
exc1 = traceback.TracebackException.from_exception(exc_obj, limit=10)
|
||
exc2 = traceback.TracebackException.from_exception(exc_obj, limit=2)
|
||
|
||
self.assertEqual(exc, exc1) # limit=10 gets all frames
|
||
self.assertNotEqual(exc, exc2) # limit=2 truncates the output
|
||
|
||
# locals change the output
|
||
exc3 = traceback.TracebackException.from_exception(exc_obj, capture_locals=True)
|
||
self.assertNotEqual(exc, exc3)
|
||
|
||
# there are no locals in the innermost frame
|
||
exc4 = traceback.TracebackException.from_exception(exc_obj, limit=-1)
|
||
exc5 = traceback.TracebackException.from_exception(exc_obj, limit=-1, capture_locals=True)
|
||
self.assertEqual(exc4, exc5)
|
||
|
||
# there are locals in the next-to-innermost frame
|
||
exc6 = traceback.TracebackException.from_exception(exc_obj, limit=-2)
|
||
exc7 = traceback.TracebackException.from_exception(exc_obj, limit=-2, capture_locals=True)
|
||
self.assertNotEqual(exc6, exc7)
|
||
|
||
def test_comparison_equivalent_exceptions_are_equal(self):
|
||
excs = []
|
||
for _ in range(2):
|
||
try:
|
||
1/0
|
||
except Exception as e:
|
||
excs.append(traceback.TracebackException.from_exception(e))
|
||
self.assertEqual(excs[0], excs[1])
|
||
self.assertEqual(list(excs[0].format()), list(excs[1].format()))
|
||
|
||
def test_unhashable(self):
|
||
class UnhashableException(Exception):
|
||
def __eq__(self, other):
|
||
return True
|
||
|
||
ex1 = UnhashableException('ex1')
|
||
ex2 = UnhashableException('ex2')
|
||
try:
|
||
raise ex2 from ex1
|
||
except UnhashableException:
|
||
try:
|
||
raise ex1
|
||
except UnhashableException as e:
|
||
exc_obj = e
|
||
exc = traceback.TracebackException.from_exception(exc_obj)
|
||
formatted = list(exc.format())
|
||
self.assertIn('UnhashableException: ex2\n', formatted[2])
|
||
self.assertIn('UnhashableException: ex1\n', formatted[6])
|
||
|
||
def test_limit(self):
|
||
def recurse(n):
|
||
if n:
|
||
recurse(n-1)
|
||
else:
|
||
1/0
|
||
try:
|
||
recurse(10)
|
||
except Exception as e:
|
||
exc = traceback.TracebackException.from_exception(e, limit=5)
|
||
expected_stack = traceback.StackSummary.extract(
|
||
traceback.walk_tb(e.__traceback__), limit=5)
|
||
self.assertEqual(expected_stack, exc.stack)
|
||
|
||
def test_lookup_lines(self):
|
||
linecache.clearcache()
|
||
e = Exception("uh oh")
|
||
c = test_code('/foo.py', 'method')
|
||
f = test_frame(c, None, None)
|
||
tb = test_tb(f, 6, None, 0)
|
||
exc = traceback.TracebackException(Exception, e, tb, lookup_lines=False)
|
||
self.assertEqual(linecache.cache, {})
|
||
linecache.updatecache('/foo.py', globals())
|
||
self.assertEqual(exc.stack[0].line, "import sys")
|
||
|
||
def test_locals(self):
|
||
linecache.updatecache('/foo.py', globals())
|
||
e = Exception("uh oh")
|
||
c = test_code('/foo.py', 'method')
|
||
f = test_frame(c, globals(), {'something': 1, 'other': 'string', 'unrepresentable': Unrepresentable()})
|
||
tb = test_tb(f, 6, None, 0)
|
||
exc = traceback.TracebackException(
|
||
Exception, e, tb, capture_locals=True)
|
||
self.assertEqual(
|
||
exc.stack[0].locals,
|
||
{'something': '1', 'other': "'string'", 'unrepresentable': '<local repr() failed>'})
|
||
|
||
def test_no_locals(self):
|
||
linecache.updatecache('/foo.py', globals())
|
||
e = Exception("uh oh")
|
||
c = test_code('/foo.py', 'method')
|
||
f = test_frame(c, globals(), {'something': 1})
|
||
tb = test_tb(f, 6, None, 0)
|
||
exc = traceback.TracebackException(Exception, e, tb)
|
||
self.assertEqual(exc.stack[0].locals, None)
|
||
|
||
def test_traceback_header(self):
|
||
# do not print a traceback header if exc_traceback is None
|
||
# see issue #24695
|
||
exc = traceback.TracebackException(Exception, Exception("haven"), None)
|
||
self.assertEqual(list(exc.format()), ["Exception: haven\n"])
|
||
|
||
@requires_debug_ranges()
|
||
def test_print(self):
|
||
def f():
|
||
x = 12
|
||
try:
|
||
x/0
|
||
except Exception as e:
|
||
return e
|
||
exc = traceback.TracebackException.from_exception(f(), capture_locals=True)
|
||
output = StringIO()
|
||
exc.print(file=output)
|
||
self.assertEqual(
|
||
output.getvalue().split('\n')[-5:],
|
||
[' x/0',
|
||
' ~^~',
|
||
' x = 12',
|
||
'ZeroDivisionError: division by zero',
|
||
''])
|
||
|
||
|
||
class TestTracebackException_ExceptionGroups(unittest.TestCase):
|
||
def setUp(self):
|
||
super().setUp()
|
||
self.eg = self._get_exception_group()
|
||
|
||
def _get_exception_group(self):
|
||
def f():
|
||
1/0
|
||
|
||
def g(v):
|
||
raise ValueError(v)
|
||
|
||
self.lno_f = f.__code__.co_firstlineno
|
||
self.lno_g = g.__code__.co_firstlineno
|
||
|
||
try:
|
||
try:
|
||
try:
|
||
f()
|
||
except Exception as e:
|
||
exc1 = e
|
||
try:
|
||
g(42)
|
||
except Exception as e:
|
||
exc2 = e
|
||
raise ExceptionGroup("eg1", [exc1, exc2])
|
||
except ExceptionGroup as e:
|
||
exc3 = e
|
||
try:
|
||
g(24)
|
||
except Exception as e:
|
||
exc4 = e
|
||
raise ExceptionGroup("eg2", [exc3, exc4])
|
||
except ExceptionGroup as eg:
|
||
return eg
|
||
self.fail('Exception Not Raised')
|
||
|
||
def test_exception_group_construction(self):
|
||
eg = self.eg
|
||
teg1 = traceback.TracebackException(type(eg), eg, eg.__traceback__)
|
||
teg2 = traceback.TracebackException.from_exception(eg)
|
||
self.assertIsNot(teg1, teg2)
|
||
self.assertEqual(teg1, teg2)
|
||
|
||
def test_exception_group_format_exception_only(self):
|
||
teg = traceback.TracebackException.from_exception(self.eg)
|
||
formatted = ''.join(teg.format_exception_only()).split('\n')
|
||
expected = "ExceptionGroup: eg2 (2 sub-exceptions)\n".split('\n')
|
||
|
||
self.assertEqual(formatted, expected)
|
||
|
||
def test_exception_group_format(self):
|
||
teg = traceback.TracebackException.from_exception(self.eg)
|
||
|
||
formatted = ''.join(teg.format()).split('\n')
|
||
lno_f = self.lno_f
|
||
lno_g = self.lno_g
|
||
|
||
expected = [
|
||
f' + Exception Group Traceback (most recent call last):',
|
||
f' | File "{__file__}", line {lno_g+23}, in _get_exception_group',
|
||
f' | raise ExceptionGroup("eg2", [exc3, exc4])',
|
||
f' | ExceptionGroup: eg2 (2 sub-exceptions)',
|
||
f' +-+---------------- 1 ----------------',
|
||
f' | Exception Group Traceback (most recent call last):',
|
||
f' | File "{__file__}", line {lno_g+16}, in _get_exception_group',
|
||
f' | raise ExceptionGroup("eg1", [exc1, exc2])',
|
||
f' | ExceptionGroup: eg1 (2 sub-exceptions)',
|
||
f' +-+---------------- 1 ----------------',
|
||
f' | Traceback (most recent call last):',
|
||
f' | File "{__file__}", line {lno_g+9}, in _get_exception_group',
|
||
f' | f()',
|
||
f' | File "{__file__}", line {lno_f+1}, in f',
|
||
f' | 1/0',
|
||
f' | ~^~',
|
||
f' | ZeroDivisionError: division by zero',
|
||
f' +---------------- 2 ----------------',
|
||
f' | Traceback (most recent call last):',
|
||
f' | File "{__file__}", line {lno_g+13}, in _get_exception_group',
|
||
f' | g(42)',
|
||
f' | File "{__file__}", line {lno_g+1}, in g',
|
||
f' | raise ValueError(v)',
|
||
f' | ValueError: 42',
|
||
f' +------------------------------------',
|
||
f' +---------------- 2 ----------------',
|
||
f' | Traceback (most recent call last):',
|
||
f' | File "{__file__}", line {lno_g+20}, in _get_exception_group',
|
||
f' | g(24)',
|
||
f' | File "{__file__}", line {lno_g+1}, in g',
|
||
f' | raise ValueError(v)',
|
||
f' | ValueError: 24',
|
||
f' +------------------------------------',
|
||
f'']
|
||
|
||
self.assertEqual(formatted, expected)
|
||
|
||
def test_max_group_width(self):
|
||
excs1 = []
|
||
excs2 = []
|
||
for i in range(3):
|
||
excs1.append(ValueError(i))
|
||
for i in range(10):
|
||
excs2.append(TypeError(i))
|
||
|
||
EG = ExceptionGroup
|
||
eg = EG('eg', [EG('eg1', excs1), EG('eg2', excs2)])
|
||
|
||
teg = traceback.TracebackException.from_exception(eg, max_group_width=2)
|
||
formatted = ''.join(teg.format()).split('\n')
|
||
|
||
expected = [
|
||
' | ExceptionGroup: eg (2 sub-exceptions)',
|
||
' +-+---------------- 1 ----------------',
|
||
' | ExceptionGroup: eg1 (3 sub-exceptions)',
|
||
' +-+---------------- 1 ----------------',
|
||
' | ValueError: 0',
|
||
' +---------------- 2 ----------------',
|
||
' | ValueError: 1',
|
||
' +---------------- ... ----------------',
|
||
' | and 1 more exception',
|
||
' +------------------------------------',
|
||
' +---------------- 2 ----------------',
|
||
' | ExceptionGroup: eg2 (10 sub-exceptions)',
|
||
' +-+---------------- 1 ----------------',
|
||
' | TypeError: 0',
|
||
' +---------------- 2 ----------------',
|
||
' | TypeError: 1',
|
||
' +---------------- ... ----------------',
|
||
' | and 8 more exceptions',
|
||
' +------------------------------------',
|
||
'']
|
||
|
||
self.assertEqual(formatted, expected)
|
||
|
||
def test_max_group_depth(self):
|
||
exc = TypeError('bad type')
|
||
for i in range(3):
|
||
exc = ExceptionGroup('exc', [ValueError(-i), exc, ValueError(i)])
|
||
|
||
teg = traceback.TracebackException.from_exception(exc, max_group_depth=2)
|
||
formatted = ''.join(teg.format()).split('\n')
|
||
|
||
expected = [
|
||
' | ExceptionGroup: exc (3 sub-exceptions)',
|
||
' +-+---------------- 1 ----------------',
|
||
' | ValueError: -2',
|
||
' +---------------- 2 ----------------',
|
||
' | ExceptionGroup: exc (3 sub-exceptions)',
|
||
' +-+---------------- 1 ----------------',
|
||
' | ValueError: -1',
|
||
' +---------------- 2 ----------------',
|
||
' | ... (max_group_depth is 2)',
|
||
' +---------------- 3 ----------------',
|
||
' | ValueError: 1',
|
||
' +------------------------------------',
|
||
' +---------------- 3 ----------------',
|
||
' | ValueError: 2',
|
||
' +------------------------------------',
|
||
'']
|
||
|
||
self.assertEqual(formatted, expected)
|
||
|
||
def test_comparison(self):
|
||
try:
|
||
raise self.eg
|
||
except ExceptionGroup as e:
|
||
exc = e
|
||
for _ in range(5):
|
||
try:
|
||
raise exc
|
||
except Exception as e:
|
||
exc_obj = e
|
||
exc = traceback.TracebackException.from_exception(exc_obj)
|
||
exc2 = traceback.TracebackException.from_exception(exc_obj)
|
||
exc3 = traceback.TracebackException.from_exception(exc_obj, limit=300)
|
||
ne = traceback.TracebackException.from_exception(exc_obj, limit=3)
|
||
self.assertIsNot(exc, exc2)
|
||
self.assertEqual(exc, exc2)
|
||
self.assertEqual(exc, exc3)
|
||
self.assertNotEqual(exc, ne)
|
||
self.assertNotEqual(exc, object())
|
||
self.assertEqual(exc, ALWAYS_EQ)
|
||
|
||
|
||
global_for_suggestions = None
|
||
|
||
|
||
class SuggestionFormattingTestBase:
|
||
def get_suggestion(self, obj, attr_name=None):
|
||
if attr_name is not None:
|
||
def callable():
|
||
getattr(obj, attr_name)
|
||
else:
|
||
callable = obj
|
||
|
||
result_lines = self.get_exception(
|
||
callable, slice_start=-1, slice_end=None
|
||
)
|
||
return result_lines[0]
|
||
|
||
def test_getattr_suggestions(self):
|
||
class Substitution:
|
||
noise = more_noise = a = bc = None
|
||
blech = None
|
||
|
||
class Elimination:
|
||
noise = more_noise = a = bc = None
|
||
blch = None
|
||
|
||
class Addition:
|
||
noise = more_noise = a = bc = None
|
||
bluchin = None
|
||
|
||
class SubstitutionOverElimination:
|
||
blach = None
|
||
bluc = None
|
||
|
||
class SubstitutionOverAddition:
|
||
blach = None
|
||
bluchi = None
|
||
|
||
class EliminationOverAddition:
|
||
blucha = None
|
||
bluc = None
|
||
|
||
class CaseChangeOverSubstitution:
|
||
Luch = None
|
||
fluch = None
|
||
BLuch = None
|
||
|
||
for cls, suggestion in [
|
||
(Addition, "'bluchin'?"),
|
||
(Substitution, "'blech'?"),
|
||
(Elimination, "'blch'?"),
|
||
(Addition, "'bluchin'?"),
|
||
(SubstitutionOverElimination, "'blach'?"),
|
||
(SubstitutionOverAddition, "'blach'?"),
|
||
(EliminationOverAddition, "'bluc'?"),
|
||
(CaseChangeOverSubstitution, "'BLuch'?"),
|
||
]:
|
||
actual = self.get_suggestion(cls(), 'bluch')
|
||
self.assertIn(suggestion, actual)
|
||
|
||
def test_getattr_suggestions_do_not_trigger_for_long_attributes(self):
|
||
class A:
|
||
blech = None
|
||
|
||
actual = self.get_suggestion(A(), 'somethingverywrong')
|
||
self.assertNotIn("blech", actual)
|
||
|
||
def test_getattr_error_bad_suggestions_do_not_trigger_for_small_names(self):
|
||
class MyClass:
|
||
vvv = mom = w = id = pytho = None
|
||
|
||
for name in ("b", "v", "m", "py"):
|
||
with self.subTest(name=name):
|
||
actual = self.get_suggestion(MyClass, name)
|
||
self.assertNotIn("Did you mean", actual)
|
||
self.assertNotIn("'vvv", actual)
|
||
self.assertNotIn("'mom'", actual)
|
||
self.assertNotIn("'id'", actual)
|
||
self.assertNotIn("'w'", actual)
|
||
self.assertNotIn("'pytho'", actual)
|
||
|
||
def test_getattr_suggestions_do_not_trigger_for_big_dicts(self):
|
||
class A:
|
||
blech = None
|
||
# A class with a very big __dict__ will not be considered
|
||
# for suggestions.
|
||
for index in range(2000):
|
||
setattr(A, f"index_{index}", None)
|
||
|
||
actual = self.get_suggestion(A(), 'bluch')
|
||
self.assertNotIn("blech", actual)
|
||
|
||
def test_getattr_suggestions_no_args(self):
|
||
class A:
|
||
blech = None
|
||
def __getattr__(self, attr):
|
||
raise AttributeError()
|
||
|
||
actual = self.get_suggestion(A(), 'bluch')
|
||
self.assertIn("blech", actual)
|
||
|
||
class A:
|
||
blech = None
|
||
def __getattr__(self, attr):
|
||
raise AttributeError
|
||
|
||
actual = self.get_suggestion(A(), 'bluch')
|
||
self.assertIn("blech", actual)
|
||
|
||
def test_getattr_suggestions_invalid_args(self):
|
||
class NonStringifyClass:
|
||
__str__ = None
|
||
__repr__ = None
|
||
|
||
class A:
|
||
blech = None
|
||
def __getattr__(self, attr):
|
||
raise AttributeError(NonStringifyClass())
|
||
|
||
class B:
|
||
blech = None
|
||
def __getattr__(self, attr):
|
||
raise AttributeError("Error", 23)
|
||
|
||
class C:
|
||
blech = None
|
||
def __getattr__(self, attr):
|
||
raise AttributeError(23)
|
||
|
||
for cls in [A, B, C]:
|
||
actual = self.get_suggestion(cls(), 'bluch')
|
||
self.assertIn("blech", actual)
|
||
|
||
def test_getattr_suggestions_for_same_name(self):
|
||
class A:
|
||
def __dir__(self):
|
||
return ['blech']
|
||
actual = self.get_suggestion(A(), 'blech')
|
||
self.assertNotIn("Did you mean", actual)
|
||
|
||
def test_attribute_error_with_failing_dict(self):
|
||
class T:
|
||
bluch = 1
|
||
def __dir__(self):
|
||
raise AttributeError("oh no!")
|
||
|
||
actual = self.get_suggestion(T(), 'blich')
|
||
self.assertNotIn("blech", actual)
|
||
self.assertNotIn("oh no!", actual)
|
||
|
||
def test_attribute_error_with_bad_name(self):
|
||
def raise_attribute_error_with_bad_name():
|
||
raise AttributeError(name=12, obj=23)
|
||
|
||
result_lines = self.get_exception(
|
||
raise_attribute_error_with_bad_name, slice_start=-1, slice_end=None
|
||
)
|
||
self.assertNotIn("?", result_lines[-1])
|
||
|
||
def test_attribute_error_inside_nested_getattr(self):
|
||
class A:
|
||
bluch = 1
|
||
|
||
class B:
|
||
def __getattribute__(self, attr):
|
||
a = A()
|
||
return a.blich
|
||
|
||
actual = self.get_suggestion(B(), 'something')
|
||
self.assertIn("Did you mean", actual)
|
||
self.assertIn("bluch", actual)
|
||
|
||
def make_module(self, code):
|
||
tmpdir = Path(tempfile.mkdtemp())
|
||
self.addCleanup(shutil.rmtree, tmpdir)
|
||
|
||
sys.path.append(str(tmpdir))
|
||
self.addCleanup(sys.path.pop)
|
||
|
||
mod_name = ''.join(random.choices(string.ascii_letters, k=16))
|
||
module = tmpdir / (mod_name + ".py")
|
||
module.write_text(code)
|
||
|
||
return mod_name
|
||
|
||
def get_import_from_suggestion(self, mod_dict, name):
|
||
modname = self.make_module(mod_dict)
|
||
|
||
def callable():
|
||
try:
|
||
exec(f"from {modname} import {name}")
|
||
except ImportError as e:
|
||
raise e from None
|
||
except Exception as e:
|
||
self.fail(f"Expected ImportError but got {type(e)}")
|
||
self.addCleanup(forget, modname)
|
||
|
||
result_lines = self.get_exception(
|
||
callable, slice_start=-1, slice_end=None
|
||
)
|
||
return result_lines[0]
|
||
|
||
def test_import_from_suggestions(self):
|
||
substitution = textwrap.dedent("""\
|
||
noise = more_noise = a = bc = None
|
||
blech = None
|
||
""")
|
||
|
||
elimination = textwrap.dedent("""
|
||
noise = more_noise = a = bc = None
|
||
blch = None
|
||
""")
|
||
|
||
addition = textwrap.dedent("""
|
||
noise = more_noise = a = bc = None
|
||
bluchin = None
|
||
""")
|
||
|
||
substitutionOverElimination = textwrap.dedent("""
|
||
blach = None
|
||
bluc = None
|
||
""")
|
||
|
||
substitutionOverAddition = textwrap.dedent("""
|
||
blach = None
|
||
bluchi = None
|
||
""")
|
||
|
||
eliminationOverAddition = textwrap.dedent("""
|
||
blucha = None
|
||
bluc = None
|
||
""")
|
||
|
||
caseChangeOverSubstitution = textwrap.dedent("""
|
||
Luch = None
|
||
fluch = None
|
||
BLuch = None
|
||
""")
|
||
|
||
for code, suggestion in [
|
||
(addition, "'bluchin'?"),
|
||
(substitution, "'blech'?"),
|
||
(elimination, "'blch'?"),
|
||
(addition, "'bluchin'?"),
|
||
(substitutionOverElimination, "'blach'?"),
|
||
(substitutionOverAddition, "'blach'?"),
|
||
(eliminationOverAddition, "'bluc'?"),
|
||
(caseChangeOverSubstitution, "'BLuch'?"),
|
||
]:
|
||
actual = self.get_import_from_suggestion(code, 'bluch')
|
||
self.assertIn(suggestion, actual)
|
||
|
||
def test_import_from_suggestions_do_not_trigger_for_long_attributes(self):
|
||
code = "blech = None"
|
||
|
||
actual = self.get_suggestion(code, 'somethingverywrong')
|
||
self.assertNotIn("blech", actual)
|
||
|
||
def test_import_from_error_bad_suggestions_do_not_trigger_for_small_names(self):
|
||
code = "vvv = mom = w = id = pytho = None"
|
||
|
||
for name in ("b", "v", "m", "py"):
|
||
with self.subTest(name=name):
|
||
actual = self.get_import_from_suggestion(code, name)
|
||
self.assertNotIn("Did you mean", actual)
|
||
self.assertNotIn("'vvv'", actual)
|
||
self.assertNotIn("'mom'", actual)
|
||
self.assertNotIn("'id'", actual)
|
||
self.assertNotIn("'w'", actual)
|
||
self.assertNotIn("'pytho'", actual)
|
||
|
||
def test_import_from_suggestions_do_not_trigger_for_big_namespaces(self):
|
||
# A module with lots of names will not be considered for suggestions.
|
||
chunks = [f"index_{index} = " for index in range(200)]
|
||
chunks.append(" None")
|
||
code = " ".join(chunks)
|
||
actual = self.get_import_from_suggestion(code, 'bluch')
|
||
self.assertNotIn("blech", actual)
|
||
|
||
def test_import_from_error_with_bad_name(self):
|
||
def raise_attribute_error_with_bad_name():
|
||
raise ImportError(name=12, obj=23, name_from=11)
|
||
|
||
result_lines = self.get_exception(
|
||
raise_attribute_error_with_bad_name, slice_start=-1, slice_end=None
|
||
)
|
||
self.assertNotIn("?", result_lines[-1])
|
||
|
||
def test_name_error_suggestions(self):
|
||
def Substitution():
|
||
noise = more_noise = a = bc = None
|
||
blech = None
|
||
print(bluch)
|
||
|
||
def Elimination():
|
||
noise = more_noise = a = bc = None
|
||
blch = None
|
||
print(bluch)
|
||
|
||
def Addition():
|
||
noise = more_noise = a = bc = None
|
||
bluchin = None
|
||
print(bluch)
|
||
|
||
def SubstitutionOverElimination():
|
||
blach = None
|
||
bluc = None
|
||
print(bluch)
|
||
|
||
def SubstitutionOverAddition():
|
||
blach = None
|
||
bluchi = None
|
||
print(bluch)
|
||
|
||
def EliminationOverAddition():
|
||
blucha = None
|
||
bluc = None
|
||
print(bluch)
|
||
|
||
for func, suggestion in [(Substitution, "'blech'?"),
|
||
(Elimination, "'blch'?"),
|
||
(Addition, "'bluchin'?"),
|
||
(EliminationOverAddition, "'blucha'?"),
|
||
(SubstitutionOverElimination, "'blach'?"),
|
||
(SubstitutionOverAddition, "'blach'?")]:
|
||
actual = self.get_suggestion(func)
|
||
self.assertIn(suggestion, actual)
|
||
|
||
def test_name_error_suggestions_from_globals(self):
|
||
def func():
|
||
print(global_for_suggestio)
|
||
actual = self.get_suggestion(func)
|
||
self.assertIn("'global_for_suggestions'?", actual)
|
||
|
||
def test_name_error_suggestions_from_builtins(self):
|
||
def func():
|
||
print(ZeroDivisionErrrrr)
|
||
actual = self.get_suggestion(func)
|
||
self.assertIn("'ZeroDivisionError'?", actual)
|
||
|
||
def test_name_error_suggestions_from_builtins_when_builtins_is_module(self):
|
||
def func():
|
||
custom_globals = globals().copy()
|
||
custom_globals["__builtins__"] = builtins
|
||
print(eval("ZeroDivisionErrrrr", custom_globals))
|
||
actual = self.get_suggestion(func)
|
||
self.assertIn("'ZeroDivisionError'?", actual)
|
||
|
||
def test_name_error_suggestions_do_not_trigger_for_long_names(self):
|
||
def func():
|
||
somethingverywronghehehehehehe = None
|
||
print(somethingverywronghe)
|
||
actual = self.get_suggestion(func)
|
||
self.assertNotIn("somethingverywronghehe", actual)
|
||
|
||
def test_name_error_bad_suggestions_do_not_trigger_for_small_names(self):
|
||
|
||
def f_b():
|
||
vvv = mom = w = id = pytho = None
|
||
b
|
||
|
||
def f_v():
|
||
vvv = mom = w = id = pytho = None
|
||
v
|
||
|
||
def f_m():
|
||
vvv = mom = w = id = pytho = None
|
||
m
|
||
|
||
def f_py():
|
||
vvv = mom = w = id = pytho = None
|
||
py
|
||
|
||
for name, func in (("b", f_b), ("v", f_v), ("m", f_m), ("py", f_py)):
|
||
with self.subTest(name=name):
|
||
actual = self.get_suggestion(func)
|
||
self.assertNotIn("you mean", actual)
|
||
self.assertNotIn("vvv", actual)
|
||
self.assertNotIn("mom", actual)
|
||
self.assertNotIn("'id'", actual)
|
||
self.assertNotIn("'w'", actual)
|
||
self.assertNotIn("'pytho'", actual)
|
||
|
||
def test_name_error_suggestions_do_not_trigger_for_too_many_locals(self):
|
||
def func():
|
||
# Mutating locals() is unreliable, so we need to do it by hand
|
||
a1 = a2 = a3 = a4 = a5 = a6 = a7 = a8 = a9 = a10 = \
|
||
a11 = a12 = a13 = a14 = a15 = a16 = a17 = a18 = a19 = a20 = \
|
||
a21 = a22 = a23 = a24 = a25 = a26 = a27 = a28 = a29 = a30 = \
|
||
a31 = a32 = a33 = a34 = a35 = a36 = a37 = a38 = a39 = a40 = \
|
||
a41 = a42 = a43 = a44 = a45 = a46 = a47 = a48 = a49 = a50 = \
|
||
a51 = a52 = a53 = a54 = a55 = a56 = a57 = a58 = a59 = a60 = \
|
||
a61 = a62 = a63 = a64 = a65 = a66 = a67 = a68 = a69 = a70 = \
|
||
a71 = a72 = a73 = a74 = a75 = a76 = a77 = a78 = a79 = a80 = \
|
||
a81 = a82 = a83 = a84 = a85 = a86 = a87 = a88 = a89 = a90 = \
|
||
a91 = a92 = a93 = a94 = a95 = a96 = a97 = a98 = a99 = a100 = \
|
||
a101 = a102 = a103 = a104 = a105 = a106 = a107 = a108 = a109 = a110 = \
|
||
a111 = a112 = a113 = a114 = a115 = a116 = a117 = a118 = a119 = a120 = \
|
||
a121 = a122 = a123 = a124 = a125 = a126 = a127 = a128 = a129 = a130 = \
|
||
a131 = a132 = a133 = a134 = a135 = a136 = a137 = a138 = a139 = a140 = \
|
||
a141 = a142 = a143 = a144 = a145 = a146 = a147 = a148 = a149 = a150 = \
|
||
a151 = a152 = a153 = a154 = a155 = a156 = a157 = a158 = a159 = a160 = \
|
||
a161 = a162 = a163 = a164 = a165 = a166 = a167 = a168 = a169 = a170 = \
|
||
a171 = a172 = a173 = a174 = a175 = a176 = a177 = a178 = a179 = a180 = \
|
||
a181 = a182 = a183 = a184 = a185 = a186 = a187 = a188 = a189 = a190 = \
|
||
a191 = a192 = a193 = a194 = a195 = a196 = a197 = a198 = a199 = a200 = \
|
||
a201 = a202 = a203 = a204 = a205 = a206 = a207 = a208 = a209 = a210 = \
|
||
a211 = a212 = a213 = a214 = a215 = a216 = a217 = a218 = a219 = a220 = \
|
||
a221 = a222 = a223 = a224 = a225 = a226 = a227 = a228 = a229 = a230 = \
|
||
a231 = a232 = a233 = a234 = a235 = a236 = a237 = a238 = a239 = a240 = \
|
||
a241 = a242 = a243 = a244 = a245 = a246 = a247 = a248 = a249 = a250 = \
|
||
a251 = a252 = a253 = a254 = a255 = a256 = a257 = a258 = a259 = a260 = \
|
||
a261 = a262 = a263 = a264 = a265 = a266 = a267 = a268 = a269 = a270 = \
|
||
a271 = a272 = a273 = a274 = a275 = a276 = a277 = a278 = a279 = a280 = \
|
||
a281 = a282 = a283 = a284 = a285 = a286 = a287 = a288 = a289 = a290 = \
|
||
a291 = a292 = a293 = a294 = a295 = a296 = a297 = a298 = a299 = a300 = \
|
||
a301 = a302 = a303 = a304 = a305 = a306 = a307 = a308 = a309 = a310 = \
|
||
a311 = a312 = a313 = a314 = a315 = a316 = a317 = a318 = a319 = a320 = \
|
||
a321 = a322 = a323 = a324 = a325 = a326 = a327 = a328 = a329 = a330 = \
|
||
a331 = a332 = a333 = a334 = a335 = a336 = a337 = a338 = a339 = a340 = \
|
||
a341 = a342 = a343 = a344 = a345 = a346 = a347 = a348 = a349 = a350 = \
|
||
a351 = a352 = a353 = a354 = a355 = a356 = a357 = a358 = a359 = a360 = \
|
||
a361 = a362 = a363 = a364 = a365 = a366 = a367 = a368 = a369 = a370 = \
|
||
a371 = a372 = a373 = a374 = a375 = a376 = a377 = a378 = a379 = a380 = \
|
||
a381 = a382 = a383 = a384 = a385 = a386 = a387 = a388 = a389 = a390 = \
|
||
a391 = a392 = a393 = a394 = a395 = a396 = a397 = a398 = a399 = a400 = \
|
||
a401 = a402 = a403 = a404 = a405 = a406 = a407 = a408 = a409 = a410 = \
|
||
a411 = a412 = a413 = a414 = a415 = a416 = a417 = a418 = a419 = a420 = \
|
||
a421 = a422 = a423 = a424 = a425 = a426 = a427 = a428 = a429 = a430 = \
|
||
a431 = a432 = a433 = a434 = a435 = a436 = a437 = a438 = a439 = a440 = \
|
||
a441 = a442 = a443 = a444 = a445 = a446 = a447 = a448 = a449 = a450 = \
|
||
a451 = a452 = a453 = a454 = a455 = a456 = a457 = a458 = a459 = a460 = \
|
||
a461 = a462 = a463 = a464 = a465 = a466 = a467 = a468 = a469 = a470 = \
|
||
a471 = a472 = a473 = a474 = a475 = a476 = a477 = a478 = a479 = a480 = \
|
||
a481 = a482 = a483 = a484 = a485 = a486 = a487 = a488 = a489 = a490 = \
|
||
a491 = a492 = a493 = a494 = a495 = a496 = a497 = a498 = a499 = a500 = \
|
||
a501 = a502 = a503 = a504 = a505 = a506 = a507 = a508 = a509 = a510 = \
|
||
a511 = a512 = a513 = a514 = a515 = a516 = a517 = a518 = a519 = a520 = \
|
||
a521 = a522 = a523 = a524 = a525 = a526 = a527 = a528 = a529 = a530 = \
|
||
a531 = a532 = a533 = a534 = a535 = a536 = a537 = a538 = a539 = a540 = \
|
||
a541 = a542 = a543 = a544 = a545 = a546 = a547 = a548 = a549 = a550 = \
|
||
a551 = a552 = a553 = a554 = a555 = a556 = a557 = a558 = a559 = a560 = \
|
||
a561 = a562 = a563 = a564 = a565 = a566 = a567 = a568 = a569 = a570 = \
|
||
a571 = a572 = a573 = a574 = a575 = a576 = a577 = a578 = a579 = a580 = \
|
||
a581 = a582 = a583 = a584 = a585 = a586 = a587 = a588 = a589 = a590 = \
|
||
a591 = a592 = a593 = a594 = a595 = a596 = a597 = a598 = a599 = a600 = \
|
||
a601 = a602 = a603 = a604 = a605 = a606 = a607 = a608 = a609 = a610 = \
|
||
a611 = a612 = a613 = a614 = a615 = a616 = a617 = a618 = a619 = a620 = \
|
||
a621 = a622 = a623 = a624 = a625 = a626 = a627 = a628 = a629 = a630 = \
|
||
a631 = a632 = a633 = a634 = a635 = a636 = a637 = a638 = a639 = a640 = \
|
||
a641 = a642 = a643 = a644 = a645 = a646 = a647 = a648 = a649 = a650 = \
|
||
a651 = a652 = a653 = a654 = a655 = a656 = a657 = a658 = a659 = a660 = \
|
||
a661 = a662 = a663 = a664 = a665 = a666 = a667 = a668 = a669 = a670 = \
|
||
a671 = a672 = a673 = a674 = a675 = a676 = a677 = a678 = a679 = a680 = \
|
||
a681 = a682 = a683 = a684 = a685 = a686 = a687 = a688 = a689 = a690 = \
|
||
a691 = a692 = a693 = a694 = a695 = a696 = a697 = a698 = a699 = a700 = \
|
||
a701 = a702 = a703 = a704 = a705 = a706 = a707 = a708 = a709 = a710 = \
|
||
a711 = a712 = a713 = a714 = a715 = a716 = a717 = a718 = a719 = a720 = \
|
||
a721 = a722 = a723 = a724 = a725 = a726 = a727 = a728 = a729 = a730 = \
|
||
a731 = a732 = a733 = a734 = a735 = a736 = a737 = a738 = a739 = a740 = \
|
||
a741 = a742 = a743 = a744 = a745 = a746 = a747 = a748 = a749 = a750 = \
|
||
a751 = a752 = a753 = a754 = a755 = a756 = a757 = a758 = a759 = a760 = \
|
||
a761 = a762 = a763 = a764 = a765 = a766 = a767 = a768 = a769 = a770 = \
|
||
a771 = a772 = a773 = a774 = a775 = a776 = a777 = a778 = a779 = a780 = \
|
||
a781 = a782 = a783 = a784 = a785 = a786 = a787 = a788 = a789 = a790 = \
|
||
a791 = a792 = a793 = a794 = a795 = a796 = a797 = a798 = a799 = a800 \
|
||
= None
|
||
print(a0)
|
||
|
||
actual = self.get_suggestion(func)
|
||
self.assertNotRegex(actual, r"NameError.*a1")
|
||
|
||
def test_name_error_with_custom_exceptions(self):
|
||
def func():
|
||
blech = None
|
||
raise NameError()
|
||
|
||
actual = self.get_suggestion(func)
|
||
self.assertNotIn("blech", actual)
|
||
|
||
def func():
|
||
blech = None
|
||
raise NameError
|
||
|
||
actual = self.get_suggestion(func)
|
||
self.assertNotIn("blech", actual)
|
||
|
||
def test_name_error_with_instance(self):
|
||
class A:
|
||
def __init__(self):
|
||
self.blech = None
|
||
def foo(self):
|
||
blich = 1
|
||
x = blech
|
||
|
||
instance = A()
|
||
actual = self.get_suggestion(instance.foo)
|
||
self.assertIn("self.blech", actual)
|
||
|
||
def test_unbound_local_error_with_instance(self):
|
||
class A:
|
||
def __init__(self):
|
||
self.blech = None
|
||
def foo(self):
|
||
blich = 1
|
||
x = blech
|
||
blech = 1
|
||
|
||
instance = A()
|
||
actual = self.get_suggestion(instance.foo)
|
||
self.assertNotIn("self.blech", actual)
|
||
|
||
def test_unbound_local_error_does_not_match(self):
|
||
def func():
|
||
something = 3
|
||
print(somethong)
|
||
somethong = 3
|
||
|
||
actual = self.get_suggestion(func)
|
||
self.assertNotIn("something", actual)
|
||
|
||
def test_name_error_for_stdlib_modules(self):
|
||
def func():
|
||
stream = io.StringIO()
|
||
|
||
actual = self.get_suggestion(func)
|
||
self.assertIn("forget to import 'io'", actual)
|
||
|
||
def test_name_error_for_private_stdlib_modules(self):
|
||
def func():
|
||
stream = _io.StringIO()
|
||
|
||
actual = self.get_suggestion(func)
|
||
self.assertIn("forget to import '_io'", actual)
|
||
|
||
|
||
|
||
class PurePythonSuggestionFormattingTests(
|
||
PurePythonExceptionFormattingMixin,
|
||
SuggestionFormattingTestBase,
|
||
unittest.TestCase,
|
||
):
|
||
"""
|
||
Same set of tests as above using the pure Python implementation of
|
||
traceback printing in traceback.py.
|
||
"""
|
||
|
||
|
||
@cpython_only
|
||
class CPythonSuggestionFormattingTests(
|
||
CAPIExceptionFormattingMixin,
|
||
SuggestionFormattingTestBase,
|
||
unittest.TestCase,
|
||
):
|
||
"""
|
||
Same set of tests as above but with Python's internal traceback printing.
|
||
"""
|
||
|
||
|
||
class MiscTest(unittest.TestCase):
|
||
|
||
def test_all(self):
|
||
expected = set()
|
||
denylist = {'print_list'}
|
||
for name in dir(traceback):
|
||
if name.startswith('_') or name in denylist:
|
||
continue
|
||
module_object = getattr(traceback, name)
|
||
if getattr(module_object, '__module__', None) == 'traceback':
|
||
expected.add(name)
|
||
self.assertCountEqual(traceback.__all__, expected)
|
||
|
||
def test_levenshtein_distance(self):
|
||
# copied from _testinternalcapi.test_edit_cost
|
||
# to also exercise the Python implementation
|
||
|
||
def CHECK(a, b, expected):
|
||
actual = traceback._levenshtein_distance(a, b, 4044)
|
||
self.assertEqual(actual, expected)
|
||
|
||
CHECK("", "", 0)
|
||
CHECK("", "a", 2)
|
||
CHECK("a", "A", 1)
|
||
CHECK("Apple", "Aple", 2)
|
||
CHECK("Banana", "B@n@n@", 6)
|
||
CHECK("Cherry", "Cherry!", 2)
|
||
CHECK("---0---", "------", 2)
|
||
CHECK("abc", "y", 6)
|
||
CHECK("aa", "bb", 4)
|
||
CHECK("aaaaa", "AAAAA", 5)
|
||
CHECK("wxyz", "wXyZ", 2)
|
||
CHECK("wxyz", "wXyZ123", 8)
|
||
CHECK("Python", "Java", 12)
|
||
CHECK("Java", "C#", 8)
|
||
CHECK("AbstractFoobarManager", "abstract_foobar_manager", 3+2*2)
|
||
CHECK("CPython", "PyPy", 10)
|
||
CHECK("CPython", "pypy", 11)
|
||
CHECK("AttributeError", "AttributeErrop", 2)
|
||
CHECK("AttributeError", "AttributeErrorTests", 10)
|
||
CHECK("ABA", "AAB", 4)
|
||
|
||
def test_levenshtein_distance_short_circuit(self):
|
||
if not LEVENSHTEIN_DATA_FILE.is_file():
|
||
self.fail(
|
||
f"{LEVENSHTEIN_DATA_FILE} is missing."
|
||
f" Run `make regen-test-levenshtein`"
|
||
)
|
||
|
||
with LEVENSHTEIN_DATA_FILE.open("r") as f:
|
||
examples = json.load(f)
|
||
for a, b, expected in examples:
|
||
res1 = traceback._levenshtein_distance(a, b, 1000)
|
||
self.assertEqual(res1, expected, msg=(a, b))
|
||
|
||
for threshold in [expected, expected + 1, expected + 2]:
|
||
# big enough thresholds shouldn't change the result
|
||
res2 = traceback._levenshtein_distance(a, b, threshold)
|
||
self.assertEqual(res2, expected, msg=(a, b, threshold))
|
||
|
||
for threshold in range(expected):
|
||
# for small thresholds, the only piece of information
|
||
# we receive is "strings not close enough".
|
||
res3 = traceback._levenshtein_distance(a, b, threshold)
|
||
self.assertGreater(res3, threshold, msg=(a, b, threshold))
|
||
|
||
|
||
if __name__ == "__main__":
|
||
unittest.main()
|