blob: aefdd7b769f289f76c3e4b2b1be2c6a707d0ec64 [file] [log] [blame]
Guido van Rossum3bead091992-01-27 17:00:37 +00001# Python test set -- part 5, built-in exceptions
2
Serhiy Storchakab7853962017-04-08 09:55:07 +03003import copy
Pablo Galindo9b648a92020-09-01 19:39:46 +01004import gc
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00005import os
6import sys
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00007import unittest
Guido van Rossumbf12cdb2006-08-17 20:24:18 +00008import pickle
Barry Warsaw8d109cb2008-05-08 04:26:35 +00009import weakref
Antoine Pitroua7622852011-09-01 21:37:43 +020010import errno
Pablo Galindoa77aac42021-04-23 14:27:05 +010011from textwrap import dedent
Thomas Wouters73e5a5b2006-06-08 15:35:45 +000012
Hai Shi46605972020-08-04 00:49:18 +080013from test.support import (captured_stderr, check_impl_detail,
14 cpython_only, gc_collect,
15 no_tracing, script_helper,
xdegaye56d1f5c2017-10-26 15:09:06 +020016 SuppressCrashReport)
Hai Shi46605972020-08-04 00:49:18 +080017from test.support.import_helper import import_module
18from test.support.os_helper import TESTFN, unlink
19from test.support.warnings_helper import check_warnings
Victor Stinnere4d300e2019-05-22 23:44:02 +020020from test import support
21
22
Richard Oudkerk5562d9d2012-07-28 17:45:28 +010023class NaiveException(Exception):
24 def __init__(self, x):
25 self.x = x
26
27class SlottedNaiveException(Exception):
28 __slots__ = ('x',)
29 def __init__(self, x):
30 self.x = x
31
Martin Panter3263f682016-02-28 03:16:11 +000032class BrokenStrException(Exception):
33 def __str__(self):
34 raise Exception("str() is broken")
35
Guido van Rossum3bead091992-01-27 17:00:37 +000036# XXX This is not really enough, each *operation* should be tested!
37
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000038class ExceptionTests(unittest.TestCase):
Barry Warsawb9c1d3d2001-08-13 23:07:00 +000039
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000040 def raise_catch(self, exc, excname):
Pablo Galindoaf8e5f82020-05-17 01:22:00 +010041 with self.subTest(exc=exc, excname=excname):
42 try:
43 raise exc("spam")
44 except exc as err:
45 buf1 = str(err)
46 try:
47 raise exc("spam")
48 except exc as err:
49 buf2 = str(err)
50 self.assertEqual(buf1, buf2)
51 self.assertEqual(exc.__name__, excname)
Guido van Rossum3bead091992-01-27 17:00:37 +000052
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000053 def testRaising(self):
54 self.raise_catch(AttributeError, "AttributeError")
55 self.assertRaises(AttributeError, getattr, sys, "undefined_attribute")
Guido van Rossum3bead091992-01-27 17:00:37 +000056
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000057 self.raise_catch(EOFError, "EOFError")
Inada Naoki8bbfeb32021-04-02 12:53:46 +090058 fp = open(TESTFN, 'w', encoding="utf-8")
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000059 fp.close()
Inada Naoki8bbfeb32021-04-02 12:53:46 +090060 fp = open(TESTFN, 'r', encoding="utf-8")
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000061 savestdin = sys.stdin
62 try:
63 try:
64 import marshal
Antoine Pitrou4a90ef02012-03-03 02:35:32 +010065 marshal.loads(b'')
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000066 except EOFError:
67 pass
68 finally:
69 sys.stdin = savestdin
70 fp.close()
71 unlink(TESTFN)
Guido van Rossum3bead091992-01-27 17:00:37 +000072
Antoine Pitrou6b4883d2011-10-12 02:54:14 +020073 self.raise_catch(OSError, "OSError")
74 self.assertRaises(OSError, open, 'this file does not exist', 'r')
Guido van Rossum3bead091992-01-27 17:00:37 +000075
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000076 self.raise_catch(ImportError, "ImportError")
77 self.assertRaises(ImportError, __import__, "undefined_module")
Guido van Rossum3bead091992-01-27 17:00:37 +000078
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000079 self.raise_catch(IndexError, "IndexError")
80 x = []
81 self.assertRaises(IndexError, x.__getitem__, 10)
Guido van Rossum3bead091992-01-27 17:00:37 +000082
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000083 self.raise_catch(KeyError, "KeyError")
84 x = {}
85 self.assertRaises(KeyError, x.__getitem__, 'key')
Guido van Rossum3bead091992-01-27 17:00:37 +000086
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000087 self.raise_catch(KeyboardInterrupt, "KeyboardInterrupt")
Guido van Rossum3bead091992-01-27 17:00:37 +000088
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000089 self.raise_catch(MemoryError, "MemoryError")
Guido van Rossum3bead091992-01-27 17:00:37 +000090
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000091 self.raise_catch(NameError, "NameError")
92 try: x = undefined_variable
93 except NameError: pass
Guido van Rossum3bead091992-01-27 17:00:37 +000094
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000095 self.raise_catch(OverflowError, "OverflowError")
96 x = 1
97 for dummy in range(128):
98 x += x # this simply shouldn't blow up
Guido van Rossum3bead091992-01-27 17:00:37 +000099
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000100 self.raise_catch(RuntimeError, "RuntimeError")
Yury Selivanovf488fb42015-07-03 01:04:23 -0400101 self.raise_catch(RecursionError, "RecursionError")
Guido van Rossum3bead091992-01-27 17:00:37 +0000102
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000103 self.raise_catch(SyntaxError, "SyntaxError")
Georg Brandl7cae87c2006-09-06 06:51:57 +0000104 try: exec('/\n')
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000105 except SyntaxError: pass
Guido van Rossum3bead091992-01-27 17:00:37 +0000106
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000107 self.raise_catch(IndentationError, "IndentationError")
Fred Drake72e48bd2000-09-08 16:32:34 +0000108
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000109 self.raise_catch(TabError, "TabError")
Georg Brandle1b5ac62008-06-04 13:06:58 +0000110 try: compile("try:\n\t1/0\n \t1/0\nfinally:\n pass\n",
111 '<string>', 'exec')
112 except TabError: pass
113 else: self.fail("TabError not raised")
Fred Drake72e48bd2000-09-08 16:32:34 +0000114
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000115 self.raise_catch(SystemError, "SystemError")
Fred Drake72e48bd2000-09-08 16:32:34 +0000116
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000117 self.raise_catch(SystemExit, "SystemExit")
118 self.assertRaises(SystemExit, sys.exit, 0)
Fred Drake85f36392000-07-11 17:53:00 +0000119
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000120 self.raise_catch(TypeError, "TypeError")
121 try: [] + ()
122 except TypeError: pass
Fred Drake85f36392000-07-11 17:53:00 +0000123
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000124 self.raise_catch(ValueError, "ValueError")
Guido van Rossume63bae62007-07-17 00:34:25 +0000125 self.assertRaises(ValueError, chr, 17<<16)
Guido van Rossum3bead091992-01-27 17:00:37 +0000126
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000127 self.raise_catch(ZeroDivisionError, "ZeroDivisionError")
128 try: x = 1/0
129 except ZeroDivisionError: pass
Guido van Rossum3bead091992-01-27 17:00:37 +0000130
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000131 self.raise_catch(Exception, "Exception")
132 try: x = 1/0
Guido van Rossumb940e112007-01-10 16:19:56 +0000133 except Exception as e: pass
Guido van Rossum3bead091992-01-27 17:00:37 +0000134
Yury Selivanovccc897f2015-07-03 01:16:04 -0400135 self.raise_catch(StopAsyncIteration, "StopAsyncIteration")
136
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000137 def testSyntaxErrorMessage(self):
138 # make sure the right exception message is raised for each of
139 # these code fragments
Guido van Rossum3bead091992-01-27 17:00:37 +0000140
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000141 def ckmsg(src, msg):
Pablo Galindoaf8e5f82020-05-17 01:22:00 +0100142 with self.subTest(src=src, msg=msg):
143 try:
144 compile(src, '<fragment>', 'exec')
145 except SyntaxError as e:
146 if e.msg != msg:
147 self.fail("expected %s, got %s" % (msg, e.msg))
148 else:
149 self.fail("failed to get expected SyntaxError")
Guido van Rossum3bead091992-01-27 17:00:37 +0000150
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000151 s = '''if 1:
152 try:
153 continue
154 except:
155 pass'''
Jeremy Hyltonede049b2001-09-26 20:01:13 +0000156
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000157 ckmsg(s, "'continue' not properly in loop")
158 ckmsg("continue\n", "'continue' not properly in loop")
Thomas Wouters303de6a2006-04-20 22:42:37 +0000159
Martijn Pieters772d8092017-08-22 21:16:23 +0100160 def testSyntaxErrorMissingParens(self):
161 def ckmsg(src, msg, exception=SyntaxError):
162 try:
163 compile(src, '<fragment>', 'exec')
164 except exception as e:
165 if e.msg != msg:
166 self.fail("expected %s, got %s" % (msg, e.msg))
167 else:
168 self.fail("failed to get expected SyntaxError")
169
170 s = '''print "old style"'''
171 ckmsg(s, "Missing parentheses in call to 'print'. "
172 "Did you mean print(\"old style\")?")
173
174 s = '''print "old style",'''
175 ckmsg(s, "Missing parentheses in call to 'print'. "
176 "Did you mean print(\"old style\", end=\" \")?")
177
178 s = '''exec "old style"'''
179 ckmsg(s, "Missing parentheses in call to 'exec'")
180
181 # should not apply to subclasses, see issue #31161
182 s = '''if True:\nprint "No indent"'''
Pablo Galindo56c95df2021-04-21 15:28:21 +0100183 ckmsg(s, "expected an indented block after 'if' statement on line 1", IndentationError)
Martijn Pieters772d8092017-08-22 21:16:23 +0100184
185 s = '''if True:\n print()\n\texec "mixed tabs and spaces"'''
186 ckmsg(s, "inconsistent use of tabs and spaces in indentation", TabError)
187
Batuhan Taskaya76c1b4d2020-05-01 16:13:43 +0300188 def check(self, src, lineno, offset, encoding='utf-8'):
Pablo Galindoaf8e5f82020-05-17 01:22:00 +0100189 with self.subTest(source=src, lineno=lineno, offset=offset):
190 with self.assertRaises(SyntaxError) as cm:
191 compile(src, '<fragment>', 'exec')
192 self.assertEqual(cm.exception.lineno, lineno)
193 self.assertEqual(cm.exception.offset, offset)
194 if cm.exception.text is not None:
195 if not isinstance(src, str):
196 src = src.decode(encoding, 'replace')
197 line = src.split('\n')[lineno-1]
198 self.assertIn(line, cm.exception.text)
Serhiy Storchaka65fd0592014-01-21 22:26:52 +0200199
Batuhan Taskaya76c1b4d2020-05-01 16:13:43 +0300200 def testSyntaxErrorOffset(self):
201 check = self.check
Serhiy Storchaka65fd0592014-01-21 22:26:52 +0200202 check('def fact(x):\n\treturn x!\n', 2, 10)
203 check('1 +\n', 1, 4)
204 check('def spam():\n print(1)\n print(2)', 3, 10)
205 check('Python = "Python" +', 1, 20)
206 check('Python = "\u1e54\xfd\u0163\u0125\xf2\xf1" +', 1, 20)
Serhiy Storchaka0cc6b5e2020-02-12 12:17:00 +0200207 check(b'# -*- coding: cp1251 -*-\nPython = "\xcf\xb3\xf2\xee\xed" +',
208 2, 19, encoding='cp1251')
209 check(b'Python = "\xcf\xb3\xf2\xee\xed" +', 1, 18)
Batuhan Taskayaa698d522021-01-21 00:38:47 +0300210 check('x = "a', 1, 5)
Ammar Askar025eb982018-09-24 17:12:49 -0400211 check('lambda x: x = 2', 1, 1)
Lysandros Nikolaou15acc4e2020-10-27 20:54:20 +0200212 check('f{a + b + c}', 1, 2)
Miss Islington (bot)756b7b92021-05-03 18:06:45 -0700213 check('[file for str(file) in []\n])', 2, 2)
Lysandros Nikolaoue5fe5092021-01-14 23:36:30 +0200214 check('[\nfile\nfor str(file)\nin\n[]\n]', 3, 5)
215 check('[file for\n str(file) in []]', 2, 2)
Ammar Askar025eb982018-09-24 17:12:49 -0400216
217 # Errors thrown by compile.c
218 check('class foo:return 1', 1, 11)
219 check('def f():\n continue', 2, 3)
220 check('def f():\n break', 2, 3)
Mark Shannon8d4b1842021-05-06 13:38:50 +0100221 check('try:\n pass\nexcept:\n pass\nexcept ValueError:\n pass', 3, 1)
Ammar Askar025eb982018-09-24 17:12:49 -0400222
223 # Errors thrown by tokenizer.c
224 check('(0x+1)', 1, 3)
225 check('x = 0xI', 1, 6)
226 check('0010 + 2', 1, 4)
227 check('x = 32e-+4', 1, 8)
228 check('x = 0o9', 1, 6)
Serhiy Storchaka0cc6b5e2020-02-12 12:17:00 +0200229 check('\u03b1 = 0xI', 1, 6)
230 check(b'\xce\xb1 = 0xI', 1, 6)
231 check(b'# -*- coding: iso8859-7 -*-\n\xe1 = 0xI', 2, 6,
232 encoding='iso8859-7')
Pablo Galindo11a7f152020-04-21 01:53:04 +0100233 check(b"""if 1:
234 def foo():
235 '''
236
237 def bar():
238 pass
239
240 def baz():
241 '''quux'''
Batuhan Taskayaa698d522021-01-21 00:38:47 +0300242 """, 9, 24)
Pablo Galindobcc30362020-05-14 21:11:48 +0100243 check("pass\npass\npass\n(1+)\npass\npass\npass", 4, 4)
244 check("(1+)", 1, 4)
Miss Islington (bot)1afaaf52021-05-15 10:39:18 -0700245 check("[interesting\nfoo()\n", 1, 1)
Ammar Askar025eb982018-09-24 17:12:49 -0400246
247 # Errors thrown by symtable.c
Serhiy Storchakab619b092018-11-27 09:40:29 +0200248 check('x = [(yield i) for i in range(3)]', 1, 5)
Ammar Askar025eb982018-09-24 17:12:49 -0400249 check('def f():\n from _ import *', 1, 1)
250 check('def f(x, x):\n pass', 1, 1)
251 check('def f(x):\n nonlocal x', 2, 3)
252 check('def f(x):\n x = 1\n global x', 3, 3)
253 check('nonlocal x', 1, 1)
254 check('def f():\n global x\n nonlocal x', 2, 3)
255
Ammar Askar025eb982018-09-24 17:12:49 -0400256 # Errors thrown by future.c
257 check('from __future__ import doesnt_exist', 1, 1)
258 check('from __future__ import braces', 1, 1)
259 check('x=1\nfrom __future__ import division', 2, 1)
Pablo Galindoa77aac42021-04-23 14:27:05 +0100260 check('foo(1=2)', 1, 5)
Lysandros Nikolaoua15c9b32020-05-13 22:36:27 +0300261 check('def f():\n x, y: int', 2, 3)
262 check('[*x for x in xs]', 1, 2)
263 check('foo(x for x in range(10), 100)', 1, 5)
Lysandros Nikolaou01ece632020-06-19 02:10:43 +0300264 check('for 1 in []: pass', 1, 5)
Pablo Galindoa77aac42021-04-23 14:27:05 +0100265 check('(yield i) = 2', 1, 2)
266 check('def f(*):\n pass', 1, 7)
Serhiy Storchaka65fd0592014-01-21 22:26:52 +0200267
Benjamin Peterson17e0bbc2010-06-28 15:39:55 +0000268 @cpython_only
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000269 def testSettingException(self):
270 # test that setting an exception at the C level works even if the
271 # exception object can't be constructed.
Jeremy Hyltonede049b2001-09-26 20:01:13 +0000272
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000273 class BadException(Exception):
274 def __init__(self_):
Collin Winter828f04a2007-08-31 00:04:24 +0000275 raise RuntimeError("can't instantiate BadException")
Finn Bockaa3dc452001-12-08 10:15:48 +0000276
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000277 class InvalidException:
278 pass
Thomas Wouters303de6a2006-04-20 22:42:37 +0000279
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000280 def test_capi1():
281 import _testcapi
282 try:
283 _testcapi.raise_exception(BadException, 1)
Guido van Rossumb940e112007-01-10 16:19:56 +0000284 except TypeError as err:
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000285 exc, err, tb = sys.exc_info()
286 co = tb.tb_frame.f_code
Ezio Melottib3aedd42010-11-20 19:04:17 +0000287 self.assertEqual(co.co_name, "test_capi1")
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000288 self.assertTrue(co.co_filename.endswith('test_exceptions.py'))
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000289 else:
290 self.fail("Expected exception")
Jeremy Hyltonede049b2001-09-26 20:01:13 +0000291
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000292 def test_capi2():
293 import _testcapi
294 try:
295 _testcapi.raise_exception(BadException, 0)
Guido van Rossumb940e112007-01-10 16:19:56 +0000296 except RuntimeError as err:
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000297 exc, err, tb = sys.exc_info()
298 co = tb.tb_frame.f_code
Ezio Melottib3aedd42010-11-20 19:04:17 +0000299 self.assertEqual(co.co_name, "__init__")
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000300 self.assertTrue(co.co_filename.endswith('test_exceptions.py'))
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000301 co2 = tb.tb_frame.f_back.f_code
Ezio Melottib3aedd42010-11-20 19:04:17 +0000302 self.assertEqual(co2.co_name, "test_capi2")
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000303 else:
304 self.fail("Expected exception")
Thomas Wouters477c8d52006-05-27 19:21:47 +0000305
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000306 def test_capi3():
307 import _testcapi
308 self.assertRaises(SystemError, _testcapi.raise_exception,
309 InvalidException, 1)
310
311 if not sys.platform.startswith('java'):
312 test_capi1()
313 test_capi2()
314 test_capi3()
315
Thomas Wouters89f507f2006-12-13 04:49:30 +0000316 def test_WindowsError(self):
317 try:
318 WindowsError
319 except NameError:
320 pass
321 else:
Antoine Pitrou6b4883d2011-10-12 02:54:14 +0200322 self.assertIs(WindowsError, OSError)
323 self.assertEqual(str(OSError(1001)), "1001")
324 self.assertEqual(str(OSError(1001, "message")),
325 "[Errno 1001] message")
326 # POSIX errno (9 aka EBADF) is untranslated
327 w = OSError(9, 'foo', 'bar')
328 self.assertEqual(w.errno, 9)
329 self.assertEqual(w.winerror, None)
330 self.assertEqual(str(w), "[Errno 9] foo: 'bar'")
331 # ERROR_PATH_NOT_FOUND (win error 3) becomes ENOENT (2)
332 w = OSError(0, 'foo', 'bar', 3)
333 self.assertEqual(w.errno, 2)
334 self.assertEqual(w.winerror, 3)
335 self.assertEqual(w.strerror, 'foo')
336 self.assertEqual(w.filename, 'bar')
Martin Panter5487c132015-10-26 11:05:42 +0000337 self.assertEqual(w.filename2, None)
Richard Oudkerk30147712012-08-28 19:33:26 +0100338 self.assertEqual(str(w), "[WinError 3] foo: 'bar'")
Antoine Pitrou6b4883d2011-10-12 02:54:14 +0200339 # Unknown win error becomes EINVAL (22)
340 w = OSError(0, 'foo', None, 1001)
341 self.assertEqual(w.errno, 22)
342 self.assertEqual(w.winerror, 1001)
343 self.assertEqual(w.strerror, 'foo')
344 self.assertEqual(w.filename, None)
Martin Panter5487c132015-10-26 11:05:42 +0000345 self.assertEqual(w.filename2, None)
Richard Oudkerk30147712012-08-28 19:33:26 +0100346 self.assertEqual(str(w), "[WinError 1001] foo")
Antoine Pitrou6b4883d2011-10-12 02:54:14 +0200347 # Non-numeric "errno"
348 w = OSError('bar', 'foo')
349 self.assertEqual(w.errno, 'bar')
350 self.assertEqual(w.winerror, None)
351 self.assertEqual(w.strerror, 'foo')
352 self.assertEqual(w.filename, None)
Martin Panter5487c132015-10-26 11:05:42 +0000353 self.assertEqual(w.filename2, None)
Thomas Wouters89f507f2006-12-13 04:49:30 +0000354
Victor Stinnerd223fa62015-04-02 14:17:38 +0200355 @unittest.skipUnless(sys.platform == 'win32',
356 'test specific to Windows')
Serhiy Storchakaf41f8f92015-04-02 09:47:27 +0300357 def test_windows_message(self):
358 """Should fill in unknown error code in Windows error message"""
Victor Stinnerd223fa62015-04-02 14:17:38 +0200359 ctypes = import_module('ctypes')
360 # this error code has no message, Python formats it as hexadecimal
361 code = 3765269347
362 with self.assertRaisesRegex(OSError, 'Windows Error 0x%x' % code):
363 ctypes.pythonapi.PyErr_SetFromWindowsErr(code)
Serhiy Storchakaf41f8f92015-04-02 09:47:27 +0300364
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000365 def testAttributes(self):
366 # test that exception attributes are happy
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000367
368 exceptionList = [
Guido van Rossumebe3e162007-05-17 18:20:34 +0000369 (BaseException, (), {'args' : ()}),
370 (BaseException, (1, ), {'args' : (1,)}),
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000371 (BaseException, ('foo',),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000372 {'args' : ('foo',)}),
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000373 (BaseException, ('foo', 1),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000374 {'args' : ('foo', 1)}),
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000375 (SystemExit, ('foo',),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000376 {'args' : ('foo',), 'code' : 'foo'}),
Andrew Svetlovf7a17b42012-12-25 16:47:37 +0200377 (OSError, ('foo',),
Martin Panter5487c132015-10-26 11:05:42 +0000378 {'args' : ('foo',), 'filename' : None, 'filename2' : None,
Thomas Wouters89f507f2006-12-13 04:49:30 +0000379 'errno' : None, 'strerror' : None}),
Andrew Svetlovf7a17b42012-12-25 16:47:37 +0200380 (OSError, ('foo', 'bar'),
Martin Panter5487c132015-10-26 11:05:42 +0000381 {'args' : ('foo', 'bar'),
382 'filename' : None, 'filename2' : None,
Thomas Wouters89f507f2006-12-13 04:49:30 +0000383 'errno' : 'foo', 'strerror' : 'bar'}),
Andrew Svetlovf7a17b42012-12-25 16:47:37 +0200384 (OSError, ('foo', 'bar', 'baz'),
Martin Panter5487c132015-10-26 11:05:42 +0000385 {'args' : ('foo', 'bar'),
386 'filename' : 'baz', 'filename2' : None,
Thomas Wouters89f507f2006-12-13 04:49:30 +0000387 'errno' : 'foo', 'strerror' : 'bar'}),
Larry Hastingsb0827312014-02-09 22:05:19 -0800388 (OSError, ('foo', 'bar', 'baz', None, 'quux'),
389 {'args' : ('foo', 'bar'), 'filename' : 'baz', 'filename2': 'quux'}),
Andrew Svetlov3438fa42012-12-17 23:35:18 +0200390 (OSError, ('errnoStr', 'strErrorStr', 'filenameStr'),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000391 {'args' : ('errnoStr', 'strErrorStr'),
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000392 'strerror' : 'strErrorStr', 'errno' : 'errnoStr',
393 'filename' : 'filenameStr'}),
Andrew Svetlov3438fa42012-12-17 23:35:18 +0200394 (OSError, (1, 'strErrorStr', 'filenameStr'),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000395 {'args' : (1, 'strErrorStr'), 'errno' : 1,
Martin Panter5487c132015-10-26 11:05:42 +0000396 'strerror' : 'strErrorStr',
397 'filename' : 'filenameStr', 'filename2' : None}),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000398 (SyntaxError, (), {'msg' : None, 'text' : None,
Guido van Rossumd8faa362007-04-27 19:54:29 +0000399 'filename' : None, 'lineno' : None, 'offset' : None,
Pablo Galindoa77aac42021-04-23 14:27:05 +0100400 'end_offset': None, 'print_file_and_line' : None}),
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000401 (SyntaxError, ('msgStr',),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000402 {'args' : ('msgStr',), 'text' : None,
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000403 'print_file_and_line' : None, 'msg' : 'msgStr',
Pablo Galindoa77aac42021-04-23 14:27:05 +0100404 'filename' : None, 'lineno' : None, 'offset' : None,
405 'end_offset': None}),
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000406 (SyntaxError, ('msgStr', ('filenameStr', 'linenoStr', 'offsetStr',
Pablo Galindoa77aac42021-04-23 14:27:05 +0100407 'textStr', 'endLinenoStr', 'endOffsetStr')),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000408 {'offset' : 'offsetStr', 'text' : 'textStr',
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000409 'args' : ('msgStr', ('filenameStr', 'linenoStr',
Pablo Galindoa77aac42021-04-23 14:27:05 +0100410 'offsetStr', 'textStr',
411 'endLinenoStr', 'endOffsetStr')),
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000412 'print_file_and_line' : None, 'msg' : 'msgStr',
Pablo Galindoa77aac42021-04-23 14:27:05 +0100413 'filename' : 'filenameStr', 'lineno' : 'linenoStr',
414 'end_lineno': 'endLinenoStr', 'end_offset': 'endOffsetStr'}),
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000415 (SyntaxError, ('msgStr', 'filenameStr', 'linenoStr', 'offsetStr',
Pablo Galindoa77aac42021-04-23 14:27:05 +0100416 'textStr', 'endLinenoStr', 'endOffsetStr',
417 'print_file_and_lineStr'),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000418 {'text' : None,
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000419 'args' : ('msgStr', 'filenameStr', 'linenoStr', 'offsetStr',
Pablo Galindoa77aac42021-04-23 14:27:05 +0100420 'textStr', 'endLinenoStr', 'endOffsetStr',
421 'print_file_and_lineStr'),
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000422 'print_file_and_line' : None, 'msg' : 'msgStr',
Pablo Galindoa77aac42021-04-23 14:27:05 +0100423 'filename' : None, 'lineno' : None, 'offset' : None,
424 'end_lineno': None, 'end_offset': None}),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000425 (UnicodeError, (), {'args' : (),}),
Walter Dörwaldeceb0fb2007-05-24 17:49:56 +0000426 (UnicodeEncodeError, ('ascii', 'a', 0, 1,
427 'ordinal not in range'),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000428 {'args' : ('ascii', 'a', 0, 1,
Thomas Wouters89f507f2006-12-13 04:49:30 +0000429 'ordinal not in range'),
Guido van Rossumef87d6e2007-05-02 19:09:54 +0000430 'encoding' : 'ascii', 'object' : 'a',
Thomas Wouters89f507f2006-12-13 04:49:30 +0000431 'start' : 0, 'reason' : 'ordinal not in range'}),
Guido van Rossum254348e2007-11-21 19:29:53 +0000432 (UnicodeDecodeError, ('ascii', bytearray(b'\xff'), 0, 1,
Guido van Rossum98297ee2007-11-06 21:34:58 +0000433 'ordinal not in range'),
Guido van Rossum254348e2007-11-21 19:29:53 +0000434 {'args' : ('ascii', bytearray(b'\xff'), 0, 1,
Guido van Rossum98297ee2007-11-06 21:34:58 +0000435 'ordinal not in range'),
436 'encoding' : 'ascii', 'object' : b'\xff',
437 'start' : 0, 'reason' : 'ordinal not in range'}),
Walter Dörwaldeceb0fb2007-05-24 17:49:56 +0000438 (UnicodeDecodeError, ('ascii', b'\xff', 0, 1,
439 'ordinal not in range'),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000440 {'args' : ('ascii', b'\xff', 0, 1,
Thomas Wouters89f507f2006-12-13 04:49:30 +0000441 'ordinal not in range'),
Guido van Rossumb8142c32007-05-08 17:49:10 +0000442 'encoding' : 'ascii', 'object' : b'\xff',
Thomas Wouters89f507f2006-12-13 04:49:30 +0000443 'start' : 0, 'reason' : 'ordinal not in range'}),
Walter Dörwaldeceb0fb2007-05-24 17:49:56 +0000444 (UnicodeTranslateError, ("\u3042", 0, 1, "ouch"),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000445 {'args' : ('\u3042', 0, 1, 'ouch'),
Guido van Rossumef87d6e2007-05-02 19:09:54 +0000446 'object' : '\u3042', 'reason' : 'ouch',
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000447 'start' : 0, 'end' : 1}),
Richard Oudkerk5562d9d2012-07-28 17:45:28 +0100448 (NaiveException, ('foo',),
449 {'args': ('foo',), 'x': 'foo'}),
450 (SlottedNaiveException, ('foo',),
451 {'args': ('foo',), 'x': 'foo'}),
Thomas Wouters477c8d52006-05-27 19:21:47 +0000452 ]
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000453 try:
Antoine Pitrou6b4883d2011-10-12 02:54:14 +0200454 # More tests are in test_WindowsError
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000455 exceptionList.append(
456 (WindowsError, (1, 'strErrorStr', 'filenameStr'),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000457 {'args' : (1, 'strErrorStr'),
Antoine Pitrou6b4883d2011-10-12 02:54:14 +0200458 'strerror' : 'strErrorStr', 'winerror' : None,
Martin Panter5487c132015-10-26 11:05:42 +0000459 'errno' : 1,
460 'filename' : 'filenameStr', 'filename2' : None})
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000461 )
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000462 except NameError:
463 pass
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000464
Guido van Rossumebe3e162007-05-17 18:20:34 +0000465 for exc, args, expected in exceptionList:
466 try:
467 e = exc(*args)
468 except:
Guido van Rossum98297ee2007-11-06 21:34:58 +0000469 print("\nexc=%r, args=%r" % (exc, args), file=sys.stderr)
Pablo Galindoa77aac42021-04-23 14:27:05 +0100470 # raise
Guido van Rossumebe3e162007-05-17 18:20:34 +0000471 else:
472 # Verify module name
Richard Oudkerk5562d9d2012-07-28 17:45:28 +0100473 if not type(e).__name__.endswith('NaiveException'):
474 self.assertEqual(type(e).__module__, 'builtins')
Guido van Rossumebe3e162007-05-17 18:20:34 +0000475 # Verify no ref leaks in Exc_str()
476 s = str(e)
477 for checkArgName in expected:
478 value = getattr(e, checkArgName)
Ezio Melottib3aedd42010-11-20 19:04:17 +0000479 self.assertEqual(repr(value),
480 repr(expected[checkArgName]),
481 '%r.%s == %r, expected %r' % (
482 e, checkArgName,
483 value, expected[checkArgName]))
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000484
Guido van Rossumebe3e162007-05-17 18:20:34 +0000485 # test for pickling support
Guido van Rossum99603b02007-07-20 00:22:32 +0000486 for p in [pickle]:
Guido van Rossumebe3e162007-05-17 18:20:34 +0000487 for protocol in range(p.HIGHEST_PROTOCOL + 1):
488 s = p.dumps(e, protocol)
489 new = p.loads(s)
490 for checkArgName in expected:
491 got = repr(getattr(new, checkArgName))
492 want = repr(expected[checkArgName])
Ezio Melottib3aedd42010-11-20 19:04:17 +0000493 self.assertEqual(got, want,
494 'pickled "%r", attribute "%s' %
495 (e, checkArgName))
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000496
Collin Winter828f04a2007-08-31 00:04:24 +0000497 def testWithTraceback(self):
498 try:
499 raise IndexError(4)
500 except:
501 tb = sys.exc_info()[2]
502
503 e = BaseException().with_traceback(tb)
Ezio Melottie9615932010-01-24 19:26:24 +0000504 self.assertIsInstance(e, BaseException)
Collin Winter828f04a2007-08-31 00:04:24 +0000505 self.assertEqual(e.__traceback__, tb)
506
507 e = IndexError(5).with_traceback(tb)
Ezio Melottie9615932010-01-24 19:26:24 +0000508 self.assertIsInstance(e, IndexError)
Collin Winter828f04a2007-08-31 00:04:24 +0000509 self.assertEqual(e.__traceback__, tb)
510
511 class MyException(Exception):
512 pass
513
514 e = MyException().with_traceback(tb)
Ezio Melottie9615932010-01-24 19:26:24 +0000515 self.assertIsInstance(e, MyException)
Collin Winter828f04a2007-08-31 00:04:24 +0000516 self.assertEqual(e.__traceback__, tb)
517
518 def testInvalidTraceback(self):
519 try:
520 Exception().__traceback__ = 5
521 except TypeError as e:
Benjamin Peterson577473f2010-01-19 00:09:57 +0000522 self.assertIn("__traceback__ must be a traceback", str(e))
Collin Winter828f04a2007-08-31 00:04:24 +0000523 else:
524 self.fail("No exception raised")
525
Georg Brandlab6f2f62009-03-31 04:16:10 +0000526 def testInvalidAttrs(self):
527 self.assertRaises(TypeError, setattr, Exception(), '__cause__', 1)
528 self.assertRaises(TypeError, delattr, Exception(), '__cause__')
529 self.assertRaises(TypeError, setattr, Exception(), '__context__', 1)
530 self.assertRaises(TypeError, delattr, Exception(), '__context__')
531
Collin Winter828f04a2007-08-31 00:04:24 +0000532 def testNoneClearsTracebackAttr(self):
533 try:
534 raise IndexError(4)
535 except:
536 tb = sys.exc_info()[2]
537
538 e = Exception()
539 e.__traceback__ = tb
540 e.__traceback__ = None
541 self.assertEqual(e.__traceback__, None)
542
543 def testChainingAttrs(self):
544 e = Exception()
Nick Coghlanab7bf212012-02-26 17:49:52 +1000545 self.assertIsNone(e.__context__)
Benjamin Petersond5a1c442012-05-14 22:09:31 -0700546 self.assertIsNone(e.__cause__)
Collin Winter828f04a2007-08-31 00:04:24 +0000547
548 e = TypeError()
Nick Coghlanab7bf212012-02-26 17:49:52 +1000549 self.assertIsNone(e.__context__)
Benjamin Petersond5a1c442012-05-14 22:09:31 -0700550 self.assertIsNone(e.__cause__)
Collin Winter828f04a2007-08-31 00:04:24 +0000551
Andrew Svetlov3438fa42012-12-17 23:35:18 +0200552 class MyException(OSError):
Collin Winter828f04a2007-08-31 00:04:24 +0000553 pass
554
555 e = MyException()
Nick Coghlanab7bf212012-02-26 17:49:52 +1000556 self.assertIsNone(e.__context__)
Benjamin Petersond5a1c442012-05-14 22:09:31 -0700557 self.assertIsNone(e.__cause__)
Nick Coghlanab7bf212012-02-26 17:49:52 +1000558
559 def testChainingDescriptors(self):
560 try:
561 raise Exception()
562 except Exception as exc:
563 e = exc
564
565 self.assertIsNone(e.__context__)
Benjamin Petersond5a1c442012-05-14 22:09:31 -0700566 self.assertIsNone(e.__cause__)
567 self.assertFalse(e.__suppress_context__)
Nick Coghlanab7bf212012-02-26 17:49:52 +1000568
569 e.__context__ = NameError()
570 e.__cause__ = None
571 self.assertIsInstance(e.__context__, NameError)
572 self.assertIsNone(e.__cause__)
Benjamin Petersond5a1c442012-05-14 22:09:31 -0700573 self.assertTrue(e.__suppress_context__)
574 e.__suppress_context__ = False
575 self.assertFalse(e.__suppress_context__)
Collin Winter828f04a2007-08-31 00:04:24 +0000576
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000577 def testKeywordArgs(self):
578 # test that builtin exception don't take keyword args,
579 # but user-defined subclasses can if they want
580 self.assertRaises(TypeError, BaseException, a=1)
581
582 class DerivedException(BaseException):
583 def __init__(self, fancy_arg):
584 BaseException.__init__(self)
585 self.fancy_arg = fancy_arg
586
587 x = DerivedException(fancy_arg=42)
Ezio Melottib3aedd42010-11-20 19:04:17 +0000588 self.assertEqual(x.fancy_arg, 42)
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000589
Brett Cannon31f59292011-02-21 19:29:56 +0000590 @no_tracing
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000591 def testInfiniteRecursion(self):
592 def f():
593 return f()
Yury Selivanovf488fb42015-07-03 01:04:23 -0400594 self.assertRaises(RecursionError, f)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000595
596 def g():
597 try:
598 return g()
599 except ValueError:
600 return -1
Yury Selivanovf488fb42015-07-03 01:04:23 -0400601 self.assertRaises(RecursionError, g)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000602
Ezio Melotti2f5a78c2009-12-24 22:54:06 +0000603 def test_str(self):
604 # Make sure both instances and classes have a str representation.
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000605 self.assertTrue(str(Exception))
606 self.assertTrue(str(Exception('a')))
Ezio Melotti2f5a78c2009-12-24 22:54:06 +0000607 self.assertTrue(str(Exception('a', 'b')))
Thomas Wouters89f507f2006-12-13 04:49:30 +0000608
Barry Warsaw8d109cb2008-05-08 04:26:35 +0000609 def testExceptionCleanupNames(self):
610 # Make sure the local variable bound to the exception instance by
611 # an "except" statement is only visible inside the except block.
Guido van Rossumb940e112007-01-10 16:19:56 +0000612 try:
613 raise Exception()
614 except Exception as e:
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000615 self.assertTrue(e)
Guido van Rossumb940e112007-01-10 16:19:56 +0000616 del e
Ezio Melottib58e0bd2010-01-23 15:40:09 +0000617 self.assertNotIn('e', locals())
Guido van Rossumb940e112007-01-10 16:19:56 +0000618
Barry Warsaw8d109cb2008-05-08 04:26:35 +0000619 def testExceptionCleanupState(self):
620 # Make sure exception state is cleaned up as soon as the except
621 # block is left. See #2507
622
623 class MyException(Exception):
624 def __init__(self, obj):
625 self.obj = obj
626 class MyObj:
627 pass
628
629 def inner_raising_func():
630 # Create some references in exception value and traceback
631 local_ref = obj
632 raise MyException(obj)
633
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000634 # Qualified "except" with "as"
Barry Warsaw8d109cb2008-05-08 04:26:35 +0000635 obj = MyObj()
636 wr = weakref.ref(obj)
637 try:
638 inner_raising_func()
639 except MyException as e:
640 pass
641 obj = None
642 obj = wr()
Serhiy Storchakaf15c4d32017-03-30 18:05:08 +0300643 self.assertIsNone(obj)
Barry Warsaw8d109cb2008-05-08 04:26:35 +0000644
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000645 # Qualified "except" without "as"
646 obj = MyObj()
647 wr = weakref.ref(obj)
648 try:
649 inner_raising_func()
650 except MyException:
651 pass
652 obj = None
653 obj = wr()
Serhiy Storchakaf15c4d32017-03-30 18:05:08 +0300654 self.assertIsNone(obj)
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000655
656 # Bare "except"
657 obj = MyObj()
658 wr = weakref.ref(obj)
659 try:
660 inner_raising_func()
661 except:
662 pass
663 obj = None
664 obj = wr()
Serhiy Storchakaf15c4d32017-03-30 18:05:08 +0300665 self.assertIsNone(obj)
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000666
667 # "except" with premature block leave
668 obj = MyObj()
669 wr = weakref.ref(obj)
670 for i in [0]:
671 try:
672 inner_raising_func()
673 except:
674 break
675 obj = None
676 obj = wr()
Serhiy Storchakaf15c4d32017-03-30 18:05:08 +0300677 self.assertIsNone(obj)
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000678
679 # "except" block raising another exception
680 obj = MyObj()
681 wr = weakref.ref(obj)
682 try:
683 try:
684 inner_raising_func()
685 except:
686 raise KeyError
Guido van Rossumb4fb6e42008-06-14 20:20:24 +0000687 except KeyError as e:
688 # We want to test that the except block above got rid of
689 # the exception raised in inner_raising_func(), but it
690 # also ends up in the __context__ of the KeyError, so we
691 # must clear the latter manually for our test to succeed.
692 e.__context__ = None
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000693 obj = None
694 obj = wr()
Philip Jenveyb37ac8e2012-11-14 14:37:24 -0800695 # guarantee no ref cycles on CPython (don't gc_collect)
696 if check_impl_detail(cpython=False):
697 gc_collect()
Serhiy Storchakaf15c4d32017-03-30 18:05:08 +0300698 self.assertIsNone(obj)
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000699
700 # Some complicated construct
701 obj = MyObj()
702 wr = weakref.ref(obj)
703 try:
704 inner_raising_func()
705 except MyException:
706 try:
707 try:
708 raise
709 finally:
710 raise
711 except MyException:
712 pass
713 obj = None
Philip Jenveyb37ac8e2012-11-14 14:37:24 -0800714 if check_impl_detail(cpython=False):
715 gc_collect()
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000716 obj = wr()
Serhiy Storchakaf15c4d32017-03-30 18:05:08 +0300717 self.assertIsNone(obj)
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000718
719 # Inside an exception-silencing "with" block
720 class Context:
721 def __enter__(self):
722 return self
723 def __exit__ (self, exc_type, exc_value, exc_tb):
724 return True
725 obj = MyObj()
726 wr = weakref.ref(obj)
727 with Context():
728 inner_raising_func()
729 obj = None
Philip Jenveyb37ac8e2012-11-14 14:37:24 -0800730 if check_impl_detail(cpython=False):
731 gc_collect()
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000732 obj = wr()
Serhiy Storchakaf15c4d32017-03-30 18:05:08 +0300733 self.assertIsNone(obj)
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000734
Amaury Forgeot d'Arcba117ef2010-09-10 21:39:53 +0000735 def test_exception_target_in_nested_scope(self):
736 # issue 4617: This used to raise a SyntaxError
737 # "can not delete variable 'e' referenced in nested scope"
738 def print_error():
739 e
740 try:
741 something
742 except Exception as e:
743 print_error()
744 # implicit "del e" here
745
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000746 def test_generator_leaking(self):
747 # Test that generator exception state doesn't leak into the calling
748 # frame
749 def yield_raise():
750 try:
751 raise KeyError("caught")
752 except KeyError:
753 yield sys.exc_info()[0]
754 yield sys.exc_info()[0]
755 yield sys.exc_info()[0]
756 g = yield_raise()
Ezio Melottib3aedd42010-11-20 19:04:17 +0000757 self.assertEqual(next(g), KeyError)
758 self.assertEqual(sys.exc_info()[0], None)
759 self.assertEqual(next(g), KeyError)
760 self.assertEqual(sys.exc_info()[0], None)
761 self.assertEqual(next(g), None)
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000762
763 # Same test, but inside an exception handler
764 try:
765 raise TypeError("foo")
766 except TypeError:
767 g = yield_raise()
Ezio Melottib3aedd42010-11-20 19:04:17 +0000768 self.assertEqual(next(g), KeyError)
769 self.assertEqual(sys.exc_info()[0], TypeError)
770 self.assertEqual(next(g), KeyError)
771 self.assertEqual(sys.exc_info()[0], TypeError)
772 self.assertEqual(next(g), TypeError)
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000773 del g
Ezio Melottib3aedd42010-11-20 19:04:17 +0000774 self.assertEqual(sys.exc_info()[0], TypeError)
Thomas Wouters89f507f2006-12-13 04:49:30 +0000775
Benjamin Peterson83195c32011-07-03 13:44:00 -0500776 def test_generator_leaking2(self):
777 # See issue 12475.
778 def g():
779 yield
780 try:
781 raise RuntimeError
782 except RuntimeError:
783 it = g()
784 next(it)
785 try:
786 next(it)
787 except StopIteration:
788 pass
789 self.assertEqual(sys.exc_info(), (None, None, None))
790
Antoine Pitrouc4c19b32015-03-18 22:22:46 +0100791 def test_generator_leaking3(self):
792 # See issue #23353. When gen.throw() is called, the caller's
793 # exception state should be save and restored.
794 def g():
795 try:
796 yield
797 except ZeroDivisionError:
798 yield sys.exc_info()[1]
799 it = g()
800 next(it)
801 try:
802 1/0
803 except ZeroDivisionError as e:
804 self.assertIs(sys.exc_info()[1], e)
805 gen_exc = it.throw(e)
806 self.assertIs(sys.exc_info()[1], e)
807 self.assertIs(gen_exc, e)
808 self.assertEqual(sys.exc_info(), (None, None, None))
809
810 def test_generator_leaking4(self):
811 # See issue #23353. When an exception is raised by a generator,
812 # the caller's exception state should still be restored.
813 def g():
814 try:
815 1/0
816 except ZeroDivisionError:
817 yield sys.exc_info()[0]
818 raise
819 it = g()
820 try:
821 raise TypeError
822 except TypeError:
823 # The caller's exception state (TypeError) is temporarily
824 # saved in the generator.
825 tp = next(it)
826 self.assertIs(tp, ZeroDivisionError)
827 try:
828 next(it)
829 # We can't check it immediately, but while next() returns
830 # with an exception, it shouldn't have restored the old
831 # exception state (TypeError).
832 except ZeroDivisionError as e:
833 self.assertIs(sys.exc_info()[1], e)
834 # We used to find TypeError here.
835 self.assertEqual(sys.exc_info(), (None, None, None))
836
Benjamin Petersonac913412011-07-03 16:25:11 -0500837 def test_generator_doesnt_retain_old_exc(self):
838 def g():
839 self.assertIsInstance(sys.exc_info()[1], RuntimeError)
840 yield
841 self.assertEqual(sys.exc_info(), (None, None, None))
842 it = g()
843 try:
844 raise RuntimeError
845 except RuntimeError:
846 next(it)
847 self.assertRaises(StopIteration, next, it)
848
Benjamin Petersonae5f2f42010-03-07 17:10:51 +0000849 def test_generator_finalizing_and_exc_info(self):
850 # See #7173
851 def simple_gen():
852 yield 1
853 def run_gen():
854 gen = simple_gen()
855 try:
856 raise RuntimeError
857 except RuntimeError:
858 return next(gen)
859 run_gen()
860 gc_collect()
861 self.assertEqual(sys.exc_info(), (None, None, None))
862
Antoine Pitroua370fcf2011-08-20 14:15:03 +0200863 def _check_generator_cleanup_exc_state(self, testfunc):
864 # Issue #12791: exception state is cleaned up as soon as a generator
865 # is closed (reference cycles are broken).
866 class MyException(Exception):
867 def __init__(self, obj):
868 self.obj = obj
869 class MyObj:
870 pass
871
872 def raising_gen():
873 try:
874 raise MyException(obj)
875 except MyException:
876 yield
877
878 obj = MyObj()
879 wr = weakref.ref(obj)
880 g = raising_gen()
881 next(g)
882 testfunc(g)
883 g = obj = None
884 obj = wr()
Serhiy Storchakaf15c4d32017-03-30 18:05:08 +0300885 self.assertIsNone(obj)
Antoine Pitroua370fcf2011-08-20 14:15:03 +0200886
887 def test_generator_throw_cleanup_exc_state(self):
888 def do_throw(g):
889 try:
890 g.throw(RuntimeError())
891 except RuntimeError:
892 pass
893 self._check_generator_cleanup_exc_state(do_throw)
894
895 def test_generator_close_cleanup_exc_state(self):
896 def do_close(g):
897 g.close()
898 self._check_generator_cleanup_exc_state(do_close)
899
900 def test_generator_del_cleanup_exc_state(self):
901 def do_del(g):
902 g = None
903 self._check_generator_cleanup_exc_state(do_del)
904
905 def test_generator_next_cleanup_exc_state(self):
906 def do_next(g):
907 try:
908 next(g)
909 except StopIteration:
910 pass
911 else:
912 self.fail("should have raised StopIteration")
913 self._check_generator_cleanup_exc_state(do_next)
914
915 def test_generator_send_cleanup_exc_state(self):
916 def do_send(g):
917 try:
918 g.send(None)
919 except StopIteration:
920 pass
921 else:
922 self.fail("should have raised StopIteration")
923 self._check_generator_cleanup_exc_state(do_send)
924
Benjamin Peterson27d63672008-06-15 20:09:12 +0000925 def test_3114(self):
926 # Bug #3114: in its destructor, MyObject retrieves a pointer to
927 # obsolete and/or deallocated objects.
Benjamin Peterson979f3112008-06-15 00:05:44 +0000928 class MyObject:
929 def __del__(self):
930 nonlocal e
931 e = sys.exc_info()
932 e = ()
933 try:
934 raise Exception(MyObject())
935 except:
936 pass
Ezio Melottib3aedd42010-11-20 19:04:17 +0000937 self.assertEqual(e, (None, None, None))
Benjamin Peterson979f3112008-06-15 00:05:44 +0000938
Benjamin Peterson24dfb052014-04-02 12:05:35 -0400939 def test_unicode_change_attributes(self):
Eric Smith0facd772010-02-24 15:42:29 +0000940 # See issue 7309. This was a crasher.
941
942 u = UnicodeEncodeError('baz', 'xxxxx', 1, 5, 'foo')
943 self.assertEqual(str(u), "'baz' codec can't encode characters in position 1-4: foo")
944 u.end = 2
945 self.assertEqual(str(u), "'baz' codec can't encode character '\\x78' in position 1: foo")
946 u.end = 5
947 u.reason = 0x345345345345345345
948 self.assertEqual(str(u), "'baz' codec can't encode characters in position 1-4: 965230951443685724997")
949 u.encoding = 4000
950 self.assertEqual(str(u), "'4000' codec can't encode characters in position 1-4: 965230951443685724997")
951 u.start = 1000
952 self.assertEqual(str(u), "'4000' codec can't encode characters in position 1000-4: 965230951443685724997")
953
954 u = UnicodeDecodeError('baz', b'xxxxx', 1, 5, 'foo')
955 self.assertEqual(str(u), "'baz' codec can't decode bytes in position 1-4: foo")
956 u.end = 2
957 self.assertEqual(str(u), "'baz' codec can't decode byte 0x78 in position 1: foo")
958 u.end = 5
959 u.reason = 0x345345345345345345
960 self.assertEqual(str(u), "'baz' codec can't decode bytes in position 1-4: 965230951443685724997")
961 u.encoding = 4000
962 self.assertEqual(str(u), "'4000' codec can't decode bytes in position 1-4: 965230951443685724997")
963 u.start = 1000
964 self.assertEqual(str(u), "'4000' codec can't decode bytes in position 1000-4: 965230951443685724997")
965
966 u = UnicodeTranslateError('xxxx', 1, 5, 'foo')
967 self.assertEqual(str(u), "can't translate characters in position 1-4: foo")
968 u.end = 2
969 self.assertEqual(str(u), "can't translate character '\\x78' in position 1: foo")
970 u.end = 5
971 u.reason = 0x345345345345345345
972 self.assertEqual(str(u), "can't translate characters in position 1-4: 965230951443685724997")
973 u.start = 1000
974 self.assertEqual(str(u), "can't translate characters in position 1000-4: 965230951443685724997")
Benjamin Peterson6e7740c2008-08-20 23:23:34 +0000975
Benjamin Peterson9b09ba12014-04-02 12:15:06 -0400976 def test_unicode_errors_no_object(self):
977 # See issue #21134.
Benjamin Petersone3311212014-04-02 15:51:38 -0400978 klasses = UnicodeEncodeError, UnicodeDecodeError, UnicodeTranslateError
Benjamin Peterson9b09ba12014-04-02 12:15:06 -0400979 for klass in klasses:
980 self.assertEqual(str(klass.__new__(klass)), "")
981
Brett Cannon31f59292011-02-21 19:29:56 +0000982 @no_tracing
Benjamin Peterson69c88f72008-07-31 01:47:08 +0000983 def test_badisinstance(self):
984 # Bug #2542: if issubclass(e, MyException) raises an exception,
985 # it should be ignored
986 class Meta(type):
987 def __subclasscheck__(cls, subclass):
988 raise ValueError()
989 class MyException(Exception, metaclass=Meta):
990 pass
991
Martin Panter3263f682016-02-28 03:16:11 +0000992 with captured_stderr() as stderr:
Benjamin Peterson69c88f72008-07-31 01:47:08 +0000993 try:
994 raise KeyError()
995 except MyException as e:
996 self.fail("exception should not be a MyException")
997 except KeyError:
998 pass
999 except:
Antoine Pitrouec569b72008-08-26 22:40:48 +00001000 self.fail("Should have raised KeyError")
Benjamin Peterson69c88f72008-07-31 01:47:08 +00001001 else:
Antoine Pitrouec569b72008-08-26 22:40:48 +00001002 self.fail("Should have raised KeyError")
1003
1004 def g():
1005 try:
1006 return g()
Yury Selivanovf488fb42015-07-03 01:04:23 -04001007 except RecursionError:
Antoine Pitrouec569b72008-08-26 22:40:48 +00001008 return sys.exc_info()
1009 e, v, tb = g()
Serhiy Storchakaf15c4d32017-03-30 18:05:08 +03001010 self.assertIsInstance(v, RecursionError, type(v))
Benjamin Peterson577473f2010-01-19 00:09:57 +00001011 self.assertIn("maximum recursion depth exceeded", str(v))
Benjamin Peterson69c88f72008-07-31 01:47:08 +00001012
xdegaye56d1f5c2017-10-26 15:09:06 +02001013 @cpython_only
1014 def test_recursion_normalizing_exception(self):
1015 # Issue #22898.
1016 # Test that a RecursionError is raised when tstate->recursion_depth is
1017 # equal to recursion_limit in PyErr_NormalizeException() and check
1018 # that a ResourceWarning is printed.
1019 # Prior to #22898, the recursivity of PyErr_NormalizeException() was
luzpaza5293b42017-11-05 07:37:50 -06001020 # controlled by tstate->recursion_depth and a PyExc_RecursionErrorInst
xdegaye56d1f5c2017-10-26 15:09:06 +02001021 # singleton was being used in that case, that held traceback data and
1022 # locals indefinitely and would cause a segfault in _PyExc_Fini() upon
1023 # finalization of these locals.
1024 code = """if 1:
1025 import sys
Victor Stinner3f2f4fe2020-03-13 13:07:31 +01001026 from _testinternalcapi import get_recursion_depth
xdegaye56d1f5c2017-10-26 15:09:06 +02001027
1028 class MyException(Exception): pass
1029
1030 def setrecursionlimit(depth):
1031 while 1:
1032 try:
1033 sys.setrecursionlimit(depth)
1034 return depth
1035 except RecursionError:
1036 # sys.setrecursionlimit() raises a RecursionError if
1037 # the new recursion limit is too low (issue #25274).
1038 depth += 1
1039
1040 def recurse(cnt):
1041 cnt -= 1
1042 if cnt:
1043 recurse(cnt)
1044 else:
1045 generator.throw(MyException)
1046
1047 def gen():
1048 f = open(%a, mode='rb', buffering=0)
1049 yield
1050
1051 generator = gen()
1052 next(generator)
1053 recursionlimit = sys.getrecursionlimit()
1054 depth = get_recursion_depth()
1055 try:
1056 # Upon the last recursive invocation of recurse(),
1057 # tstate->recursion_depth is equal to (recursion_limit - 1)
1058 # and is equal to recursion_limit when _gen_throw() calls
1059 # PyErr_NormalizeException().
Mark Shannon4e7a69b2020-12-02 13:30:55 +00001060 recurse(setrecursionlimit(depth + 2) - depth)
xdegaye56d1f5c2017-10-26 15:09:06 +02001061 finally:
1062 sys.setrecursionlimit(recursionlimit)
1063 print('Done.')
1064 """ % __file__
1065 rc, out, err = script_helper.assert_python_failure("-Wd", "-c", code)
1066 # Check that the program does not fail with SIGABRT.
1067 self.assertEqual(rc, 1)
1068 self.assertIn(b'RecursionError', err)
1069 self.assertIn(b'ResourceWarning', err)
1070 self.assertIn(b'Done.', out)
1071
1072 @cpython_only
1073 def test_recursion_normalizing_infinite_exception(self):
1074 # Issue #30697. Test that a RecursionError is raised when
1075 # PyErr_NormalizeException() maximum recursion depth has been
1076 # exceeded.
1077 code = """if 1:
1078 import _testcapi
1079 try:
1080 raise _testcapi.RecursingInfinitelyError
1081 finally:
1082 print('Done.')
1083 """
1084 rc, out, err = script_helper.assert_python_failure("-c", code)
1085 self.assertEqual(rc, 1)
1086 self.assertIn(b'RecursionError: maximum recursion depth exceeded '
1087 b'while normalizing an exception', err)
1088 self.assertIn(b'Done.', out)
1089
Mark Shannon4e7a69b2020-12-02 13:30:55 +00001090
1091 def test_recursion_in_except_handler(self):
1092
1093 def set_relative_recursion_limit(n):
1094 depth = 1
1095 while True:
1096 try:
1097 sys.setrecursionlimit(depth)
1098 except RecursionError:
1099 depth += 1
1100 else:
1101 break
1102 sys.setrecursionlimit(depth+n)
1103
1104 def recurse_in_except():
1105 try:
1106 1/0
1107 except:
1108 recurse_in_except()
1109
1110 def recurse_after_except():
1111 try:
1112 1/0
1113 except:
1114 pass
1115 recurse_after_except()
1116
1117 def recurse_in_body_and_except():
1118 try:
1119 recurse_in_body_and_except()
1120 except:
1121 recurse_in_body_and_except()
1122
1123 recursionlimit = sys.getrecursionlimit()
1124 try:
1125 set_relative_recursion_limit(10)
1126 for func in (recurse_in_except, recurse_after_except, recurse_in_body_and_except):
1127 with self.subTest(func=func):
1128 try:
1129 func()
1130 except RecursionError:
1131 pass
1132 else:
1133 self.fail("Should have raised a RecursionError")
1134 finally:
1135 sys.setrecursionlimit(recursionlimit)
1136
1137
xdegaye56d1f5c2017-10-26 15:09:06 +02001138 @cpython_only
1139 def test_recursion_normalizing_with_no_memory(self):
1140 # Issue #30697. Test that in the abort that occurs when there is no
1141 # memory left and the size of the Python frames stack is greater than
1142 # the size of the list of preallocated MemoryError instances, the
1143 # Fatal Python error message mentions MemoryError.
1144 code = """if 1:
1145 import _testcapi
1146 class C(): pass
1147 def recurse(cnt):
1148 cnt -= 1
1149 if cnt:
1150 recurse(cnt)
1151 else:
1152 _testcapi.set_nomemory(0)
1153 C()
1154 recurse(16)
1155 """
1156 with SuppressCrashReport():
1157 rc, out, err = script_helper.assert_python_failure("-c", code)
Victor Stinner9e5d30c2020-03-07 00:54:20 +01001158 self.assertIn(b'Fatal Python error: _PyErr_NormalizeException: '
1159 b'Cannot recover from MemoryErrors while '
1160 b'normalizing exceptions.', err)
Amaury Forgeot d'Arce19cadb2008-07-31 22:56:02 +00001161
Serhiy Storchaka5cfc79d2014-02-07 10:06:39 +02001162 @cpython_only
Amaury Forgeot d'Arce19cadb2008-07-31 22:56:02 +00001163 def test_MemoryError(self):
1164 # PyErr_NoMemory always raises the same exception instance.
1165 # Check that the traceback is not doubled.
1166 import traceback
Benjamin Peterson0067bd62008-08-16 16:11:03 +00001167 from _testcapi import raise_memoryerror
Amaury Forgeot d'Arce19cadb2008-07-31 22:56:02 +00001168 def raiseMemError():
1169 try:
Benjamin Peterson0067bd62008-08-16 16:11:03 +00001170 raise_memoryerror()
Amaury Forgeot d'Arce19cadb2008-07-31 22:56:02 +00001171 except MemoryError as e:
1172 tb = e.__traceback__
1173 else:
Mark Shannon4e7a69b2020-12-02 13:30:55 +00001174 self.fail("Should have raised a MemoryError")
Amaury Forgeot d'Arce19cadb2008-07-31 22:56:02 +00001175 return traceback.format_tb(tb)
1176
1177 tb1 = raiseMemError()
1178 tb2 = raiseMemError()
1179 self.assertEqual(tb1, tb2)
1180
Benjamin Peterson17e0bbc2010-06-28 15:39:55 +00001181 @cpython_only
Georg Brandl1e28a272009-12-28 08:41:01 +00001182 def test_exception_with_doc(self):
1183 import _testcapi
1184 doc2 = "This is a test docstring."
1185 doc4 = "This is another test docstring."
1186
1187 self.assertRaises(SystemError, _testcapi.make_exception_with_doc,
1188 "error1")
1189
1190 # test basic usage of PyErr_NewException
1191 error1 = _testcapi.make_exception_with_doc("_testcapi.error1")
1192 self.assertIs(type(error1), type)
1193 self.assertTrue(issubclass(error1, Exception))
1194 self.assertIsNone(error1.__doc__)
1195
1196 # test with given docstring
1197 error2 = _testcapi.make_exception_with_doc("_testcapi.error2", doc2)
1198 self.assertEqual(error2.__doc__, doc2)
1199
1200 # test with explicit base (without docstring)
1201 error3 = _testcapi.make_exception_with_doc("_testcapi.error3",
1202 base=error2)
1203 self.assertTrue(issubclass(error3, error2))
1204
1205 # test with explicit base tuple
1206 class C(object):
1207 pass
1208 error4 = _testcapi.make_exception_with_doc("_testcapi.error4", doc4,
1209 (error3, C))
1210 self.assertTrue(issubclass(error4, error3))
1211 self.assertTrue(issubclass(error4, C))
1212 self.assertEqual(error4.__doc__, doc4)
1213
1214 # test with explicit dictionary
1215 error5 = _testcapi.make_exception_with_doc("_testcapi.error5", "",
1216 error4, {'a': 1})
1217 self.assertTrue(issubclass(error5, error4))
1218 self.assertEqual(error5.a, 1)
1219 self.assertEqual(error5.__doc__, "")
1220
Serhiy Storchaka5cfc79d2014-02-07 10:06:39 +02001221 @cpython_only
Antoine Pitrou07e20ef2010-10-28 22:56:58 +00001222 def test_memory_error_cleanup(self):
1223 # Issue #5437: preallocated MemoryError instances should not keep
1224 # traceback objects alive.
1225 from _testcapi import raise_memoryerror
1226 class C:
1227 pass
1228 wr = None
1229 def inner():
1230 nonlocal wr
1231 c = C()
1232 wr = weakref.ref(c)
1233 raise_memoryerror()
1234 # We cannot use assertRaises since it manually deletes the traceback
1235 try:
1236 inner()
1237 except MemoryError as e:
1238 self.assertNotEqual(wr(), None)
1239 else:
1240 self.fail("MemoryError not raised")
1241 self.assertEqual(wr(), None)
1242
Brett Cannon31f59292011-02-21 19:29:56 +00001243 @no_tracing
Antoine Pitrou07e20ef2010-10-28 22:56:58 +00001244 def test_recursion_error_cleanup(self):
1245 # Same test as above, but with "recursion exceeded" errors
1246 class C:
1247 pass
1248 wr = None
1249 def inner():
1250 nonlocal wr
1251 c = C()
1252 wr = weakref.ref(c)
1253 inner()
1254 # We cannot use assertRaises since it manually deletes the traceback
1255 try:
1256 inner()
Yury Selivanovf488fb42015-07-03 01:04:23 -04001257 except RecursionError as e:
Antoine Pitrou07e20ef2010-10-28 22:56:58 +00001258 self.assertNotEqual(wr(), None)
1259 else:
Yury Selivanovf488fb42015-07-03 01:04:23 -04001260 self.fail("RecursionError not raised")
Antoine Pitrou07e20ef2010-10-28 22:56:58 +00001261 self.assertEqual(wr(), None)
Georg Brandl1e28a272009-12-28 08:41:01 +00001262
Antoine Pitroua7622852011-09-01 21:37:43 +02001263 def test_errno_ENOTDIR(self):
1264 # Issue #12802: "not a directory" errors are ENOTDIR even on Windows
1265 with self.assertRaises(OSError) as cm:
1266 os.listdir(__file__)
1267 self.assertEqual(cm.exception.errno, errno.ENOTDIR, cm.exception)
1268
Martin Panter3263f682016-02-28 03:16:11 +00001269 def test_unraisable(self):
1270 # Issue #22836: PyErr_WriteUnraisable() should give sensible reports
1271 class BrokenDel:
1272 def __del__(self):
1273 exc = ValueError("del is broken")
1274 # The following line is included in the traceback report:
1275 raise exc
1276
Victor Stinnere4d300e2019-05-22 23:44:02 +02001277 obj = BrokenDel()
1278 with support.catch_unraisable_exception() as cm:
1279 del obj
Martin Panter3263f682016-02-28 03:16:11 +00001280
Victor Stinnere4d300e2019-05-22 23:44:02 +02001281 self.assertEqual(cm.unraisable.object, BrokenDel.__del__)
1282 self.assertIsNotNone(cm.unraisable.exc_traceback)
Martin Panter3263f682016-02-28 03:16:11 +00001283
1284 def test_unhandled(self):
1285 # Check for sensible reporting of unhandled exceptions
1286 for exc_type in (ValueError, BrokenStrException):
1287 with self.subTest(exc_type):
1288 try:
1289 exc = exc_type("test message")
1290 # The following line is included in the traceback report:
1291 raise exc
1292 except exc_type:
1293 with captured_stderr() as stderr:
1294 sys.__excepthook__(*sys.exc_info())
1295 report = stderr.getvalue()
1296 self.assertIn("test_exceptions.py", report)
1297 self.assertIn("raise exc", report)
1298 self.assertIn(exc_type.__name__, report)
1299 if exc_type is BrokenStrException:
1300 self.assertIn("<exception str() failed>", report)
1301 else:
1302 self.assertIn("test message", report)
1303 self.assertTrue(report.endswith("\n"))
1304
xdegaye66caacf2017-10-23 18:08:41 +02001305 @cpython_only
1306 def test_memory_error_in_PyErr_PrintEx(self):
1307 code = """if 1:
1308 import _testcapi
1309 class C(): pass
1310 _testcapi.set_nomemory(0, %d)
1311 C()
1312 """
1313
1314 # Issue #30817: Abort in PyErr_PrintEx() when no memory.
1315 # Span a large range of tests as the CPython code always evolves with
1316 # changes that add or remove memory allocations.
1317 for i in range(1, 20):
1318 rc, out, err = script_helper.assert_python_failure("-c", code % i)
1319 self.assertIn(rc, (1, 120))
1320 self.assertIn(b'MemoryError', err)
1321
Mark Shannonae3087c2017-10-22 22:41:51 +01001322 def test_yield_in_nested_try_excepts(self):
1323 #Issue #25612
1324 class MainError(Exception):
1325 pass
1326
1327 class SubError(Exception):
1328 pass
1329
1330 def main():
1331 try:
1332 raise MainError()
1333 except MainError:
1334 try:
1335 yield
1336 except SubError:
1337 pass
1338 raise
1339
1340 coro = main()
1341 coro.send(None)
1342 with self.assertRaises(MainError):
1343 coro.throw(SubError())
1344
1345 def test_generator_doesnt_retain_old_exc2(self):
1346 #Issue 28884#msg282532
1347 def g():
1348 try:
1349 raise ValueError
1350 except ValueError:
1351 yield 1
1352 self.assertEqual(sys.exc_info(), (None, None, None))
1353 yield 2
1354
1355 gen = g()
1356
1357 try:
1358 raise IndexError
1359 except IndexError:
1360 self.assertEqual(next(gen), 1)
1361 self.assertEqual(next(gen), 2)
1362
1363 def test_raise_in_generator(self):
1364 #Issue 25612#msg304117
1365 def g():
1366 yield 1
1367 raise
1368 yield 2
1369
1370 with self.assertRaises(ZeroDivisionError):
1371 i = g()
1372 try:
1373 1/0
1374 except:
1375 next(i)
1376 next(i)
1377
Zackery Spytzce6a0702019-08-25 03:44:09 -06001378 @unittest.skipUnless(__debug__, "Won't work if __debug__ is False")
1379 def test_assert_shadowing(self):
1380 # Shadowing AssertionError would cause the assert statement to
1381 # misbehave.
1382 global AssertionError
1383 AssertionError = TypeError
1384 try:
1385 assert False, 'hello'
1386 except BaseException as e:
1387 del AssertionError
1388 self.assertIsInstance(e, AssertionError)
1389 self.assertEqual(str(e), 'hello')
1390 else:
1391 del AssertionError
1392 self.fail('Expected exception')
1393
Pablo Galindo9b648a92020-09-01 19:39:46 +01001394 def test_memory_error_subclasses(self):
1395 # bpo-41654: MemoryError instances use a freelist of objects that are
1396 # linked using the 'dict' attribute when they are inactive/dead.
1397 # Subclasses of MemoryError should not participate in the freelist
1398 # schema. This test creates a MemoryError object and keeps it alive
1399 # (therefore advancing the freelist) and then it creates and destroys a
1400 # subclass object. Finally, it checks that creating a new MemoryError
1401 # succeeds, proving that the freelist is not corrupted.
1402
1403 class TestException(MemoryError):
1404 pass
1405
1406 try:
1407 raise MemoryError
1408 except MemoryError as exc:
1409 inst = exc
1410
1411 try:
1412 raise TestException
1413 except Exception:
1414 pass
1415
1416 for _ in range(10):
1417 try:
1418 raise MemoryError
1419 except MemoryError as exc:
1420 pass
1421
1422 gc_collect()
1423
Pablo Galindo5bf8bf22021-04-14 15:10:33 +01001424global_for_suggestions = None
1425
1426class NameErrorTests(unittest.TestCase):
1427 def test_name_error_has_name(self):
1428 try:
1429 bluch
1430 except NameError as exc:
1431 self.assertEqual("bluch", exc.name)
1432
1433 def test_name_error_suggestions(self):
1434 def Substitution():
1435 noise = more_noise = a = bc = None
1436 blech = None
1437 print(bluch)
1438
1439 def Elimination():
1440 noise = more_noise = a = bc = None
1441 blch = None
1442 print(bluch)
1443
1444 def Addition():
1445 noise = more_noise = a = bc = None
1446 bluchin = None
1447 print(bluch)
1448
1449 def SubstitutionOverElimination():
1450 blach = None
1451 bluc = None
1452 print(bluch)
1453
1454 def SubstitutionOverAddition():
1455 blach = None
1456 bluchi = None
1457 print(bluch)
1458
1459 def EliminationOverAddition():
1460 blucha = None
1461 bluc = None
1462 print(bluch)
1463
Pablo Galindo7a041162021-04-19 23:35:53 +01001464 for func, suggestion in [(Substitution, "'blech'?"),
1465 (Elimination, "'blch'?"),
1466 (Addition, "'bluchin'?"),
1467 (EliminationOverAddition, "'blucha'?"),
1468 (SubstitutionOverElimination, "'blach'?"),
1469 (SubstitutionOverAddition, "'blach'?")]:
Pablo Galindo5bf8bf22021-04-14 15:10:33 +01001470 err = None
1471 try:
1472 func()
1473 except NameError as exc:
1474 with support.captured_stderr() as err:
1475 sys.__excepthook__(*sys.exc_info())
1476 self.assertIn(suggestion, err.getvalue())
1477
1478 def test_name_error_suggestions_from_globals(self):
1479 def func():
1480 print(global_for_suggestio)
1481 try:
1482 func()
1483 except NameError as exc:
1484 with support.captured_stderr() as err:
1485 sys.__excepthook__(*sys.exc_info())
Pablo Galindo7a041162021-04-19 23:35:53 +01001486 self.assertIn("'global_for_suggestions'?", err.getvalue())
Pablo Galindo5bf8bf22021-04-14 15:10:33 +01001487
Pablo Galindo3ab4bea2021-04-17 22:26:54 +01001488 def test_name_error_suggestions_from_builtins(self):
1489 def func():
Dennis Sweeney80a2a4e2021-05-03 11:47:27 -04001490 print(ZeroDivisionErrrrr)
Pablo Galindo3ab4bea2021-04-17 22:26:54 +01001491 try:
1492 func()
1493 except NameError as exc:
1494 with support.captured_stderr() as err:
1495 sys.__excepthook__(*sys.exc_info())
Dennis Sweeney80a2a4e2021-05-03 11:47:27 -04001496 self.assertIn("'ZeroDivisionError'?", err.getvalue())
Pablo Galindo3ab4bea2021-04-17 22:26:54 +01001497
Pablo Galindo5bf8bf22021-04-14 15:10:33 +01001498 def test_name_error_suggestions_do_not_trigger_for_long_names(self):
1499 def f():
1500 somethingverywronghehehehehehe = None
1501 print(somethingverywronghe)
1502
1503 try:
1504 f()
1505 except NameError as exc:
1506 with support.captured_stderr() as err:
1507 sys.__excepthook__(*sys.exc_info())
1508
1509 self.assertNotIn("somethingverywronghehe", err.getvalue())
1510
Dennis Sweeney284c52d2021-04-26 20:22:27 -04001511 def test_name_error_bad_suggestions_do_not_trigger_for_small_names(self):
1512 vvv = mom = w = id = pytho = None
1513
1514 with self.subTest(name="b"):
1515 try:
1516 b
1517 except NameError as exc:
1518 with support.captured_stderr() as err:
1519 sys.__excepthook__(*sys.exc_info())
1520 self.assertNotIn("you mean", err.getvalue())
1521 self.assertNotIn("vvv", err.getvalue())
1522 self.assertNotIn("mom", err.getvalue())
1523 self.assertNotIn("'id'", err.getvalue())
1524 self.assertNotIn("'w'", err.getvalue())
1525 self.assertNotIn("'pytho'", err.getvalue())
1526
1527 with self.subTest(name="v"):
1528 try:
1529 v
1530 except NameError as exc:
1531 with support.captured_stderr() as err:
1532 sys.__excepthook__(*sys.exc_info())
1533 self.assertNotIn("you mean", err.getvalue())
1534 self.assertNotIn("vvv", err.getvalue())
1535 self.assertNotIn("mom", err.getvalue())
1536 self.assertNotIn("'id'", err.getvalue())
1537 self.assertNotIn("'w'", err.getvalue())
1538 self.assertNotIn("'pytho'", err.getvalue())
1539
1540 with self.subTest(name="m"):
1541 try:
1542 m
1543 except NameError as exc:
1544 with support.captured_stderr() as err:
1545 sys.__excepthook__(*sys.exc_info())
1546 self.assertNotIn("you mean", err.getvalue())
1547 self.assertNotIn("vvv", err.getvalue())
1548 self.assertNotIn("mom", err.getvalue())
1549 self.assertNotIn("'id'", err.getvalue())
1550 self.assertNotIn("'w'", err.getvalue())
1551 self.assertNotIn("'pytho'", err.getvalue())
1552
1553 with self.subTest(name="py"):
1554 try:
1555 py
1556 except NameError as exc:
1557 with support.captured_stderr() as err:
1558 sys.__excepthook__(*sys.exc_info())
1559 self.assertNotIn("you mean", err.getvalue())
1560 self.assertNotIn("vvv", err.getvalue())
1561 self.assertNotIn("mom", err.getvalue())
1562 self.assertNotIn("'id'", err.getvalue())
1563 self.assertNotIn("'w'", err.getvalue())
1564 self.assertNotIn("'pytho'", err.getvalue())
1565
Pablo Galindo3ab4bea2021-04-17 22:26:54 +01001566 def test_name_error_suggestions_do_not_trigger_for_too_many_locals(self):
Pablo Galindo5bf8bf22021-04-14 15:10:33 +01001567 def f():
1568 # Mutating locals() is unreliable, so we need to do it by hand
Dennis Sweeney80a2a4e2021-05-03 11:47:27 -04001569 a1 = a2 = a3 = a4 = a5 = a6 = a7 = a8 = a9 = a10 = \
1570 a11 = a12 = a13 = a14 = a15 = a16 = a17 = a18 = a19 = a20 = \
1571 a21 = a22 = a23 = a24 = a25 = a26 = a27 = a28 = a29 = a30 = \
1572 a31 = a32 = a33 = a34 = a35 = a36 = a37 = a38 = a39 = a40 = \
1573 a41 = a42 = a43 = a44 = a45 = a46 = a47 = a48 = a49 = a50 = \
1574 a51 = a52 = a53 = a54 = a55 = a56 = a57 = a58 = a59 = a60 = \
1575 a61 = a62 = a63 = a64 = a65 = a66 = a67 = a68 = a69 = a70 = \
1576 a71 = a72 = a73 = a74 = a75 = a76 = a77 = a78 = a79 = a80 = \
1577 a81 = a82 = a83 = a84 = a85 = a86 = a87 = a88 = a89 = a90 = \
1578 a91 = a92 = a93 = a94 = a95 = a96 = a97 = a98 = a99 = a100 = \
1579 a101 = a102 = a103 = a104 = a105 = a106 = a107 = a108 = a109 = a110 = \
1580 a111 = a112 = a113 = a114 = a115 = a116 = a117 = a118 = a119 = a120 = \
1581 a121 = a122 = a123 = a124 = a125 = a126 = a127 = a128 = a129 = a130 = \
1582 a131 = a132 = a133 = a134 = a135 = a136 = a137 = a138 = a139 = a140 = \
1583 a141 = a142 = a143 = a144 = a145 = a146 = a147 = a148 = a149 = a150 = \
1584 a151 = a152 = a153 = a154 = a155 = a156 = a157 = a158 = a159 = a160 = \
1585 a161 = a162 = a163 = a164 = a165 = a166 = a167 = a168 = a169 = a170 = \
1586 a171 = a172 = a173 = a174 = a175 = a176 = a177 = a178 = a179 = a180 = \
1587 a181 = a182 = a183 = a184 = a185 = a186 = a187 = a188 = a189 = a190 = \
1588 a191 = a192 = a193 = a194 = a195 = a196 = a197 = a198 = a199 = a200 = \
1589 a201 = a202 = a203 = a204 = a205 = a206 = a207 = a208 = a209 = a210 = \
1590 a211 = a212 = a213 = a214 = a215 = a216 = a217 = a218 = a219 = a220 = \
1591 a221 = a222 = a223 = a224 = a225 = a226 = a227 = a228 = a229 = a230 = \
1592 a231 = a232 = a233 = a234 = a235 = a236 = a237 = a238 = a239 = a240 = \
1593 a241 = a242 = a243 = a244 = a245 = a246 = a247 = a248 = a249 = a250 = \
1594 a251 = a252 = a253 = a254 = a255 = a256 = a257 = a258 = a259 = a260 = \
1595 a261 = a262 = a263 = a264 = a265 = a266 = a267 = a268 = a269 = a270 = \
1596 a271 = a272 = a273 = a274 = a275 = a276 = a277 = a278 = a279 = a280 = \
1597 a281 = a282 = a283 = a284 = a285 = a286 = a287 = a288 = a289 = a290 = \
1598 a291 = a292 = a293 = a294 = a295 = a296 = a297 = a298 = a299 = a300 = \
1599 a301 = a302 = a303 = a304 = a305 = a306 = a307 = a308 = a309 = a310 = \
1600 a311 = a312 = a313 = a314 = a315 = a316 = a317 = a318 = a319 = a320 = \
1601 a321 = a322 = a323 = a324 = a325 = a326 = a327 = a328 = a329 = a330 = \
1602 a331 = a332 = a333 = a334 = a335 = a336 = a337 = a338 = a339 = a340 = \
1603 a341 = a342 = a343 = a344 = a345 = a346 = a347 = a348 = a349 = a350 = \
1604 a351 = a352 = a353 = a354 = a355 = a356 = a357 = a358 = a359 = a360 = \
1605 a361 = a362 = a363 = a364 = a365 = a366 = a367 = a368 = a369 = a370 = \
1606 a371 = a372 = a373 = a374 = a375 = a376 = a377 = a378 = a379 = a380 = \
1607 a381 = a382 = a383 = a384 = a385 = a386 = a387 = a388 = a389 = a390 = \
1608 a391 = a392 = a393 = a394 = a395 = a396 = a397 = a398 = a399 = a400 = \
1609 a401 = a402 = a403 = a404 = a405 = a406 = a407 = a408 = a409 = a410 = \
1610 a411 = a412 = a413 = a414 = a415 = a416 = a417 = a418 = a419 = a420 = \
1611 a421 = a422 = a423 = a424 = a425 = a426 = a427 = a428 = a429 = a430 = \
1612 a431 = a432 = a433 = a434 = a435 = a436 = a437 = a438 = a439 = a440 = \
1613 a441 = a442 = a443 = a444 = a445 = a446 = a447 = a448 = a449 = a450 = \
1614 a451 = a452 = a453 = a454 = a455 = a456 = a457 = a458 = a459 = a460 = \
1615 a461 = a462 = a463 = a464 = a465 = a466 = a467 = a468 = a469 = a470 = \
1616 a471 = a472 = a473 = a474 = a475 = a476 = a477 = a478 = a479 = a480 = \
1617 a481 = a482 = a483 = a484 = a485 = a486 = a487 = a488 = a489 = a490 = \
1618 a491 = a492 = a493 = a494 = a495 = a496 = a497 = a498 = a499 = a500 = \
1619 a501 = a502 = a503 = a504 = a505 = a506 = a507 = a508 = a509 = a510 = \
1620 a511 = a512 = a513 = a514 = a515 = a516 = a517 = a518 = a519 = a520 = \
1621 a521 = a522 = a523 = a524 = a525 = a526 = a527 = a528 = a529 = a530 = \
1622 a531 = a532 = a533 = a534 = a535 = a536 = a537 = a538 = a539 = a540 = \
1623 a541 = a542 = a543 = a544 = a545 = a546 = a547 = a548 = a549 = a550 = \
1624 a551 = a552 = a553 = a554 = a555 = a556 = a557 = a558 = a559 = a560 = \
1625 a561 = a562 = a563 = a564 = a565 = a566 = a567 = a568 = a569 = a570 = \
1626 a571 = a572 = a573 = a574 = a575 = a576 = a577 = a578 = a579 = a580 = \
1627 a581 = a582 = a583 = a584 = a585 = a586 = a587 = a588 = a589 = a590 = \
1628 a591 = a592 = a593 = a594 = a595 = a596 = a597 = a598 = a599 = a600 = \
1629 a601 = a602 = a603 = a604 = a605 = a606 = a607 = a608 = a609 = a610 = \
1630 a611 = a612 = a613 = a614 = a615 = a616 = a617 = a618 = a619 = a620 = \
1631 a621 = a622 = a623 = a624 = a625 = a626 = a627 = a628 = a629 = a630 = \
1632 a631 = a632 = a633 = a634 = a635 = a636 = a637 = a638 = a639 = a640 = \
1633 a641 = a642 = a643 = a644 = a645 = a646 = a647 = a648 = a649 = a650 = \
1634 a651 = a652 = a653 = a654 = a655 = a656 = a657 = a658 = a659 = a660 = \
1635 a661 = a662 = a663 = a664 = a665 = a666 = a667 = a668 = a669 = a670 = \
1636 a671 = a672 = a673 = a674 = a675 = a676 = a677 = a678 = a679 = a680 = \
1637 a681 = a682 = a683 = a684 = a685 = a686 = a687 = a688 = a689 = a690 = \
1638 a691 = a692 = a693 = a694 = a695 = a696 = a697 = a698 = a699 = a700 = \
1639 a701 = a702 = a703 = a704 = a705 = a706 = a707 = a708 = a709 = a710 = \
1640 a711 = a712 = a713 = a714 = a715 = a716 = a717 = a718 = a719 = a720 = \
1641 a721 = a722 = a723 = a724 = a725 = a726 = a727 = a728 = a729 = a730 = \
1642 a731 = a732 = a733 = a734 = a735 = a736 = a737 = a738 = a739 = a740 = \
1643 a741 = a742 = a743 = a744 = a745 = a746 = a747 = a748 = a749 = a750 = \
1644 a751 = a752 = a753 = a754 = a755 = a756 = a757 = a758 = a759 = a760 = \
1645 a761 = a762 = a763 = a764 = a765 = a766 = a767 = a768 = a769 = a770 = \
1646 a771 = a772 = a773 = a774 = a775 = a776 = a777 = a778 = a779 = a780 = \
1647 a781 = a782 = a783 = a784 = a785 = a786 = a787 = a788 = a789 = a790 = \
1648 a791 = a792 = a793 = a794 = a795 = a796 = a797 = a798 = a799 = a800 \
1649 = None
Pablo Galindo5bf8bf22021-04-14 15:10:33 +01001650 print(a0)
1651
1652 try:
1653 f()
1654 except NameError as exc:
1655 with support.captured_stderr() as err:
1656 sys.__excepthook__(*sys.exc_info())
1657
Pablo Galindo3ab4bea2021-04-17 22:26:54 +01001658 self.assertNotIn("a1", err.getvalue())
Pablo Galindo5bf8bf22021-04-14 15:10:33 +01001659
1660 def test_name_error_with_custom_exceptions(self):
1661 def f():
1662 blech = None
1663 raise NameError()
1664
1665 try:
1666 f()
1667 except NameError as exc:
1668 with support.captured_stderr() as err:
1669 sys.__excepthook__(*sys.exc_info())
1670
1671 self.assertNotIn("blech", err.getvalue())
1672
1673 def f():
1674 blech = None
1675 raise NameError
1676
1677 try:
1678 f()
1679 except NameError as exc:
1680 with support.captured_stderr() as err:
1681 sys.__excepthook__(*sys.exc_info())
1682
1683 self.assertNotIn("blech", err.getvalue())
Antoine Pitroua7622852011-09-01 21:37:43 +02001684
Pablo Galindo0ad81d42021-04-16 17:12:03 +01001685 def test_unbound_local_error_doesn_not_match(self):
1686 def foo():
1687 something = 3
1688 print(somethong)
1689 somethong = 3
1690
1691 try:
1692 foo()
1693 except UnboundLocalError as exc:
1694 with support.captured_stderr() as err:
1695 sys.__excepthook__(*sys.exc_info())
1696
1697 self.assertNotIn("something", err.getvalue())
1698
1699
Pablo Galindo37494b42021-04-14 02:36:07 +01001700class AttributeErrorTests(unittest.TestCase):
1701 def test_attributes(self):
1702 # Setting 'attr' should not be a problem.
1703 exc = AttributeError('Ouch!')
1704 self.assertIsNone(exc.name)
1705 self.assertIsNone(exc.obj)
1706
1707 sentinel = object()
1708 exc = AttributeError('Ouch', name='carry', obj=sentinel)
1709 self.assertEqual(exc.name, 'carry')
1710 self.assertIs(exc.obj, sentinel)
1711
1712 def test_getattr_has_name_and_obj(self):
1713 class A:
1714 blech = None
1715
1716 obj = A()
1717 try:
1718 obj.bluch
1719 except AttributeError as exc:
1720 self.assertEqual("bluch", exc.name)
1721 self.assertEqual(obj, exc.obj)
1722
1723 def test_getattr_has_name_and_obj_for_method(self):
1724 class A:
1725 def blech(self):
1726 return
1727
1728 obj = A()
1729 try:
1730 obj.bluch()
1731 except AttributeError as exc:
1732 self.assertEqual("bluch", exc.name)
1733 self.assertEqual(obj, exc.obj)
1734
1735 def test_getattr_suggestions(self):
1736 class Substitution:
1737 noise = more_noise = a = bc = None
1738 blech = None
1739
1740 class Elimination:
1741 noise = more_noise = a = bc = None
1742 blch = None
1743
1744 class Addition:
1745 noise = more_noise = a = bc = None
1746 bluchin = None
1747
1748 class SubstitutionOverElimination:
1749 blach = None
1750 bluc = None
1751
1752 class SubstitutionOverAddition:
1753 blach = None
1754 bluchi = None
1755
1756 class EliminationOverAddition:
1757 blucha = None
1758 bluc = None
1759
Pablo Galindo7a041162021-04-19 23:35:53 +01001760 for cls, suggestion in [(Substitution, "'blech'?"),
1761 (Elimination, "'blch'?"),
1762 (Addition, "'bluchin'?"),
1763 (EliminationOverAddition, "'bluc'?"),
1764 (SubstitutionOverElimination, "'blach'?"),
1765 (SubstitutionOverAddition, "'blach'?")]:
Pablo Galindo37494b42021-04-14 02:36:07 +01001766 try:
1767 cls().bluch
1768 except AttributeError as exc:
1769 with support.captured_stderr() as err:
1770 sys.__excepthook__(*sys.exc_info())
1771
1772 self.assertIn(suggestion, err.getvalue())
1773
1774 def test_getattr_suggestions_do_not_trigger_for_long_attributes(self):
1775 class A:
1776 blech = None
1777
1778 try:
1779 A().somethingverywrong
1780 except AttributeError as exc:
1781 with support.captured_stderr() as err:
1782 sys.__excepthook__(*sys.exc_info())
1783
1784 self.assertNotIn("blech", err.getvalue())
1785
Dennis Sweeney284c52d2021-04-26 20:22:27 -04001786 def test_getattr_error_bad_suggestions_do_not_trigger_for_small_names(self):
1787 class MyClass:
1788 vvv = mom = w = id = pytho = None
1789
1790 with self.subTest(name="b"):
1791 try:
1792 MyClass.b
1793 except AttributeError as exc:
1794 with support.captured_stderr() as err:
1795 sys.__excepthook__(*sys.exc_info())
1796 self.assertNotIn("you mean", err.getvalue())
1797 self.assertNotIn("vvv", err.getvalue())
1798 self.assertNotIn("mom", err.getvalue())
1799 self.assertNotIn("'id'", err.getvalue())
1800 self.assertNotIn("'w'", err.getvalue())
1801 self.assertNotIn("'pytho'", err.getvalue())
1802
1803 with self.subTest(name="v"):
1804 try:
1805 MyClass.v
1806 except AttributeError as exc:
1807 with support.captured_stderr() as err:
1808 sys.__excepthook__(*sys.exc_info())
1809 self.assertNotIn("you mean", err.getvalue())
1810 self.assertNotIn("vvv", err.getvalue())
1811 self.assertNotIn("mom", err.getvalue())
1812 self.assertNotIn("'id'", err.getvalue())
1813 self.assertNotIn("'w'", err.getvalue())
1814 self.assertNotIn("'pytho'", err.getvalue())
1815
1816 with self.subTest(name="m"):
1817 try:
1818 MyClass.m
1819 except AttributeError as exc:
1820 with support.captured_stderr() as err:
1821 sys.__excepthook__(*sys.exc_info())
1822 self.assertNotIn("you mean", err.getvalue())
1823 self.assertNotIn("vvv", err.getvalue())
1824 self.assertNotIn("mom", err.getvalue())
1825 self.assertNotIn("'id'", err.getvalue())
1826 self.assertNotIn("'w'", err.getvalue())
1827 self.assertNotIn("'pytho'", err.getvalue())
1828
1829 with self.subTest(name="py"):
1830 try:
1831 MyClass.py
1832 except AttributeError as exc:
1833 with support.captured_stderr() as err:
1834 sys.__excepthook__(*sys.exc_info())
1835 self.assertNotIn("you mean", err.getvalue())
1836 self.assertNotIn("vvv", err.getvalue())
1837 self.assertNotIn("mom", err.getvalue())
1838 self.assertNotIn("'id'", err.getvalue())
1839 self.assertNotIn("'w'", err.getvalue())
1840 self.assertNotIn("'pytho'", err.getvalue())
1841
1842
Pablo Galindo37494b42021-04-14 02:36:07 +01001843 def test_getattr_suggestions_do_not_trigger_for_big_dicts(self):
1844 class A:
1845 blech = None
1846 # A class with a very big __dict__ will not be consider
1847 # for suggestions.
Dennis Sweeney80a2a4e2021-05-03 11:47:27 -04001848 for index in range(2000):
Pablo Galindo37494b42021-04-14 02:36:07 +01001849 setattr(A, f"index_{index}", None)
1850
1851 try:
1852 A().bluch
1853 except AttributeError as exc:
1854 with support.captured_stderr() as err:
1855 sys.__excepthook__(*sys.exc_info())
1856
1857 self.assertNotIn("blech", err.getvalue())
1858
1859 def test_getattr_suggestions_no_args(self):
1860 class A:
1861 blech = None
1862 def __getattr__(self, attr):
1863 raise AttributeError()
1864
1865 try:
1866 A().bluch
1867 except AttributeError as exc:
1868 with support.captured_stderr() as err:
1869 sys.__excepthook__(*sys.exc_info())
1870
1871 self.assertIn("blech", err.getvalue())
1872
1873 class A:
1874 blech = None
1875 def __getattr__(self, attr):
1876 raise AttributeError
1877
1878 try:
1879 A().bluch
1880 except AttributeError as exc:
1881 with support.captured_stderr() as err:
1882 sys.__excepthook__(*sys.exc_info())
1883
1884 self.assertIn("blech", err.getvalue())
1885
1886 def test_getattr_suggestions_invalid_args(self):
1887 class NonStringifyClass:
1888 __str__ = None
1889 __repr__ = None
1890
1891 class A:
1892 blech = None
1893 def __getattr__(self, attr):
1894 raise AttributeError(NonStringifyClass())
1895
1896 class B:
1897 blech = None
1898 def __getattr__(self, attr):
1899 raise AttributeError("Error", 23)
1900
1901 class C:
1902 blech = None
1903 def __getattr__(self, attr):
1904 raise AttributeError(23)
1905
1906 for cls in [A, B, C]:
1907 try:
1908 cls().bluch
1909 except AttributeError as exc:
1910 with support.captured_stderr() as err:
1911 sys.__excepthook__(*sys.exc_info())
1912
1913 self.assertIn("blech", err.getvalue())
1914
Pablo Galindoe07f4ab2021-04-14 18:58:28 +01001915 def test_attribute_error_with_failing_dict(self):
1916 class T:
1917 bluch = 1
1918 def __dir__(self):
1919 raise AttributeError("oh no!")
1920
1921 try:
1922 T().blich
1923 except AttributeError as exc:
1924 with support.captured_stderr() as err:
1925 sys.__excepthook__(*sys.exc_info())
1926
1927 self.assertNotIn("blech", err.getvalue())
1928 self.assertNotIn("oh no!", err.getvalue())
Pablo Galindo37494b42021-04-14 02:36:07 +01001929
Pablo Galindo0b1c1692021-04-17 23:28:45 +01001930 def test_attribute_error_with_bad_name(self):
1931 try:
1932 raise AttributeError(name=12, obj=23)
1933 except AttributeError as exc:
1934 with support.captured_stderr() as err:
1935 sys.__excepthook__(*sys.exc_info())
1936
1937 self.assertNotIn("?", err.getvalue())
1938
1939
Brett Cannon79ec55e2012-04-12 20:24:54 -04001940class ImportErrorTests(unittest.TestCase):
1941
1942 def test_attributes(self):
1943 # Setting 'name' and 'path' should not be a problem.
1944 exc = ImportError('test')
1945 self.assertIsNone(exc.name)
1946 self.assertIsNone(exc.path)
1947
1948 exc = ImportError('test', name='somemodule')
1949 self.assertEqual(exc.name, 'somemodule')
1950 self.assertIsNone(exc.path)
1951
1952 exc = ImportError('test', path='somepath')
1953 self.assertEqual(exc.path, 'somepath')
1954 self.assertIsNone(exc.name)
1955
1956 exc = ImportError('test', path='somepath', name='somename')
1957 self.assertEqual(exc.name, 'somename')
1958 self.assertEqual(exc.path, 'somepath')
1959
Michael Seifert64c8f702017-04-09 09:47:12 +02001960 msg = "'invalid' is an invalid keyword argument for ImportError"
Serhiy Storchaka47dee112016-09-27 20:45:35 +03001961 with self.assertRaisesRegex(TypeError, msg):
1962 ImportError('test', invalid='keyword')
1963
1964 with self.assertRaisesRegex(TypeError, msg):
1965 ImportError('test', name='name', invalid='keyword')
1966
1967 with self.assertRaisesRegex(TypeError, msg):
1968 ImportError('test', path='path', invalid='keyword')
1969
1970 with self.assertRaisesRegex(TypeError, msg):
1971 ImportError(invalid='keyword')
1972
Serhiy Storchaka47dee112016-09-27 20:45:35 +03001973 with self.assertRaisesRegex(TypeError, msg):
1974 ImportError('test', invalid='keyword', another=True)
1975
Serhiy Storchakae9e44482016-09-28 07:53:32 +03001976 def test_reset_attributes(self):
1977 exc = ImportError('test', name='name', path='path')
1978 self.assertEqual(exc.args, ('test',))
1979 self.assertEqual(exc.msg, 'test')
1980 self.assertEqual(exc.name, 'name')
1981 self.assertEqual(exc.path, 'path')
1982
1983 # Reset not specified attributes
1984 exc.__init__()
1985 self.assertEqual(exc.args, ())
1986 self.assertEqual(exc.msg, None)
1987 self.assertEqual(exc.name, None)
1988 self.assertEqual(exc.path, None)
1989
Brett Cannon07c6e712012-08-24 13:05:09 -04001990 def test_non_str_argument(self):
1991 # Issue #15778
Nadeem Vawda6d708702012-10-14 01:42:32 +02001992 with check_warnings(('', BytesWarning), quiet=True):
1993 arg = b'abc'
1994 exc = ImportError(arg)
1995 self.assertEqual(str(arg), str(exc))
Brett Cannon79ec55e2012-04-12 20:24:54 -04001996
Serhiy Storchakab7853962017-04-08 09:55:07 +03001997 def test_copy_pickle(self):
1998 for kwargs in (dict(),
1999 dict(name='somename'),
2000 dict(path='somepath'),
2001 dict(name='somename', path='somepath')):
2002 orig = ImportError('test', **kwargs)
2003 for proto in range(pickle.HIGHEST_PROTOCOL + 1):
2004 exc = pickle.loads(pickle.dumps(orig, proto))
2005 self.assertEqual(exc.args, ('test',))
2006 self.assertEqual(exc.msg, 'test')
2007 self.assertEqual(exc.name, orig.name)
2008 self.assertEqual(exc.path, orig.path)
2009 for c in copy.copy, copy.deepcopy:
2010 exc = c(orig)
2011 self.assertEqual(exc.args, ('test',))
2012 self.assertEqual(exc.msg, 'test')
2013 self.assertEqual(exc.name, orig.name)
2014 self.assertEqual(exc.path, orig.path)
2015
Pablo Galindoa77aac42021-04-23 14:27:05 +01002016class SyntaxErrorTests(unittest.TestCase):
2017 def test_range_of_offsets(self):
2018 cases = [
2019 # Basic range from 2->7
2020 (("bad.py", 1, 2, "abcdefg", 1, 7),
2021 dedent(
2022 """
2023 File "bad.py", line 1
2024 abcdefg
2025 ^^^^^
2026 SyntaxError: bad bad
2027 """)),
2028 # end_offset = start_offset + 1
2029 (("bad.py", 1, 2, "abcdefg", 1, 3),
2030 dedent(
2031 """
2032 File "bad.py", line 1
2033 abcdefg
2034 ^
2035 SyntaxError: bad bad
2036 """)),
2037 # Negative end offset
2038 (("bad.py", 1, 2, "abcdefg", 1, -2),
2039 dedent(
2040 """
2041 File "bad.py", line 1
2042 abcdefg
2043 ^
2044 SyntaxError: bad bad
2045 """)),
2046 # end offset before starting offset
2047 (("bad.py", 1, 4, "abcdefg", 1, 2),
2048 dedent(
2049 """
2050 File "bad.py", line 1
2051 abcdefg
2052 ^
2053 SyntaxError: bad bad
2054 """)),
2055 # Both offsets negative
2056 (("bad.py", 1, -4, "abcdefg", 1, -2),
2057 dedent(
2058 """
2059 File "bad.py", line 1
2060 abcdefg
2061 SyntaxError: bad bad
2062 """)),
2063 # Both offsets negative and the end more negative
2064 (("bad.py", 1, -4, "abcdefg", 1, -5),
2065 dedent(
2066 """
2067 File "bad.py", line 1
2068 abcdefg
2069 SyntaxError: bad bad
2070 """)),
2071 # Both offsets 0
2072 (("bad.py", 1, 0, "abcdefg", 1, 0),
2073 dedent(
2074 """
2075 File "bad.py", line 1
2076 abcdefg
2077 SyntaxError: bad bad
2078 """)),
2079 # Start offset 0 and end offset not 0
2080 (("bad.py", 1, 0, "abcdefg", 1, 5),
2081 dedent(
2082 """
2083 File "bad.py", line 1
2084 abcdefg
2085 SyntaxError: bad bad
2086 """)),
2087 # End offset pass the source lenght
2088 (("bad.py", 1, 2, "abcdefg", 1, 100),
2089 dedent(
2090 """
2091 File "bad.py", line 1
2092 abcdefg
2093 ^^^^^^
2094 SyntaxError: bad bad
2095 """)),
2096 ]
2097 for args, expected in cases:
2098 with self.subTest(args=args):
2099 try:
2100 raise SyntaxError("bad bad", args)
2101 except SyntaxError as exc:
2102 with support.captured_stderr() as err:
2103 sys.__excepthook__(*sys.exc_info())
2104 the_exception = exc
2105
2106 def test_attributes_new_constructor(self):
2107 args = ("bad.py", 1, 2, "abcdefg", 1, 100)
2108 the_exception = SyntaxError("bad bad", args)
2109 filename, lineno, offset, error, end_lineno, end_offset = args
2110 self.assertEqual(filename, the_exception.filename)
2111 self.assertEqual(lineno, the_exception.lineno)
2112 self.assertEqual(end_lineno, the_exception.end_lineno)
2113 self.assertEqual(offset, the_exception.offset)
2114 self.assertEqual(end_offset, the_exception.end_offset)
2115 self.assertEqual(error, the_exception.text)
2116 self.assertEqual("bad bad", the_exception.msg)
2117
2118 def test_attributes_old_constructor(self):
2119 args = ("bad.py", 1, 2, "abcdefg")
2120 the_exception = SyntaxError("bad bad", args)
2121 filename, lineno, offset, error = args
2122 self.assertEqual(filename, the_exception.filename)
2123 self.assertEqual(lineno, the_exception.lineno)
2124 self.assertEqual(None, the_exception.end_lineno)
2125 self.assertEqual(offset, the_exception.offset)
2126 self.assertEqual(None, the_exception.end_offset)
2127 self.assertEqual(error, the_exception.text)
2128 self.assertEqual("bad bad", the_exception.msg)
2129
2130 def test_incorrect_constructor(self):
2131 args = ("bad.py", 1, 2)
2132 self.assertRaises(TypeError, SyntaxError, "bad bad", args)
2133
2134 args = ("bad.py", 1, 2, 4, 5, 6, 7)
2135 self.assertRaises(TypeError, SyntaxError, "bad bad", args)
2136
2137 args = ("bad.py", 1, 2, "abcdefg", 1)
2138 self.assertRaises(TypeError, SyntaxError, "bad bad", args)
2139
Brett Cannon79ec55e2012-04-12 20:24:54 -04002140
Mark Shannonbf353f32020-12-17 13:55:28 +00002141class PEP626Tests(unittest.TestCase):
2142
2143 def lineno_after_raise(self, f, line):
2144 try:
2145 f()
2146 except Exception as ex:
2147 t = ex.__traceback__
2148 while t.tb_next:
2149 t = t.tb_next
2150 frame = t.tb_frame
Mark Shannon088a15c2021-04-29 19:28:50 +01002151 if line is None:
2152 self.assertEqual(frame.f_lineno, line)
2153 else:
2154 self.assertEqual(frame.f_lineno-frame.f_code.co_firstlineno, line)
Mark Shannonbf353f32020-12-17 13:55:28 +00002155
2156 def test_lineno_after_raise_simple(self):
2157 def simple():
2158 1/0
2159 pass
2160 self.lineno_after_raise(simple, 1)
2161
2162 def test_lineno_after_raise_in_except(self):
2163 def in_except():
2164 try:
2165 1/0
2166 except:
2167 1/0
2168 pass
2169 self.lineno_after_raise(in_except, 4)
2170
2171 def test_lineno_after_other_except(self):
2172 def other_except():
2173 try:
2174 1/0
2175 except TypeError as ex:
2176 pass
2177 self.lineno_after_raise(other_except, 3)
2178
2179 def test_lineno_in_named_except(self):
2180 def in_named_except():
2181 try:
2182 1/0
2183 except Exception as ex:
2184 1/0
2185 pass
2186 self.lineno_after_raise(in_named_except, 4)
2187
2188 def test_lineno_in_try(self):
2189 def in_try():
2190 try:
2191 1/0
2192 finally:
2193 pass
2194 self.lineno_after_raise(in_try, 4)
2195
2196 def test_lineno_in_finally_normal(self):
2197 def in_finally_normal():
2198 try:
2199 pass
2200 finally:
2201 1/0
2202 pass
2203 self.lineno_after_raise(in_finally_normal, 4)
2204
2205 def test_lineno_in_finally_except(self):
2206 def in_finally_except():
2207 try:
2208 1/0
2209 finally:
2210 1/0
2211 pass
2212 self.lineno_after_raise(in_finally_except, 4)
2213
2214 def test_lineno_after_with(self):
2215 class Noop:
2216 def __enter__(self):
2217 return self
2218 def __exit__(self, *args):
2219 pass
2220 def after_with():
2221 with Noop():
2222 1/0
2223 pass
2224 self.lineno_after_raise(after_with, 2)
2225
Mark Shannon088a15c2021-04-29 19:28:50 +01002226 def test_missing_lineno_shows_as_none(self):
2227 def f():
2228 1/0
2229 self.lineno_after_raise(f, 1)
2230 f.__code__ = f.__code__.replace(co_linetable=b'\x04\x80\xff\x80')
2231 self.lineno_after_raise(f, None)
Mark Shannonbf353f32020-12-17 13:55:28 +00002232
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00002233if __name__ == '__main__':
Guido van Rossumb8142c32007-05-08 17:49:10 +00002234 unittest.main()