blob: 413f4dd62bc6024b1d6d70013af11c93f12f80dd [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()
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000554 self.assertTrue(obj is None, "%s" % obj)
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000555
556 # Some complicated construct
557 obj = MyObj()
558 wr = weakref.ref(obj)
559 try:
560 inner_raising_func()
561 except MyException:
562 try:
563 try:
564 raise
565 finally:
566 raise
567 except MyException:
568 pass
569 obj = None
570 obj = wr()
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000571 self.assertTrue(obj is None, "%s" % obj)
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000572
573 # Inside an exception-silencing "with" block
574 class Context:
575 def __enter__(self):
576 return self
577 def __exit__ (self, exc_type, exc_value, exc_tb):
578 return True
579 obj = MyObj()
580 wr = weakref.ref(obj)
581 with Context():
582 inner_raising_func()
583 obj = None
584 obj = wr()
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000585 self.assertTrue(obj is None, "%s" % obj)
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000586
Amaury Forgeot d'Arcba117ef2010-09-10 21:39:53 +0000587 def test_exception_target_in_nested_scope(self):
588 # issue 4617: This used to raise a SyntaxError
589 # "can not delete variable 'e' referenced in nested scope"
590 def print_error():
591 e
592 try:
593 something
594 except Exception as e:
595 print_error()
596 # implicit "del e" here
597
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000598 def test_generator_leaking(self):
599 # Test that generator exception state doesn't leak into the calling
600 # frame
601 def yield_raise():
602 try:
603 raise KeyError("caught")
604 except KeyError:
605 yield sys.exc_info()[0]
606 yield sys.exc_info()[0]
607 yield sys.exc_info()[0]
608 g = yield_raise()
Ezio Melottib3aedd42010-11-20 19:04:17 +0000609 self.assertEqual(next(g), KeyError)
610 self.assertEqual(sys.exc_info()[0], None)
611 self.assertEqual(next(g), KeyError)
612 self.assertEqual(sys.exc_info()[0], None)
613 self.assertEqual(next(g), None)
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000614
615 # Same test, but inside an exception handler
616 try:
617 raise TypeError("foo")
618 except TypeError:
619 g = yield_raise()
Ezio Melottib3aedd42010-11-20 19:04:17 +0000620 self.assertEqual(next(g), KeyError)
621 self.assertEqual(sys.exc_info()[0], TypeError)
622 self.assertEqual(next(g), KeyError)
623 self.assertEqual(sys.exc_info()[0], TypeError)
624 self.assertEqual(next(g), TypeError)
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000625 del g
Ezio Melottib3aedd42010-11-20 19:04:17 +0000626 self.assertEqual(sys.exc_info()[0], TypeError)
Thomas Wouters89f507f2006-12-13 04:49:30 +0000627
Benjamin Peterson83195c32011-07-03 13:44:00 -0500628 def test_generator_leaking2(self):
629 # See issue 12475.
630 def g():
631 yield
632 try:
633 raise RuntimeError
634 except RuntimeError:
635 it = g()
636 next(it)
637 try:
638 next(it)
639 except StopIteration:
640 pass
641 self.assertEqual(sys.exc_info(), (None, None, None))
642
Benjamin Petersonac913412011-07-03 16:25:11 -0500643 def test_generator_doesnt_retain_old_exc(self):
644 def g():
645 self.assertIsInstance(sys.exc_info()[1], RuntimeError)
646 yield
647 self.assertEqual(sys.exc_info(), (None, None, None))
648 it = g()
649 try:
650 raise RuntimeError
651 except RuntimeError:
652 next(it)
653 self.assertRaises(StopIteration, next, it)
654
Benjamin Petersonae5f2f42010-03-07 17:10:51 +0000655 def test_generator_finalizing_and_exc_info(self):
656 # See #7173
657 def simple_gen():
658 yield 1
659 def run_gen():
660 gen = simple_gen()
661 try:
662 raise RuntimeError
663 except RuntimeError:
664 return next(gen)
665 run_gen()
666 gc_collect()
667 self.assertEqual(sys.exc_info(), (None, None, None))
668
Antoine Pitroua370fcf2011-08-20 14:15:03 +0200669 def _check_generator_cleanup_exc_state(self, testfunc):
670 # Issue #12791: exception state is cleaned up as soon as a generator
671 # is closed (reference cycles are broken).
672 class MyException(Exception):
673 def __init__(self, obj):
674 self.obj = obj
675 class MyObj:
676 pass
677
678 def raising_gen():
679 try:
680 raise MyException(obj)
681 except MyException:
682 yield
683
684 obj = MyObj()
685 wr = weakref.ref(obj)
686 g = raising_gen()
687 next(g)
688 testfunc(g)
689 g = obj = None
690 obj = wr()
691 self.assertIs(obj, None)
692
693 def test_generator_throw_cleanup_exc_state(self):
694 def do_throw(g):
695 try:
696 g.throw(RuntimeError())
697 except RuntimeError:
698 pass
699 self._check_generator_cleanup_exc_state(do_throw)
700
701 def test_generator_close_cleanup_exc_state(self):
702 def do_close(g):
703 g.close()
704 self._check_generator_cleanup_exc_state(do_close)
705
706 def test_generator_del_cleanup_exc_state(self):
707 def do_del(g):
708 g = None
709 self._check_generator_cleanup_exc_state(do_del)
710
711 def test_generator_next_cleanup_exc_state(self):
712 def do_next(g):
713 try:
714 next(g)
715 except StopIteration:
716 pass
717 else:
718 self.fail("should have raised StopIteration")
719 self._check_generator_cleanup_exc_state(do_next)
720
721 def test_generator_send_cleanup_exc_state(self):
722 def do_send(g):
723 try:
724 g.send(None)
725 except StopIteration:
726 pass
727 else:
728 self.fail("should have raised StopIteration")
729 self._check_generator_cleanup_exc_state(do_send)
730
Benjamin Peterson27d63672008-06-15 20:09:12 +0000731 def test_3114(self):
732 # Bug #3114: in its destructor, MyObject retrieves a pointer to
733 # obsolete and/or deallocated objects.
Benjamin Peterson979f3112008-06-15 00:05:44 +0000734 class MyObject:
735 def __del__(self):
736 nonlocal e
737 e = sys.exc_info()
738 e = ()
739 try:
740 raise Exception(MyObject())
741 except:
742 pass
Ezio Melottib3aedd42010-11-20 19:04:17 +0000743 self.assertEqual(e, (None, None, None))
Benjamin Peterson979f3112008-06-15 00:05:44 +0000744
Eric Smith0facd772010-02-24 15:42:29 +0000745 def testUnicodeChangeAttributes(self):
746 # See issue 7309. This was a crasher.
747
748 u = UnicodeEncodeError('baz', 'xxxxx', 1, 5, 'foo')
749 self.assertEqual(str(u), "'baz' codec can't encode characters in position 1-4: foo")
750 u.end = 2
751 self.assertEqual(str(u), "'baz' codec can't encode character '\\x78' in position 1: foo")
752 u.end = 5
753 u.reason = 0x345345345345345345
754 self.assertEqual(str(u), "'baz' codec can't encode characters in position 1-4: 965230951443685724997")
755 u.encoding = 4000
756 self.assertEqual(str(u), "'4000' codec can't encode characters in position 1-4: 965230951443685724997")
757 u.start = 1000
758 self.assertEqual(str(u), "'4000' codec can't encode characters in position 1000-4: 965230951443685724997")
759
760 u = UnicodeDecodeError('baz', b'xxxxx', 1, 5, 'foo')
761 self.assertEqual(str(u), "'baz' codec can't decode bytes in position 1-4: foo")
762 u.end = 2
763 self.assertEqual(str(u), "'baz' codec can't decode byte 0x78 in position 1: foo")
764 u.end = 5
765 u.reason = 0x345345345345345345
766 self.assertEqual(str(u), "'baz' codec can't decode bytes in position 1-4: 965230951443685724997")
767 u.encoding = 4000
768 self.assertEqual(str(u), "'4000' codec can't decode bytes in position 1-4: 965230951443685724997")
769 u.start = 1000
770 self.assertEqual(str(u), "'4000' codec can't decode bytes in position 1000-4: 965230951443685724997")
771
772 u = UnicodeTranslateError('xxxx', 1, 5, 'foo')
773 self.assertEqual(str(u), "can't translate characters in position 1-4: foo")
774 u.end = 2
775 self.assertEqual(str(u), "can't translate character '\\x78' in position 1: foo")
776 u.end = 5
777 u.reason = 0x345345345345345345
778 self.assertEqual(str(u), "can't translate characters in position 1-4: 965230951443685724997")
779 u.start = 1000
780 self.assertEqual(str(u), "can't translate characters in position 1000-4: 965230951443685724997")
Benjamin Peterson6e7740c2008-08-20 23:23:34 +0000781
Brett Cannon31f59292011-02-21 19:29:56 +0000782 @no_tracing
Benjamin Peterson69c88f72008-07-31 01:47:08 +0000783 def test_badisinstance(self):
784 # Bug #2542: if issubclass(e, MyException) raises an exception,
785 # it should be ignored
786 class Meta(type):
787 def __subclasscheck__(cls, subclass):
788 raise ValueError()
789 class MyException(Exception, metaclass=Meta):
790 pass
791
792 with captured_output("stderr") as stderr:
793 try:
794 raise KeyError()
795 except MyException as e:
796 self.fail("exception should not be a MyException")
797 except KeyError:
798 pass
799 except:
Antoine Pitrouec569b72008-08-26 22:40:48 +0000800 self.fail("Should have raised KeyError")
Benjamin Peterson69c88f72008-07-31 01:47:08 +0000801 else:
Antoine Pitrouec569b72008-08-26 22:40:48 +0000802 self.fail("Should have raised KeyError")
803
804 def g():
805 try:
806 return g()
807 except RuntimeError:
808 return sys.exc_info()
809 e, v, tb = g()
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000810 self.assertTrue(isinstance(v, RuntimeError), type(v))
Benjamin Peterson577473f2010-01-19 00:09:57 +0000811 self.assertIn("maximum recursion depth exceeded", str(v))
Benjamin Peterson69c88f72008-07-31 01:47:08 +0000812
Amaury Forgeot d'Arce19cadb2008-07-31 22:56:02 +0000813
814 def test_MemoryError(self):
815 # PyErr_NoMemory always raises the same exception instance.
816 # Check that the traceback is not doubled.
817 import traceback
Benjamin Peterson0067bd62008-08-16 16:11:03 +0000818 from _testcapi import raise_memoryerror
Amaury Forgeot d'Arce19cadb2008-07-31 22:56:02 +0000819 def raiseMemError():
820 try:
Benjamin Peterson0067bd62008-08-16 16:11:03 +0000821 raise_memoryerror()
Amaury Forgeot d'Arce19cadb2008-07-31 22:56:02 +0000822 except MemoryError as e:
823 tb = e.__traceback__
824 else:
825 self.fail("Should have raises a MemoryError")
826 return traceback.format_tb(tb)
827
828 tb1 = raiseMemError()
829 tb2 = raiseMemError()
830 self.assertEqual(tb1, tb2)
831
Benjamin Peterson17e0bbc2010-06-28 15:39:55 +0000832 @cpython_only
Georg Brandl1e28a272009-12-28 08:41:01 +0000833 def test_exception_with_doc(self):
834 import _testcapi
835 doc2 = "This is a test docstring."
836 doc4 = "This is another test docstring."
837
838 self.assertRaises(SystemError, _testcapi.make_exception_with_doc,
839 "error1")
840
841 # test basic usage of PyErr_NewException
842 error1 = _testcapi.make_exception_with_doc("_testcapi.error1")
843 self.assertIs(type(error1), type)
844 self.assertTrue(issubclass(error1, Exception))
845 self.assertIsNone(error1.__doc__)
846
847 # test with given docstring
848 error2 = _testcapi.make_exception_with_doc("_testcapi.error2", doc2)
849 self.assertEqual(error2.__doc__, doc2)
850
851 # test with explicit base (without docstring)
852 error3 = _testcapi.make_exception_with_doc("_testcapi.error3",
853 base=error2)
854 self.assertTrue(issubclass(error3, error2))
855
856 # test with explicit base tuple
857 class C(object):
858 pass
859 error4 = _testcapi.make_exception_with_doc("_testcapi.error4", doc4,
860 (error3, C))
861 self.assertTrue(issubclass(error4, error3))
862 self.assertTrue(issubclass(error4, C))
863 self.assertEqual(error4.__doc__, doc4)
864
865 # test with explicit dictionary
866 error5 = _testcapi.make_exception_with_doc("_testcapi.error5", "",
867 error4, {'a': 1})
868 self.assertTrue(issubclass(error5, error4))
869 self.assertEqual(error5.a, 1)
870 self.assertEqual(error5.__doc__, "")
871
Antoine Pitrou07e20ef2010-10-28 22:56:58 +0000872 def test_memory_error_cleanup(self):
873 # Issue #5437: preallocated MemoryError instances should not keep
874 # traceback objects alive.
875 from _testcapi import raise_memoryerror
876 class C:
877 pass
878 wr = None
879 def inner():
880 nonlocal wr
881 c = C()
882 wr = weakref.ref(c)
883 raise_memoryerror()
884 # We cannot use assertRaises since it manually deletes the traceback
885 try:
886 inner()
887 except MemoryError as e:
888 self.assertNotEqual(wr(), None)
889 else:
890 self.fail("MemoryError not raised")
891 self.assertEqual(wr(), None)
892
Brett Cannon31f59292011-02-21 19:29:56 +0000893 @no_tracing
Antoine Pitrou07e20ef2010-10-28 22:56:58 +0000894 def test_recursion_error_cleanup(self):
895 # Same test as above, but with "recursion exceeded" errors
896 class C:
897 pass
898 wr = None
899 def inner():
900 nonlocal wr
901 c = C()
902 wr = weakref.ref(c)
903 inner()
904 # We cannot use assertRaises since it manually deletes the traceback
905 try:
906 inner()
907 except RuntimeError as e:
908 self.assertNotEqual(wr(), None)
909 else:
910 self.fail("RuntimeError not raised")
911 self.assertEqual(wr(), None)
Georg Brandl1e28a272009-12-28 08:41:01 +0000912
Antoine Pitroua7622852011-09-01 21:37:43 +0200913 def test_errno_ENOTDIR(self):
914 # Issue #12802: "not a directory" errors are ENOTDIR even on Windows
915 with self.assertRaises(OSError) as cm:
916 os.listdir(__file__)
917 self.assertEqual(cm.exception.errno, errno.ENOTDIR, cm.exception)
918
919
Brett Cannon79ec55e2012-04-12 20:24:54 -0400920class ImportErrorTests(unittest.TestCase):
921
922 def test_attributes(self):
923 # Setting 'name' and 'path' should not be a problem.
924 exc = ImportError('test')
925 self.assertIsNone(exc.name)
926 self.assertIsNone(exc.path)
927
928 exc = ImportError('test', name='somemodule')
929 self.assertEqual(exc.name, 'somemodule')
930 self.assertIsNone(exc.path)
931
932 exc = ImportError('test', path='somepath')
933 self.assertEqual(exc.path, 'somepath')
934 self.assertIsNone(exc.name)
935
936 exc = ImportError('test', path='somepath', name='somename')
937 self.assertEqual(exc.name, 'somename')
938 self.assertEqual(exc.path, 'somepath')
939
Brett Cannon07c6e712012-08-24 13:05:09 -0400940 def test_non_str_argument(self):
941 # Issue #15778
942 arg = b'abc'
943 exc = ImportError(arg)
944 self.assertEqual(str(arg), str(exc))
Brett Cannon79ec55e2012-04-12 20:24:54 -0400945
946
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000947def test_main():
Brett Cannon79ec55e2012-04-12 20:24:54 -0400948 run_unittest(ExceptionTests, ImportErrorTests)
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000949
950if __name__ == '__main__':
Guido van Rossumb8142c32007-05-08 17:49:10 +0000951 unittest.main()