blob: 1d897f29583b6f6d2780503ab1bc25068b5e1e1f [file] [log] [blame]
Guido van Rossum3bead091992-01-27 17:00:37 +00001# Python test set -- part 5, built-in exceptions
2
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00003import os
4import sys
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00005import unittest
Guido van Rossumbf12cdb2006-08-17 20:24:18 +00006import pickle
Barry Warsaw8d109cb2008-05-08 04:26:35 +00007import weakref
Antoine Pitroua7622852011-09-01 21:37:43 +02008import errno
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00009
Benjamin Petersonae5f2f42010-03-07 17:10:51 +000010from test.support import (TESTFN, unlink, run_unittest, captured_output,
Brett Cannon31f59292011-02-21 19:29:56 +000011 gc_collect, cpython_only, no_tracing)
Guido van Rossum83b120d2001-08-23 03:23:03 +000012
Richard Oudkerk5562d9d2012-07-28 17:45:28 +010013class NaiveException(Exception):
14 def __init__(self, x):
15 self.x = x
16
17class SlottedNaiveException(Exception):
18 __slots__ = ('x',)
19 def __init__(self, x):
20 self.x = x
21
Guido van Rossum3bead091992-01-27 17:00:37 +000022# XXX This is not really enough, each *operation* should be tested!
23
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000024class ExceptionTests(unittest.TestCase):
Barry Warsawb9c1d3d2001-08-13 23:07:00 +000025
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000026 def raise_catch(self, exc, excname):
27 try:
Collin Winter828f04a2007-08-31 00:04:24 +000028 raise exc("spam")
Guido van Rossumb940e112007-01-10 16:19:56 +000029 except exc as err:
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000030 buf1 = str(err)
31 try:
32 raise exc("spam")
Guido van Rossumb940e112007-01-10 16:19:56 +000033 except exc as err:
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000034 buf2 = str(err)
Ezio Melottib3aedd42010-11-20 19:04:17 +000035 self.assertEqual(buf1, buf2)
36 self.assertEqual(exc.__name__, excname)
Guido van Rossum3bead091992-01-27 17:00:37 +000037
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000038 def testRaising(self):
39 self.raise_catch(AttributeError, "AttributeError")
40 self.assertRaises(AttributeError, getattr, sys, "undefined_attribute")
Guido van Rossum3bead091992-01-27 17:00:37 +000041
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000042 self.raise_catch(EOFError, "EOFError")
43 fp = open(TESTFN, 'w')
44 fp.close()
45 fp = open(TESTFN, 'r')
46 savestdin = sys.stdin
47 try:
48 try:
49 import marshal
Antoine Pitrou4a90ef02012-03-03 02:35:32 +010050 marshal.loads(b'')
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000051 except EOFError:
52 pass
53 finally:
54 sys.stdin = savestdin
55 fp.close()
56 unlink(TESTFN)
Guido van Rossum3bead091992-01-27 17:00:37 +000057
Antoine Pitrou6b4883d2011-10-12 02:54:14 +020058 self.raise_catch(OSError, "OSError")
59 self.assertRaises(OSError, open, 'this file does not exist', 'r')
Guido van Rossum3bead091992-01-27 17:00:37 +000060
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000061 self.raise_catch(ImportError, "ImportError")
62 self.assertRaises(ImportError, __import__, "undefined_module")
Guido van Rossum3bead091992-01-27 17:00:37 +000063
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000064 self.raise_catch(IndexError, "IndexError")
65 x = []
66 self.assertRaises(IndexError, x.__getitem__, 10)
Guido van Rossum3bead091992-01-27 17:00:37 +000067
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000068 self.raise_catch(KeyError, "KeyError")
69 x = {}
70 self.assertRaises(KeyError, x.__getitem__, 'key')
Guido van Rossum3bead091992-01-27 17:00:37 +000071
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000072 self.raise_catch(KeyboardInterrupt, "KeyboardInterrupt")
Guido van Rossum3bead091992-01-27 17:00:37 +000073
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000074 self.raise_catch(MemoryError, "MemoryError")
Guido van Rossum3bead091992-01-27 17:00:37 +000075
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000076 self.raise_catch(NameError, "NameError")
77 try: x = undefined_variable
78 except NameError: pass
Guido van Rossum3bead091992-01-27 17:00:37 +000079
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000080 self.raise_catch(OverflowError, "OverflowError")
81 x = 1
82 for dummy in range(128):
83 x += x # this simply shouldn't blow up
Guido van Rossum3bead091992-01-27 17:00:37 +000084
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000085 self.raise_catch(RuntimeError, "RuntimeError")
Guido van Rossum3bead091992-01-27 17:00:37 +000086
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000087 self.raise_catch(SyntaxError, "SyntaxError")
Georg Brandl7cae87c2006-09-06 06:51:57 +000088 try: exec('/\n')
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000089 except SyntaxError: pass
Guido van Rossum3bead091992-01-27 17:00:37 +000090
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000091 self.raise_catch(IndentationError, "IndentationError")
Fred Drake72e48bd2000-09-08 16:32:34 +000092
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000093 self.raise_catch(TabError, "TabError")
Georg Brandle1b5ac62008-06-04 13:06:58 +000094 try: compile("try:\n\t1/0\n \t1/0\nfinally:\n pass\n",
95 '<string>', 'exec')
96 except TabError: pass
97 else: self.fail("TabError not raised")
Fred Drake72e48bd2000-09-08 16:32:34 +000098
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000099 self.raise_catch(SystemError, "SystemError")
Fred Drake72e48bd2000-09-08 16:32:34 +0000100
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000101 self.raise_catch(SystemExit, "SystemExit")
102 self.assertRaises(SystemExit, sys.exit, 0)
Fred Drake85f36392000-07-11 17:53:00 +0000103
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000104 self.raise_catch(TypeError, "TypeError")
105 try: [] + ()
106 except TypeError: pass
Fred Drake85f36392000-07-11 17:53:00 +0000107
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000108 self.raise_catch(ValueError, "ValueError")
Guido van Rossume63bae62007-07-17 00:34:25 +0000109 self.assertRaises(ValueError, chr, 17<<16)
Guido van Rossum3bead091992-01-27 17:00:37 +0000110
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000111 self.raise_catch(ZeroDivisionError, "ZeroDivisionError")
112 try: x = 1/0
113 except ZeroDivisionError: pass
Guido van Rossum3bead091992-01-27 17:00:37 +0000114
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000115 self.raise_catch(Exception, "Exception")
116 try: x = 1/0
Guido van Rossumb940e112007-01-10 16:19:56 +0000117 except Exception as e: pass
Guido van Rossum3bead091992-01-27 17:00:37 +0000118
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000119 def testSyntaxErrorMessage(self):
120 # make sure the right exception message is raised for each of
121 # these code fragments
Guido van Rossum3bead091992-01-27 17:00:37 +0000122
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000123 def ckmsg(src, msg):
124 try:
125 compile(src, '<fragment>', 'exec')
Guido van Rossumb940e112007-01-10 16:19:56 +0000126 except SyntaxError as e:
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000127 if e.msg != msg:
128 self.fail("expected %s, got %s" % (msg, e.msg))
129 else:
130 self.fail("failed to get expected SyntaxError")
Guido van Rossum3bead091992-01-27 17:00:37 +0000131
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000132 s = '''while 1:
133 try:
134 pass
135 finally:
136 continue'''
Barry Warsaw992cb8a2000-05-25 23:16:54 +0000137
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000138 if not sys.platform.startswith('java'):
139 ckmsg(s, "'continue' not supported inside 'finally' clause")
Jeremy Hyltonede049b2001-09-26 20:01:13 +0000140
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000141 s = '''if 1:
142 try:
143 continue
144 except:
145 pass'''
Jeremy Hyltonede049b2001-09-26 20:01:13 +0000146
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000147 ckmsg(s, "'continue' not properly in loop")
148 ckmsg("continue\n", "'continue' not properly in loop")
Thomas Wouters303de6a2006-04-20 22:42:37 +0000149
Benjamin Peterson17e0bbc2010-06-28 15:39:55 +0000150 @cpython_only
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000151 def testSettingException(self):
152 # test that setting an exception at the C level works even if the
153 # exception object can't be constructed.
Jeremy Hyltonede049b2001-09-26 20:01:13 +0000154
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000155 class BadException(Exception):
156 def __init__(self_):
Collin Winter828f04a2007-08-31 00:04:24 +0000157 raise RuntimeError("can't instantiate BadException")
Finn Bockaa3dc452001-12-08 10:15:48 +0000158
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000159 class InvalidException:
160 pass
Thomas Wouters303de6a2006-04-20 22:42:37 +0000161
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000162 def test_capi1():
163 import _testcapi
164 try:
165 _testcapi.raise_exception(BadException, 1)
Guido van Rossumb940e112007-01-10 16:19:56 +0000166 except TypeError as err:
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000167 exc, err, tb = sys.exc_info()
168 co = tb.tb_frame.f_code
Ezio Melottib3aedd42010-11-20 19:04:17 +0000169 self.assertEqual(co.co_name, "test_capi1")
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000170 self.assertTrue(co.co_filename.endswith('test_exceptions.py'))
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000171 else:
172 self.fail("Expected exception")
Jeremy Hyltonede049b2001-09-26 20:01:13 +0000173
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000174 def test_capi2():
175 import _testcapi
176 try:
177 _testcapi.raise_exception(BadException, 0)
Guido van Rossumb940e112007-01-10 16:19:56 +0000178 except RuntimeError as err:
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000179 exc, err, tb = sys.exc_info()
180 co = tb.tb_frame.f_code
Ezio Melottib3aedd42010-11-20 19:04:17 +0000181 self.assertEqual(co.co_name, "__init__")
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000182 self.assertTrue(co.co_filename.endswith('test_exceptions.py'))
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000183 co2 = tb.tb_frame.f_back.f_code
Ezio Melottib3aedd42010-11-20 19:04:17 +0000184 self.assertEqual(co2.co_name, "test_capi2")
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000185 else:
186 self.fail("Expected exception")
Thomas Wouters477c8d52006-05-27 19:21:47 +0000187
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000188 def test_capi3():
189 import _testcapi
190 self.assertRaises(SystemError, _testcapi.raise_exception,
191 InvalidException, 1)
192
193 if not sys.platform.startswith('java'):
194 test_capi1()
195 test_capi2()
196 test_capi3()
197
Thomas Wouters89f507f2006-12-13 04:49:30 +0000198 def test_WindowsError(self):
199 try:
200 WindowsError
201 except NameError:
202 pass
203 else:
Antoine Pitrou6b4883d2011-10-12 02:54:14 +0200204 self.assertIs(WindowsError, OSError)
205 self.assertEqual(str(OSError(1001)), "1001")
206 self.assertEqual(str(OSError(1001, "message")),
207 "[Errno 1001] message")
208 # POSIX errno (9 aka EBADF) is untranslated
209 w = OSError(9, 'foo', 'bar')
210 self.assertEqual(w.errno, 9)
211 self.assertEqual(w.winerror, None)
212 self.assertEqual(str(w), "[Errno 9] foo: 'bar'")
213 # ERROR_PATH_NOT_FOUND (win error 3) becomes ENOENT (2)
214 w = OSError(0, 'foo', 'bar', 3)
215 self.assertEqual(w.errno, 2)
216 self.assertEqual(w.winerror, 3)
217 self.assertEqual(w.strerror, 'foo')
218 self.assertEqual(w.filename, 'bar')
Richard Oudkerk30147712012-08-28 19:33:26 +0100219 self.assertEqual(str(w), "[WinError 3] foo: 'bar'")
Antoine Pitrou6b4883d2011-10-12 02:54:14 +0200220 # Unknown win error becomes EINVAL (22)
221 w = OSError(0, 'foo', None, 1001)
222 self.assertEqual(w.errno, 22)
223 self.assertEqual(w.winerror, 1001)
224 self.assertEqual(w.strerror, 'foo')
225 self.assertEqual(w.filename, None)
Richard Oudkerk30147712012-08-28 19:33:26 +0100226 self.assertEqual(str(w), "[WinError 1001] foo")
Antoine Pitrou6b4883d2011-10-12 02:54:14 +0200227 # Non-numeric "errno"
228 w = OSError('bar', 'foo')
229 self.assertEqual(w.errno, 'bar')
230 self.assertEqual(w.winerror, None)
231 self.assertEqual(w.strerror, 'foo')
232 self.assertEqual(w.filename, None)
Thomas Wouters89f507f2006-12-13 04:49:30 +0000233
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000234 def testAttributes(self):
235 # test that exception attributes are happy
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000236
237 exceptionList = [
Guido van Rossumebe3e162007-05-17 18:20:34 +0000238 (BaseException, (), {'args' : ()}),
239 (BaseException, (1, ), {'args' : (1,)}),
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000240 (BaseException, ('foo',),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000241 {'args' : ('foo',)}),
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000242 (BaseException, ('foo', 1),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000243 {'args' : ('foo', 1)}),
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000244 (SystemExit, ('foo',),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000245 {'args' : ('foo',), 'code' : 'foo'}),
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000246 (IOError, ('foo',),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000247 {'args' : ('foo',), 'filename' : None,
Thomas Wouters89f507f2006-12-13 04:49:30 +0000248 'errno' : None, 'strerror' : None}),
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000249 (IOError, ('foo', 'bar'),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000250 {'args' : ('foo', 'bar'), 'filename' : None,
Thomas Wouters89f507f2006-12-13 04:49:30 +0000251 'errno' : 'foo', 'strerror' : 'bar'}),
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000252 (IOError, ('foo', 'bar', 'baz'),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000253 {'args' : ('foo', 'bar'), 'filename' : 'baz',
Thomas Wouters89f507f2006-12-13 04:49:30 +0000254 'errno' : 'foo', 'strerror' : 'bar'}),
255 (IOError, ('foo', 'bar', 'baz', 'quux'),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000256 {'args' : ('foo', 'bar', 'baz', 'quux')}),
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000257 (EnvironmentError, ('errnoStr', 'strErrorStr', 'filenameStr'),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000258 {'args' : ('errnoStr', 'strErrorStr'),
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000259 'strerror' : 'strErrorStr', 'errno' : 'errnoStr',
260 'filename' : 'filenameStr'}),
261 (EnvironmentError, (1, 'strErrorStr', 'filenameStr'),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000262 {'args' : (1, 'strErrorStr'), 'errno' : 1,
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000263 'strerror' : 'strErrorStr', 'filename' : 'filenameStr'}),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000264 (SyntaxError, (), {'msg' : None, 'text' : None,
Guido van Rossumd8faa362007-04-27 19:54:29 +0000265 'filename' : None, 'lineno' : None, 'offset' : None,
266 'print_file_and_line' : None}),
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000267 (SyntaxError, ('msgStr',),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000268 {'args' : ('msgStr',), 'text' : None,
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000269 'print_file_and_line' : None, 'msg' : 'msgStr',
270 'filename' : None, 'lineno' : None, 'offset' : None}),
271 (SyntaxError, ('msgStr', ('filenameStr', 'linenoStr', 'offsetStr',
272 'textStr')),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000273 {'offset' : 'offsetStr', 'text' : 'textStr',
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000274 'args' : ('msgStr', ('filenameStr', 'linenoStr',
275 'offsetStr', 'textStr')),
276 'print_file_and_line' : None, 'msg' : 'msgStr',
277 'filename' : 'filenameStr', 'lineno' : 'linenoStr'}),
278 (SyntaxError, ('msgStr', 'filenameStr', 'linenoStr', 'offsetStr',
279 'textStr', 'print_file_and_lineStr'),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000280 {'text' : None,
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000281 'args' : ('msgStr', 'filenameStr', 'linenoStr', 'offsetStr',
282 'textStr', 'print_file_and_lineStr'),
283 'print_file_and_line' : None, 'msg' : 'msgStr',
284 'filename' : None, 'lineno' : None, 'offset' : None}),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000285 (UnicodeError, (), {'args' : (),}),
Walter Dörwaldeceb0fb2007-05-24 17:49:56 +0000286 (UnicodeEncodeError, ('ascii', 'a', 0, 1,
287 'ordinal not in range'),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000288 {'args' : ('ascii', 'a', 0, 1,
Thomas Wouters89f507f2006-12-13 04:49:30 +0000289 'ordinal not in range'),
Guido van Rossumef87d6e2007-05-02 19:09:54 +0000290 'encoding' : 'ascii', 'object' : 'a',
Thomas Wouters89f507f2006-12-13 04:49:30 +0000291 'start' : 0, 'reason' : 'ordinal not in range'}),
Guido van Rossum254348e2007-11-21 19:29:53 +0000292 (UnicodeDecodeError, ('ascii', bytearray(b'\xff'), 0, 1,
Guido van Rossum98297ee2007-11-06 21:34:58 +0000293 'ordinal not in range'),
Guido van Rossum254348e2007-11-21 19:29:53 +0000294 {'args' : ('ascii', bytearray(b'\xff'), 0, 1,
Guido van Rossum98297ee2007-11-06 21:34:58 +0000295 'ordinal not in range'),
296 'encoding' : 'ascii', 'object' : b'\xff',
297 'start' : 0, 'reason' : 'ordinal not in range'}),
Walter Dörwaldeceb0fb2007-05-24 17:49:56 +0000298 (UnicodeDecodeError, ('ascii', b'\xff', 0, 1,
299 'ordinal not in range'),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000300 {'args' : ('ascii', b'\xff', 0, 1,
Thomas Wouters89f507f2006-12-13 04:49:30 +0000301 'ordinal not in range'),
Guido van Rossumb8142c32007-05-08 17:49:10 +0000302 'encoding' : 'ascii', 'object' : b'\xff',
Thomas Wouters89f507f2006-12-13 04:49:30 +0000303 'start' : 0, 'reason' : 'ordinal not in range'}),
Walter Dörwaldeceb0fb2007-05-24 17:49:56 +0000304 (UnicodeTranslateError, ("\u3042", 0, 1, "ouch"),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000305 {'args' : ('\u3042', 0, 1, 'ouch'),
Guido van Rossumef87d6e2007-05-02 19:09:54 +0000306 'object' : '\u3042', 'reason' : 'ouch',
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000307 'start' : 0, 'end' : 1}),
Richard Oudkerk5562d9d2012-07-28 17:45:28 +0100308 (NaiveException, ('foo',),
309 {'args': ('foo',), 'x': 'foo'}),
310 (SlottedNaiveException, ('foo',),
311 {'args': ('foo',), 'x': 'foo'}),
Thomas Wouters477c8d52006-05-27 19:21:47 +0000312 ]
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000313 try:
Antoine Pitrou6b4883d2011-10-12 02:54:14 +0200314 # More tests are in test_WindowsError
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000315 exceptionList.append(
316 (WindowsError, (1, 'strErrorStr', 'filenameStr'),
Guido van Rossumebe3e162007-05-17 18:20:34 +0000317 {'args' : (1, 'strErrorStr'),
Antoine Pitrou6b4883d2011-10-12 02:54:14 +0200318 'strerror' : 'strErrorStr', 'winerror' : None,
319 'errno' : 1, 'filename' : 'filenameStr'})
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000320 )
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000321 except NameError:
322 pass
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000323
Guido van Rossumebe3e162007-05-17 18:20:34 +0000324 for exc, args, expected in exceptionList:
325 try:
326 e = exc(*args)
327 except:
Guido van Rossum98297ee2007-11-06 21:34:58 +0000328 print("\nexc=%r, args=%r" % (exc, args), file=sys.stderr)
Guido van Rossumebe3e162007-05-17 18:20:34 +0000329 raise
330 else:
331 # Verify module name
Richard Oudkerk5562d9d2012-07-28 17:45:28 +0100332 if not type(e).__name__.endswith('NaiveException'):
333 self.assertEqual(type(e).__module__, 'builtins')
Guido van Rossumebe3e162007-05-17 18:20:34 +0000334 # Verify no ref leaks in Exc_str()
335 s = str(e)
336 for checkArgName in expected:
337 value = getattr(e, checkArgName)
Ezio Melottib3aedd42010-11-20 19:04:17 +0000338 self.assertEqual(repr(value),
339 repr(expected[checkArgName]),
340 '%r.%s == %r, expected %r' % (
341 e, checkArgName,
342 value, expected[checkArgName]))
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000343
Guido van Rossumebe3e162007-05-17 18:20:34 +0000344 # test for pickling support
Guido van Rossum99603b02007-07-20 00:22:32 +0000345 for p in [pickle]:
Guido van Rossumebe3e162007-05-17 18:20:34 +0000346 for protocol in range(p.HIGHEST_PROTOCOL + 1):
347 s = p.dumps(e, protocol)
348 new = p.loads(s)
349 for checkArgName in expected:
350 got = repr(getattr(new, checkArgName))
351 want = repr(expected[checkArgName])
Ezio Melottib3aedd42010-11-20 19:04:17 +0000352 self.assertEqual(got, want,
353 'pickled "%r", attribute "%s' %
354 (e, checkArgName))
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000355
Collin Winter828f04a2007-08-31 00:04:24 +0000356 def testWithTraceback(self):
357 try:
358 raise IndexError(4)
359 except:
360 tb = sys.exc_info()[2]
361
362 e = BaseException().with_traceback(tb)
Ezio Melottie9615932010-01-24 19:26:24 +0000363 self.assertIsInstance(e, BaseException)
Collin Winter828f04a2007-08-31 00:04:24 +0000364 self.assertEqual(e.__traceback__, tb)
365
366 e = IndexError(5).with_traceback(tb)
Ezio Melottie9615932010-01-24 19:26:24 +0000367 self.assertIsInstance(e, IndexError)
Collin Winter828f04a2007-08-31 00:04:24 +0000368 self.assertEqual(e.__traceback__, tb)
369
370 class MyException(Exception):
371 pass
372
373 e = MyException().with_traceback(tb)
Ezio Melottie9615932010-01-24 19:26:24 +0000374 self.assertIsInstance(e, MyException)
Collin Winter828f04a2007-08-31 00:04:24 +0000375 self.assertEqual(e.__traceback__, tb)
376
377 def testInvalidTraceback(self):
378 try:
379 Exception().__traceback__ = 5
380 except TypeError as e:
Benjamin Peterson577473f2010-01-19 00:09:57 +0000381 self.assertIn("__traceback__ must be a traceback", str(e))
Collin Winter828f04a2007-08-31 00:04:24 +0000382 else:
383 self.fail("No exception raised")
384
Georg Brandlab6f2f62009-03-31 04:16:10 +0000385 def testInvalidAttrs(self):
386 self.assertRaises(TypeError, setattr, Exception(), '__cause__', 1)
387 self.assertRaises(TypeError, delattr, Exception(), '__cause__')
388 self.assertRaises(TypeError, setattr, Exception(), '__context__', 1)
389 self.assertRaises(TypeError, delattr, Exception(), '__context__')
390
Collin Winter828f04a2007-08-31 00:04:24 +0000391 def testNoneClearsTracebackAttr(self):
392 try:
393 raise IndexError(4)
394 except:
395 tb = sys.exc_info()[2]
396
397 e = Exception()
398 e.__traceback__ = tb
399 e.__traceback__ = None
400 self.assertEqual(e.__traceback__, None)
401
402 def testChainingAttrs(self):
403 e = Exception()
Nick Coghlanab7bf212012-02-26 17:49:52 +1000404 self.assertIsNone(e.__context__)
Benjamin Petersond5a1c442012-05-14 22:09:31 -0700405 self.assertIsNone(e.__cause__)
Collin Winter828f04a2007-08-31 00:04:24 +0000406
407 e = TypeError()
Nick Coghlanab7bf212012-02-26 17:49:52 +1000408 self.assertIsNone(e.__context__)
Benjamin Petersond5a1c442012-05-14 22:09:31 -0700409 self.assertIsNone(e.__cause__)
Collin Winter828f04a2007-08-31 00:04:24 +0000410
411 class MyException(EnvironmentError):
412 pass
413
414 e = MyException()
Nick Coghlanab7bf212012-02-26 17:49:52 +1000415 self.assertIsNone(e.__context__)
Benjamin Petersond5a1c442012-05-14 22:09:31 -0700416 self.assertIsNone(e.__cause__)
Nick Coghlanab7bf212012-02-26 17:49:52 +1000417
418 def testChainingDescriptors(self):
419 try:
420 raise Exception()
421 except Exception as exc:
422 e = exc
423
424 self.assertIsNone(e.__context__)
Benjamin Petersond5a1c442012-05-14 22:09:31 -0700425 self.assertIsNone(e.__cause__)
426 self.assertFalse(e.__suppress_context__)
Nick Coghlanab7bf212012-02-26 17:49:52 +1000427
428 e.__context__ = NameError()
429 e.__cause__ = None
430 self.assertIsInstance(e.__context__, NameError)
431 self.assertIsNone(e.__cause__)
Benjamin Petersond5a1c442012-05-14 22:09:31 -0700432 self.assertTrue(e.__suppress_context__)
433 e.__suppress_context__ = False
434 self.assertFalse(e.__suppress_context__)
Collin Winter828f04a2007-08-31 00:04:24 +0000435
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000436 def testKeywordArgs(self):
437 # test that builtin exception don't take keyword args,
438 # but user-defined subclasses can if they want
439 self.assertRaises(TypeError, BaseException, a=1)
440
441 class DerivedException(BaseException):
442 def __init__(self, fancy_arg):
443 BaseException.__init__(self)
444 self.fancy_arg = fancy_arg
445
446 x = DerivedException(fancy_arg=42)
Ezio Melottib3aedd42010-11-20 19:04:17 +0000447 self.assertEqual(x.fancy_arg, 42)
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000448
Brett Cannon31f59292011-02-21 19:29:56 +0000449 @no_tracing
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000450 def testInfiniteRecursion(self):
451 def f():
452 return f()
453 self.assertRaises(RuntimeError, f)
454
455 def g():
456 try:
457 return g()
458 except ValueError:
459 return -1
460 self.assertRaises(RuntimeError, g)
461
Ezio Melotti2f5a78c2009-12-24 22:54:06 +0000462 def test_str(self):
463 # Make sure both instances and classes have a str representation.
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000464 self.assertTrue(str(Exception))
465 self.assertTrue(str(Exception('a')))
Ezio Melotti2f5a78c2009-12-24 22:54:06 +0000466 self.assertTrue(str(Exception('a', 'b')))
Thomas Wouters89f507f2006-12-13 04:49:30 +0000467
Barry Warsaw8d109cb2008-05-08 04:26:35 +0000468 def testExceptionCleanupNames(self):
469 # Make sure the local variable bound to the exception instance by
470 # an "except" statement is only visible inside the except block.
Guido van Rossumb940e112007-01-10 16:19:56 +0000471 try:
472 raise Exception()
473 except Exception as e:
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000474 self.assertTrue(e)
Guido van Rossumb940e112007-01-10 16:19:56 +0000475 del e
Ezio Melottib58e0bd2010-01-23 15:40:09 +0000476 self.assertNotIn('e', locals())
Guido van Rossumb940e112007-01-10 16:19:56 +0000477
Barry Warsaw8d109cb2008-05-08 04:26:35 +0000478 def testExceptionCleanupState(self):
479 # Make sure exception state is cleaned up as soon as the except
480 # block is left. See #2507
481
482 class MyException(Exception):
483 def __init__(self, obj):
484 self.obj = obj
485 class MyObj:
486 pass
487
488 def inner_raising_func():
489 # Create some references in exception value and traceback
490 local_ref = obj
491 raise MyException(obj)
492
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000493 # Qualified "except" with "as"
Barry Warsaw8d109cb2008-05-08 04:26:35 +0000494 obj = MyObj()
495 wr = weakref.ref(obj)
496 try:
497 inner_raising_func()
498 except MyException as e:
499 pass
500 obj = None
501 obj = wr()
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000502 self.assertTrue(obj is None, "%s" % obj)
Barry Warsaw8d109cb2008-05-08 04:26:35 +0000503
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000504 # Qualified "except" without "as"
505 obj = MyObj()
506 wr = weakref.ref(obj)
507 try:
508 inner_raising_func()
509 except MyException:
510 pass
511 obj = None
512 obj = wr()
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000513 self.assertTrue(obj is None, "%s" % obj)
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000514
515 # Bare "except"
516 obj = MyObj()
517 wr = weakref.ref(obj)
518 try:
519 inner_raising_func()
520 except:
521 pass
522 obj = None
523 obj = wr()
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000524 self.assertTrue(obj is None, "%s" % obj)
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000525
526 # "except" with premature block leave
527 obj = MyObj()
528 wr = weakref.ref(obj)
529 for i in [0]:
530 try:
531 inner_raising_func()
532 except:
533 break
534 obj = None
535 obj = wr()
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000536 self.assertTrue(obj is None, "%s" % obj)
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000537
538 # "except" block raising another exception
539 obj = MyObj()
540 wr = weakref.ref(obj)
541 try:
542 try:
543 inner_raising_func()
544 except:
545 raise KeyError
Guido van Rossumb4fb6e42008-06-14 20:20:24 +0000546 except KeyError as e:
547 # We want to test that the except block above got rid of
548 # the exception raised in inner_raising_func(), but it
549 # also ends up in the __context__ of the KeyError, so we
550 # must clear the latter manually for our test to succeed.
551 e.__context__ = None
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000552 obj = None
553 obj = wr()
Philip Jenvey3acc7ef2012-11-13 12:26:31 -0800554 gc_collect()
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000555 self.assertTrue(obj is None, "%s" % obj)
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000556
557 # Some complicated construct
558 obj = MyObj()
559 wr = weakref.ref(obj)
560 try:
561 inner_raising_func()
562 except MyException:
563 try:
564 try:
565 raise
566 finally:
567 raise
568 except MyException:
569 pass
570 obj = None
Philip Jenvey3acc7ef2012-11-13 12:26:31 -0800571 gc_collect()
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000572 obj = wr()
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000573 self.assertTrue(obj is None, "%s" % obj)
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000574
575 # Inside an exception-silencing "with" block
576 class Context:
577 def __enter__(self):
578 return self
579 def __exit__ (self, exc_type, exc_value, exc_tb):
580 return True
581 obj = MyObj()
582 wr = weakref.ref(obj)
583 with Context():
584 inner_raising_func()
585 obj = None
Philip Jenvey3acc7ef2012-11-13 12:26:31 -0800586 gc_collect()
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000587 obj = wr()
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000588 self.assertTrue(obj is None, "%s" % obj)
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000589
Amaury Forgeot d'Arcba117ef2010-09-10 21:39:53 +0000590 def test_exception_target_in_nested_scope(self):
591 # issue 4617: This used to raise a SyntaxError
592 # "can not delete variable 'e' referenced in nested scope"
593 def print_error():
594 e
595 try:
596 something
597 except Exception as e:
598 print_error()
599 # implicit "del e" here
600
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000601 def test_generator_leaking(self):
602 # Test that generator exception state doesn't leak into the calling
603 # frame
604 def yield_raise():
605 try:
606 raise KeyError("caught")
607 except KeyError:
608 yield sys.exc_info()[0]
609 yield sys.exc_info()[0]
610 yield sys.exc_info()[0]
611 g = yield_raise()
Ezio Melottib3aedd42010-11-20 19:04:17 +0000612 self.assertEqual(next(g), KeyError)
613 self.assertEqual(sys.exc_info()[0], None)
614 self.assertEqual(next(g), KeyError)
615 self.assertEqual(sys.exc_info()[0], None)
616 self.assertEqual(next(g), None)
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000617
618 # Same test, but inside an exception handler
619 try:
620 raise TypeError("foo")
621 except TypeError:
622 g = yield_raise()
Ezio Melottib3aedd42010-11-20 19:04:17 +0000623 self.assertEqual(next(g), KeyError)
624 self.assertEqual(sys.exc_info()[0], TypeError)
625 self.assertEqual(next(g), KeyError)
626 self.assertEqual(sys.exc_info()[0], TypeError)
627 self.assertEqual(next(g), TypeError)
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000628 del g
Ezio Melottib3aedd42010-11-20 19:04:17 +0000629 self.assertEqual(sys.exc_info()[0], TypeError)
Thomas Wouters89f507f2006-12-13 04:49:30 +0000630
Benjamin Peterson83195c32011-07-03 13:44:00 -0500631 def test_generator_leaking2(self):
632 # See issue 12475.
633 def g():
634 yield
635 try:
636 raise RuntimeError
637 except RuntimeError:
638 it = g()
639 next(it)
640 try:
641 next(it)
642 except StopIteration:
643 pass
644 self.assertEqual(sys.exc_info(), (None, None, None))
645
Benjamin Petersonac913412011-07-03 16:25:11 -0500646 def test_generator_doesnt_retain_old_exc(self):
647 def g():
648 self.assertIsInstance(sys.exc_info()[1], RuntimeError)
649 yield
650 self.assertEqual(sys.exc_info(), (None, None, None))
651 it = g()
652 try:
653 raise RuntimeError
654 except RuntimeError:
655 next(it)
656 self.assertRaises(StopIteration, next, it)
657
Benjamin Petersonae5f2f42010-03-07 17:10:51 +0000658 def test_generator_finalizing_and_exc_info(self):
659 # See #7173
660 def simple_gen():
661 yield 1
662 def run_gen():
663 gen = simple_gen()
664 try:
665 raise RuntimeError
666 except RuntimeError:
667 return next(gen)
668 run_gen()
669 gc_collect()
670 self.assertEqual(sys.exc_info(), (None, None, None))
671
Antoine Pitroua370fcf2011-08-20 14:15:03 +0200672 def _check_generator_cleanup_exc_state(self, testfunc):
673 # Issue #12791: exception state is cleaned up as soon as a generator
674 # is closed (reference cycles are broken).
675 class MyException(Exception):
676 def __init__(self, obj):
677 self.obj = obj
678 class MyObj:
679 pass
680
681 def raising_gen():
682 try:
683 raise MyException(obj)
684 except MyException:
685 yield
686
687 obj = MyObj()
688 wr = weakref.ref(obj)
689 g = raising_gen()
690 next(g)
691 testfunc(g)
692 g = obj = None
693 obj = wr()
694 self.assertIs(obj, None)
695
696 def test_generator_throw_cleanup_exc_state(self):
697 def do_throw(g):
698 try:
699 g.throw(RuntimeError())
700 except RuntimeError:
701 pass
702 self._check_generator_cleanup_exc_state(do_throw)
703
704 def test_generator_close_cleanup_exc_state(self):
705 def do_close(g):
706 g.close()
707 self._check_generator_cleanup_exc_state(do_close)
708
709 def test_generator_del_cleanup_exc_state(self):
710 def do_del(g):
711 g = None
712 self._check_generator_cleanup_exc_state(do_del)
713
714 def test_generator_next_cleanup_exc_state(self):
715 def do_next(g):
716 try:
717 next(g)
718 except StopIteration:
719 pass
720 else:
721 self.fail("should have raised StopIteration")
722 self._check_generator_cleanup_exc_state(do_next)
723
724 def test_generator_send_cleanup_exc_state(self):
725 def do_send(g):
726 try:
727 g.send(None)
728 except StopIteration:
729 pass
730 else:
731 self.fail("should have raised StopIteration")
732 self._check_generator_cleanup_exc_state(do_send)
733
Benjamin Peterson27d63672008-06-15 20:09:12 +0000734 def test_3114(self):
735 # Bug #3114: in its destructor, MyObject retrieves a pointer to
736 # obsolete and/or deallocated objects.
Benjamin Peterson979f3112008-06-15 00:05:44 +0000737 class MyObject:
738 def __del__(self):
739 nonlocal e
740 e = sys.exc_info()
741 e = ()
742 try:
743 raise Exception(MyObject())
744 except:
745 pass
Ezio Melottib3aedd42010-11-20 19:04:17 +0000746 self.assertEqual(e, (None, None, None))
Benjamin Peterson979f3112008-06-15 00:05:44 +0000747
Eric Smith0facd772010-02-24 15:42:29 +0000748 def testUnicodeChangeAttributes(self):
749 # See issue 7309. This was a crasher.
750
751 u = UnicodeEncodeError('baz', 'xxxxx', 1, 5, 'foo')
752 self.assertEqual(str(u), "'baz' codec can't encode characters in position 1-4: foo")
753 u.end = 2
754 self.assertEqual(str(u), "'baz' codec can't encode character '\\x78' in position 1: foo")
755 u.end = 5
756 u.reason = 0x345345345345345345
757 self.assertEqual(str(u), "'baz' codec can't encode characters in position 1-4: 965230951443685724997")
758 u.encoding = 4000
759 self.assertEqual(str(u), "'4000' codec can't encode characters in position 1-4: 965230951443685724997")
760 u.start = 1000
761 self.assertEqual(str(u), "'4000' codec can't encode characters in position 1000-4: 965230951443685724997")
762
763 u = UnicodeDecodeError('baz', b'xxxxx', 1, 5, 'foo')
764 self.assertEqual(str(u), "'baz' codec can't decode bytes in position 1-4: foo")
765 u.end = 2
766 self.assertEqual(str(u), "'baz' codec can't decode byte 0x78 in position 1: foo")
767 u.end = 5
768 u.reason = 0x345345345345345345
769 self.assertEqual(str(u), "'baz' codec can't decode bytes in position 1-4: 965230951443685724997")
770 u.encoding = 4000
771 self.assertEqual(str(u), "'4000' codec can't decode bytes in position 1-4: 965230951443685724997")
772 u.start = 1000
773 self.assertEqual(str(u), "'4000' codec can't decode bytes in position 1000-4: 965230951443685724997")
774
775 u = UnicodeTranslateError('xxxx', 1, 5, 'foo')
776 self.assertEqual(str(u), "can't translate characters in position 1-4: foo")
777 u.end = 2
778 self.assertEqual(str(u), "can't translate character '\\x78' in position 1: foo")
779 u.end = 5
780 u.reason = 0x345345345345345345
781 self.assertEqual(str(u), "can't translate characters in position 1-4: 965230951443685724997")
782 u.start = 1000
783 self.assertEqual(str(u), "can't translate characters in position 1000-4: 965230951443685724997")
Benjamin Peterson6e7740c2008-08-20 23:23:34 +0000784
Brett Cannon31f59292011-02-21 19:29:56 +0000785 @no_tracing
Benjamin Peterson69c88f72008-07-31 01:47:08 +0000786 def test_badisinstance(self):
787 # Bug #2542: if issubclass(e, MyException) raises an exception,
788 # it should be ignored
789 class Meta(type):
790 def __subclasscheck__(cls, subclass):
791 raise ValueError()
792 class MyException(Exception, metaclass=Meta):
793 pass
794
795 with captured_output("stderr") as stderr:
796 try:
797 raise KeyError()
798 except MyException as e:
799 self.fail("exception should not be a MyException")
800 except KeyError:
801 pass
802 except:
Antoine Pitrouec569b72008-08-26 22:40:48 +0000803 self.fail("Should have raised KeyError")
Benjamin Peterson69c88f72008-07-31 01:47:08 +0000804 else:
Antoine Pitrouec569b72008-08-26 22:40:48 +0000805 self.fail("Should have raised KeyError")
806
807 def g():
808 try:
809 return g()
810 except RuntimeError:
811 return sys.exc_info()
812 e, v, tb = g()
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000813 self.assertTrue(isinstance(v, RuntimeError), type(v))
Benjamin Peterson577473f2010-01-19 00:09:57 +0000814 self.assertIn("maximum recursion depth exceeded", str(v))
Benjamin Peterson69c88f72008-07-31 01:47:08 +0000815
Amaury Forgeot d'Arce19cadb2008-07-31 22:56:02 +0000816
817 def test_MemoryError(self):
818 # PyErr_NoMemory always raises the same exception instance.
819 # Check that the traceback is not doubled.
820 import traceback
Benjamin Peterson0067bd62008-08-16 16:11:03 +0000821 from _testcapi import raise_memoryerror
Amaury Forgeot d'Arce19cadb2008-07-31 22:56:02 +0000822 def raiseMemError():
823 try:
Benjamin Peterson0067bd62008-08-16 16:11:03 +0000824 raise_memoryerror()
Amaury Forgeot d'Arce19cadb2008-07-31 22:56:02 +0000825 except MemoryError as e:
826 tb = e.__traceback__
827 else:
828 self.fail("Should have raises a MemoryError")
829 return traceback.format_tb(tb)
830
831 tb1 = raiseMemError()
832 tb2 = raiseMemError()
833 self.assertEqual(tb1, tb2)
834
Benjamin Peterson17e0bbc2010-06-28 15:39:55 +0000835 @cpython_only
Georg Brandl1e28a272009-12-28 08:41:01 +0000836 def test_exception_with_doc(self):
837 import _testcapi
838 doc2 = "This is a test docstring."
839 doc4 = "This is another test docstring."
840
841 self.assertRaises(SystemError, _testcapi.make_exception_with_doc,
842 "error1")
843
844 # test basic usage of PyErr_NewException
845 error1 = _testcapi.make_exception_with_doc("_testcapi.error1")
846 self.assertIs(type(error1), type)
847 self.assertTrue(issubclass(error1, Exception))
848 self.assertIsNone(error1.__doc__)
849
850 # test with given docstring
851 error2 = _testcapi.make_exception_with_doc("_testcapi.error2", doc2)
852 self.assertEqual(error2.__doc__, doc2)
853
854 # test with explicit base (without docstring)
855 error3 = _testcapi.make_exception_with_doc("_testcapi.error3",
856 base=error2)
857 self.assertTrue(issubclass(error3, error2))
858
859 # test with explicit base tuple
860 class C(object):
861 pass
862 error4 = _testcapi.make_exception_with_doc("_testcapi.error4", doc4,
863 (error3, C))
864 self.assertTrue(issubclass(error4, error3))
865 self.assertTrue(issubclass(error4, C))
866 self.assertEqual(error4.__doc__, doc4)
867
868 # test with explicit dictionary
869 error5 = _testcapi.make_exception_with_doc("_testcapi.error5", "",
870 error4, {'a': 1})
871 self.assertTrue(issubclass(error5, error4))
872 self.assertEqual(error5.a, 1)
873 self.assertEqual(error5.__doc__, "")
874
Antoine Pitrou07e20ef2010-10-28 22:56:58 +0000875 def test_memory_error_cleanup(self):
876 # Issue #5437: preallocated MemoryError instances should not keep
877 # traceback objects alive.
878 from _testcapi import raise_memoryerror
879 class C:
880 pass
881 wr = None
882 def inner():
883 nonlocal wr
884 c = C()
885 wr = weakref.ref(c)
886 raise_memoryerror()
887 # We cannot use assertRaises since it manually deletes the traceback
888 try:
889 inner()
890 except MemoryError as e:
891 self.assertNotEqual(wr(), None)
892 else:
893 self.fail("MemoryError not raised")
894 self.assertEqual(wr(), None)
895
Brett Cannon31f59292011-02-21 19:29:56 +0000896 @no_tracing
Antoine Pitrou07e20ef2010-10-28 22:56:58 +0000897 def test_recursion_error_cleanup(self):
898 # Same test as above, but with "recursion exceeded" errors
899 class C:
900 pass
901 wr = None
902 def inner():
903 nonlocal wr
904 c = C()
905 wr = weakref.ref(c)
906 inner()
907 # We cannot use assertRaises since it manually deletes the traceback
908 try:
909 inner()
910 except RuntimeError as e:
911 self.assertNotEqual(wr(), None)
912 else:
913 self.fail("RuntimeError not raised")
914 self.assertEqual(wr(), None)
Georg Brandl1e28a272009-12-28 08:41:01 +0000915
Antoine Pitroua7622852011-09-01 21:37:43 +0200916 def test_errno_ENOTDIR(self):
917 # Issue #12802: "not a directory" errors are ENOTDIR even on Windows
918 with self.assertRaises(OSError) as cm:
919 os.listdir(__file__)
920 self.assertEqual(cm.exception.errno, errno.ENOTDIR, cm.exception)
921
922
Brett Cannon79ec55e2012-04-12 20:24:54 -0400923class ImportErrorTests(unittest.TestCase):
924
925 def test_attributes(self):
926 # Setting 'name' and 'path' should not be a problem.
927 exc = ImportError('test')
928 self.assertIsNone(exc.name)
929 self.assertIsNone(exc.path)
930
931 exc = ImportError('test', name='somemodule')
932 self.assertEqual(exc.name, 'somemodule')
933 self.assertIsNone(exc.path)
934
935 exc = ImportError('test', path='somepath')
936 self.assertEqual(exc.path, 'somepath')
937 self.assertIsNone(exc.name)
938
939 exc = ImportError('test', path='somepath', name='somename')
940 self.assertEqual(exc.name, 'somename')
941 self.assertEqual(exc.path, 'somepath')
942
Brett Cannon07c6e712012-08-24 13:05:09 -0400943 def test_non_str_argument(self):
944 # Issue #15778
945 arg = b'abc'
946 exc = ImportError(arg)
947 self.assertEqual(str(arg), str(exc))
Brett Cannon79ec55e2012-04-12 20:24:54 -0400948
949
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000950def test_main():
Brett Cannon79ec55e2012-04-12 20:24:54 -0400951 run_unittest(ExceptionTests, ImportErrorTests)
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000952
953if __name__ == '__main__':
Guido van Rossumb8142c32007-05-08 17:49:10 +0000954 unittest.main()