]>
Commit | Line | Data |
---|---|---|
4710c53d | 1 | # -*- coding: iso-8859-1 -*-\r |
2 | import unittest, test.test_support\r | |
3 | import sys, os, cStringIO\r | |
4 | import struct\r | |
5 | import operator\r | |
6 | \r | |
7 | class SysModuleTest(unittest.TestCase):\r | |
8 | \r | |
9 | def tearDown(self):\r | |
10 | test.test_support.reap_children()\r | |
11 | \r | |
12 | def test_original_displayhook(self):\r | |
13 | import __builtin__\r | |
14 | savestdout = sys.stdout\r | |
15 | out = cStringIO.StringIO()\r | |
16 | sys.stdout = out\r | |
17 | \r | |
18 | dh = sys.__displayhook__\r | |
19 | \r | |
20 | self.assertRaises(TypeError, dh)\r | |
21 | if hasattr(__builtin__, "_"):\r | |
22 | del __builtin__._\r | |
23 | \r | |
24 | dh(None)\r | |
25 | self.assertEqual(out.getvalue(), "")\r | |
26 | self.assertTrue(not hasattr(__builtin__, "_"))\r | |
27 | dh(42)\r | |
28 | self.assertEqual(out.getvalue(), "42\n")\r | |
29 | self.assertEqual(__builtin__._, 42)\r | |
30 | \r | |
31 | del sys.stdout\r | |
32 | self.assertRaises(RuntimeError, dh, 42)\r | |
33 | \r | |
34 | sys.stdout = savestdout\r | |
35 | \r | |
36 | def test_lost_displayhook(self):\r | |
37 | olddisplayhook = sys.displayhook\r | |
38 | del sys.displayhook\r | |
39 | code = compile("42", "<string>", "single")\r | |
40 | self.assertRaises(RuntimeError, eval, code)\r | |
41 | sys.displayhook = olddisplayhook\r | |
42 | \r | |
43 | def test_custom_displayhook(self):\r | |
44 | olddisplayhook = sys.displayhook\r | |
45 | def baddisplayhook(obj):\r | |
46 | raise ValueError\r | |
47 | sys.displayhook = baddisplayhook\r | |
48 | code = compile("42", "<string>", "single")\r | |
49 | self.assertRaises(ValueError, eval, code)\r | |
50 | sys.displayhook = olddisplayhook\r | |
51 | \r | |
52 | def test_original_excepthook(self):\r | |
53 | savestderr = sys.stderr\r | |
54 | err = cStringIO.StringIO()\r | |
55 | sys.stderr = err\r | |
56 | \r | |
57 | eh = sys.__excepthook__\r | |
58 | \r | |
59 | self.assertRaises(TypeError, eh)\r | |
60 | try:\r | |
61 | raise ValueError(42)\r | |
62 | except ValueError, exc:\r | |
63 | eh(*sys.exc_info())\r | |
64 | \r | |
65 | sys.stderr = savestderr\r | |
66 | self.assertTrue(err.getvalue().endswith("ValueError: 42\n"))\r | |
67 | \r | |
68 | # FIXME: testing the code for a lost or replaced excepthook in\r | |
69 | # Python/pythonrun.c::PyErr_PrintEx() is tricky.\r | |
70 | \r | |
71 | def test_exc_clear(self):\r | |
72 | self.assertRaises(TypeError, sys.exc_clear, 42)\r | |
73 | \r | |
74 | # Verify that exc_info is present and matches exc, then clear it, and\r | |
75 | # check that it worked.\r | |
76 | def clear_check(exc):\r | |
77 | typ, value, traceback = sys.exc_info()\r | |
78 | self.assertTrue(typ is not None)\r | |
79 | self.assertTrue(value is exc)\r | |
80 | self.assertTrue(traceback is not None)\r | |
81 | \r | |
82 | with test.test_support.check_py3k_warnings():\r | |
83 | sys.exc_clear()\r | |
84 | \r | |
85 | typ, value, traceback = sys.exc_info()\r | |
86 | self.assertTrue(typ is None)\r | |
87 | self.assertTrue(value is None)\r | |
88 | self.assertTrue(traceback is None)\r | |
89 | \r | |
90 | def clear():\r | |
91 | try:\r | |
92 | raise ValueError, 42\r | |
93 | except ValueError, exc:\r | |
94 | clear_check(exc)\r | |
95 | \r | |
96 | # Raise an exception and check that it can be cleared\r | |
97 | clear()\r | |
98 | \r | |
99 | # Verify that a frame currently handling an exception is\r | |
100 | # unaffected by calling exc_clear in a nested frame.\r | |
101 | try:\r | |
102 | raise ValueError, 13\r | |
103 | except ValueError, exc:\r | |
104 | typ1, value1, traceback1 = sys.exc_info()\r | |
105 | clear()\r | |
106 | typ2, value2, traceback2 = sys.exc_info()\r | |
107 | \r | |
108 | self.assertTrue(typ1 is typ2)\r | |
109 | self.assertTrue(value1 is exc)\r | |
110 | self.assertTrue(value1 is value2)\r | |
111 | self.assertTrue(traceback1 is traceback2)\r | |
112 | \r | |
113 | # Check that an exception can be cleared outside of an except block\r | |
114 | clear_check(exc)\r | |
115 | \r | |
116 | def test_exit(self):\r | |
117 | self.assertRaises(TypeError, sys.exit, 42, 42)\r | |
118 | \r | |
119 | # call without argument\r | |
120 | try:\r | |
121 | sys.exit(0)\r | |
122 | except SystemExit, exc:\r | |
123 | self.assertEqual(exc.code, 0)\r | |
124 | except:\r | |
125 | self.fail("wrong exception")\r | |
126 | else:\r | |
127 | self.fail("no exception")\r | |
128 | \r | |
129 | # call with tuple argument with one entry\r | |
130 | # entry will be unpacked\r | |
131 | try:\r | |
132 | sys.exit(42)\r | |
133 | except SystemExit, exc:\r | |
134 | self.assertEqual(exc.code, 42)\r | |
135 | except:\r | |
136 | self.fail("wrong exception")\r | |
137 | else:\r | |
138 | self.fail("no exception")\r | |
139 | \r | |
140 | # call with integer argument\r | |
141 | try:\r | |
142 | sys.exit((42,))\r | |
143 | except SystemExit, exc:\r | |
144 | self.assertEqual(exc.code, 42)\r | |
145 | except:\r | |
146 | self.fail("wrong exception")\r | |
147 | else:\r | |
148 | self.fail("no exception")\r | |
149 | \r | |
150 | # call with string argument\r | |
151 | try:\r | |
152 | sys.exit("exit")\r | |
153 | except SystemExit, exc:\r | |
154 | self.assertEqual(exc.code, "exit")\r | |
155 | except:\r | |
156 | self.fail("wrong exception")\r | |
157 | else:\r | |
158 | self.fail("no exception")\r | |
159 | \r | |
160 | # call with tuple argument with two entries\r | |
161 | try:\r | |
162 | sys.exit((17, 23))\r | |
163 | except SystemExit, exc:\r | |
164 | self.assertEqual(exc.code, (17, 23))\r | |
165 | except:\r | |
166 | self.fail("wrong exception")\r | |
167 | else:\r | |
168 | self.fail("no exception")\r | |
169 | \r | |
170 | # test that the exit machinery handles SystemExits properly\r | |
171 | import subprocess\r | |
172 | # both unnormalized...\r | |
173 | rc = subprocess.call([sys.executable, "-c",\r | |
174 | "raise SystemExit, 46"])\r | |
175 | self.assertEqual(rc, 46)\r | |
176 | # ... and normalized\r | |
177 | rc = subprocess.call([sys.executable, "-c",\r | |
178 | "raise SystemExit(47)"])\r | |
179 | self.assertEqual(rc, 47)\r | |
180 | \r | |
181 | def check_exit_message(code, expected, env=None):\r | |
182 | process = subprocess.Popen([sys.executable, "-c", code],\r | |
183 | stderr=subprocess.PIPE, env=env)\r | |
184 | stdout, stderr = process.communicate()\r | |
185 | self.assertEqual(process.returncode, 1)\r | |
186 | self.assertTrue(stderr.startswith(expected),\r | |
187 | "%s doesn't start with %s" % (repr(stderr), repr(expected)))\r | |
188 | \r | |
189 | # test that stderr buffer if flushed before the exit message is written\r | |
190 | # into stderr\r | |
191 | check_exit_message(\r | |
192 | r'import sys; sys.stderr.write("unflushed,"); sys.exit("message")',\r | |
193 | b"unflushed,message")\r | |
194 | \r | |
195 | # test that the unicode message is encoded to the stderr encoding\r | |
196 | env = os.environ.copy()\r | |
197 | env['PYTHONIOENCODING'] = 'latin-1'\r | |
198 | check_exit_message(\r | |
199 | r'import sys; sys.exit(u"h\xe9")',\r | |
200 | b"h\xe9", env=env)\r | |
201 | \r | |
202 | def test_getdefaultencoding(self):\r | |
203 | if test.test_support.have_unicode:\r | |
204 | self.assertRaises(TypeError, sys.getdefaultencoding, 42)\r | |
205 | # can't check more than the type, as the user might have changed it\r | |
206 | self.assertIsInstance(sys.getdefaultencoding(), str)\r | |
207 | \r | |
208 | # testing sys.settrace() is done in test_sys_settrace.py\r | |
209 | # testing sys.setprofile() is done in test_sys_setprofile.py\r | |
210 | \r | |
211 | def test_setcheckinterval(self):\r | |
212 | self.assertRaises(TypeError, sys.setcheckinterval)\r | |
213 | orig = sys.getcheckinterval()\r | |
214 | for n in 0, 100, 120, orig: # orig last to restore starting state\r | |
215 | sys.setcheckinterval(n)\r | |
216 | self.assertEqual(sys.getcheckinterval(), n)\r | |
217 | \r | |
218 | def test_recursionlimit(self):\r | |
219 | self.assertRaises(TypeError, sys.getrecursionlimit, 42)\r | |
220 | oldlimit = sys.getrecursionlimit()\r | |
221 | self.assertRaises(TypeError, sys.setrecursionlimit)\r | |
222 | self.assertRaises(ValueError, sys.setrecursionlimit, -42)\r | |
223 | sys.setrecursionlimit(10000)\r | |
224 | self.assertEqual(sys.getrecursionlimit(), 10000)\r | |
225 | sys.setrecursionlimit(oldlimit)\r | |
226 | \r | |
227 | def test_getwindowsversion(self):\r | |
228 | # Raise SkipTest if sys doesn't have getwindowsversion attribute\r | |
229 | test.test_support.get_attribute(sys, "getwindowsversion")\r | |
230 | v = sys.getwindowsversion()\r | |
231 | self.assertEqual(len(v), 5)\r | |
232 | self.assertIsInstance(v[0], int)\r | |
233 | self.assertIsInstance(v[1], int)\r | |
234 | self.assertIsInstance(v[2], int)\r | |
235 | self.assertIsInstance(v[3], int)\r | |
236 | self.assertIsInstance(v[4], str)\r | |
237 | self.assertRaises(IndexError, operator.getitem, v, 5)\r | |
238 | self.assertIsInstance(v.major, int)\r | |
239 | self.assertIsInstance(v.minor, int)\r | |
240 | self.assertIsInstance(v.build, int)\r | |
241 | self.assertIsInstance(v.platform, int)\r | |
242 | self.assertIsInstance(v.service_pack, str)\r | |
243 | self.assertIsInstance(v.service_pack_minor, int)\r | |
244 | self.assertIsInstance(v.service_pack_major, int)\r | |
245 | self.assertIsInstance(v.suite_mask, int)\r | |
246 | self.assertIsInstance(v.product_type, int)\r | |
247 | self.assertEqual(v[0], v.major)\r | |
248 | self.assertEqual(v[1], v.minor)\r | |
249 | self.assertEqual(v[2], v.build)\r | |
250 | self.assertEqual(v[3], v.platform)\r | |
251 | self.assertEqual(v[4], v.service_pack)\r | |
252 | \r | |
253 | # This is how platform.py calls it. Make sure tuple\r | |
254 | # still has 5 elements\r | |
255 | maj, min, buildno, plat, csd = sys.getwindowsversion()\r | |
256 | \r | |
257 | def test_dlopenflags(self):\r | |
258 | if hasattr(sys, "setdlopenflags"):\r | |
259 | self.assertTrue(hasattr(sys, "getdlopenflags"))\r | |
260 | self.assertRaises(TypeError, sys.getdlopenflags, 42)\r | |
261 | oldflags = sys.getdlopenflags()\r | |
262 | self.assertRaises(TypeError, sys.setdlopenflags)\r | |
263 | sys.setdlopenflags(oldflags+1)\r | |
264 | self.assertEqual(sys.getdlopenflags(), oldflags+1)\r | |
265 | sys.setdlopenflags(oldflags)\r | |
266 | \r | |
267 | def test_refcount(self):\r | |
268 | # n here must be a global in order for this test to pass while\r | |
269 | # tracing with a python function. Tracing calls PyFrame_FastToLocals\r | |
270 | # which will add a copy of any locals to the frame object, causing\r | |
271 | # the reference count to increase by 2 instead of 1.\r | |
272 | global n\r | |
273 | self.assertRaises(TypeError, sys.getrefcount)\r | |
274 | c = sys.getrefcount(None)\r | |
275 | n = None\r | |
276 | self.assertEqual(sys.getrefcount(None), c+1)\r | |
277 | del n\r | |
278 | self.assertEqual(sys.getrefcount(None), c)\r | |
279 | if hasattr(sys, "gettotalrefcount"):\r | |
280 | self.assertIsInstance(sys.gettotalrefcount(), int)\r | |
281 | \r | |
282 | def test_getframe(self):\r | |
283 | self.assertRaises(TypeError, sys._getframe, 42, 42)\r | |
284 | self.assertRaises(ValueError, sys._getframe, 2000000000)\r | |
285 | self.assertTrue(\r | |
286 | SysModuleTest.test_getframe.im_func.func_code \\r | |
287 | is sys._getframe().f_code\r | |
288 | )\r | |
289 | \r | |
290 | # sys._current_frames() is a CPython-only gimmick.\r | |
291 | def test_current_frames(self):\r | |
292 | have_threads = True\r | |
293 | try:\r | |
294 | import thread\r | |
295 | except ImportError:\r | |
296 | have_threads = False\r | |
297 | \r | |
298 | if have_threads:\r | |
299 | self.current_frames_with_threads()\r | |
300 | else:\r | |
301 | self.current_frames_without_threads()\r | |
302 | \r | |
303 | # Test sys._current_frames() in a WITH_THREADS build.\r | |
304 | @test.test_support.reap_threads\r | |
305 | def current_frames_with_threads(self):\r | |
306 | import threading, thread\r | |
307 | import traceback\r | |
308 | \r | |
309 | # Spawn a thread that blocks at a known place. Then the main\r | |
310 | # thread does sys._current_frames(), and verifies that the frames\r | |
311 | # returned make sense.\r | |
312 | entered_g = threading.Event()\r | |
313 | leave_g = threading.Event()\r | |
314 | thread_info = [] # the thread's id\r | |
315 | \r | |
316 | def f123():\r | |
317 | g456()\r | |
318 | \r | |
319 | def g456():\r | |
320 | thread_info.append(thread.get_ident())\r | |
321 | entered_g.set()\r | |
322 | leave_g.wait()\r | |
323 | \r | |
324 | t = threading.Thread(target=f123)\r | |
325 | t.start()\r | |
326 | entered_g.wait()\r | |
327 | \r | |
328 | # At this point, t has finished its entered_g.set(), although it's\r | |
329 | # impossible to guess whether it's still on that line or has moved on\r | |
330 | # to its leave_g.wait().\r | |
331 | self.assertEqual(len(thread_info), 1)\r | |
332 | thread_id = thread_info[0]\r | |
333 | \r | |
334 | d = sys._current_frames()\r | |
335 | \r | |
336 | main_id = thread.get_ident()\r | |
337 | self.assertIn(main_id, d)\r | |
338 | self.assertIn(thread_id, d)\r | |
339 | \r | |
340 | # Verify that the captured main-thread frame is _this_ frame.\r | |
341 | frame = d.pop(main_id)\r | |
342 | self.assertTrue(frame is sys._getframe())\r | |
343 | \r | |
344 | # Verify that the captured thread frame is blocked in g456, called\r | |
345 | # from f123. This is a litte tricky, since various bits of\r | |
346 | # threading.py are also in the thread's call stack.\r | |
347 | frame = d.pop(thread_id)\r | |
348 | stack = traceback.extract_stack(frame)\r | |
349 | for i, (filename, lineno, funcname, sourceline) in enumerate(stack):\r | |
350 | if funcname == "f123":\r | |
351 | break\r | |
352 | else:\r | |
353 | self.fail("didn't find f123() on thread's call stack")\r | |
354 | \r | |
355 | self.assertEqual(sourceline, "g456()")\r | |
356 | \r | |
357 | # And the next record must be for g456().\r | |
358 | filename, lineno, funcname, sourceline = stack[i+1]\r | |
359 | self.assertEqual(funcname, "g456")\r | |
360 | self.assertIn(sourceline, ["leave_g.wait()", "entered_g.set()"])\r | |
361 | \r | |
362 | # Reap the spawned thread.\r | |
363 | leave_g.set()\r | |
364 | t.join()\r | |
365 | \r | |
366 | # Test sys._current_frames() when thread support doesn't exist.\r | |
367 | def current_frames_without_threads(self):\r | |
368 | # Not much happens here: there is only one thread, with artificial\r | |
369 | # "thread id" 0.\r | |
370 | d = sys._current_frames()\r | |
371 | self.assertEqual(len(d), 1)\r | |
372 | self.assertIn(0, d)\r | |
373 | self.assertTrue(d[0] is sys._getframe())\r | |
374 | \r | |
375 | def test_attributes(self):\r | |
376 | self.assertIsInstance(sys.api_version, int)\r | |
377 | self.assertIsInstance(sys.argv, list)\r | |
378 | self.assertIn(sys.byteorder, ("little", "big"))\r | |
379 | self.assertIsInstance(sys.builtin_module_names, tuple)\r | |
380 | self.assertIsInstance(sys.copyright, basestring)\r | |
381 | self.assertIsInstance(sys.exec_prefix, basestring)\r | |
382 | self.assertIsInstance(sys.executable, basestring)\r | |
383 | self.assertEqual(len(sys.float_info), 11)\r | |
384 | self.assertEqual(sys.float_info.radix, 2)\r | |
385 | self.assertEqual(len(sys.long_info), 2)\r | |
386 | self.assertTrue(sys.long_info.bits_per_digit % 5 == 0)\r | |
387 | self.assertTrue(sys.long_info.sizeof_digit >= 1)\r | |
388 | self.assertEqual(type(sys.long_info.bits_per_digit), int)\r | |
389 | self.assertEqual(type(sys.long_info.sizeof_digit), int)\r | |
390 | self.assertIsInstance(sys.hexversion, int)\r | |
391 | self.assertIsInstance(sys.maxint, int)\r | |
392 | if test.test_support.have_unicode:\r | |
393 | self.assertIsInstance(sys.maxunicode, int)\r | |
394 | self.assertIsInstance(sys.platform, basestring)\r | |
395 | self.assertIsInstance(sys.prefix, basestring)\r | |
396 | self.assertIsInstance(sys.version, basestring)\r | |
397 | vi = sys.version_info\r | |
398 | self.assertIsInstance(vi[:], tuple)\r | |
399 | self.assertEqual(len(vi), 5)\r | |
400 | self.assertIsInstance(vi[0], int)\r | |
401 | self.assertIsInstance(vi[1], int)\r | |
402 | self.assertIsInstance(vi[2], int)\r | |
403 | self.assertIn(vi[3], ("alpha", "beta", "candidate", "final"))\r | |
404 | self.assertIsInstance(vi[4], int)\r | |
405 | self.assertIsInstance(vi.major, int)\r | |
406 | self.assertIsInstance(vi.minor, int)\r | |
407 | self.assertIsInstance(vi.micro, int)\r | |
408 | self.assertIn(vi.releaselevel, ("alpha", "beta", "candidate", "final"))\r | |
409 | self.assertIsInstance(vi.serial, int)\r | |
410 | self.assertEqual(vi[0], vi.major)\r | |
411 | self.assertEqual(vi[1], vi.minor)\r | |
412 | self.assertEqual(vi[2], vi.micro)\r | |
413 | self.assertEqual(vi[3], vi.releaselevel)\r | |
414 | self.assertEqual(vi[4], vi.serial)\r | |
415 | self.assertTrue(vi > (1,0,0))\r | |
416 | self.assertIsInstance(sys.float_repr_style, str)\r | |
417 | self.assertIn(sys.float_repr_style, ('short', 'legacy'))\r | |
418 | \r | |
419 | def test_43581(self):\r | |
420 | # Can't use sys.stdout, as this is a cStringIO object when\r | |
421 | # the test runs under regrtest.\r | |
422 | self.assertTrue(sys.__stdout__.encoding == sys.__stderr__.encoding)\r | |
423 | \r | |
424 | def test_sys_flags(self):\r | |
425 | self.assertTrue(sys.flags)\r | |
426 | attrs = ("debug", "py3k_warning", "division_warning", "division_new",\r | |
427 | "inspect", "interactive", "optimize", "dont_write_bytecode",\r | |
428 | "no_site", "ignore_environment", "tabcheck", "verbose",\r | |
429 | "unicode", "bytes_warning")\r | |
430 | for attr in attrs:\r | |
431 | self.assertTrue(hasattr(sys.flags, attr), attr)\r | |
432 | self.assertEqual(type(getattr(sys.flags, attr)), int, attr)\r | |
433 | self.assertTrue(repr(sys.flags))\r | |
434 | \r | |
435 | def test_clear_type_cache(self):\r | |
436 | sys._clear_type_cache()\r | |
437 | \r | |
438 | def test_ioencoding(self):\r | |
439 | import subprocess\r | |
440 | env = dict(os.environ)\r | |
441 | \r | |
442 | # Test character: cent sign, encoded as 0x4A (ASCII J) in CP424,\r | |
443 | # not representable in ASCII.\r | |
444 | \r | |
445 | env["PYTHONIOENCODING"] = "cp424"\r | |
446 | p = subprocess.Popen([sys.executable, "-c", 'print unichr(0xa2)'],\r | |
447 | stdout = subprocess.PIPE, env=env)\r | |
448 | out = p.communicate()[0].strip()\r | |
449 | self.assertEqual(out, unichr(0xa2).encode("cp424"))\r | |
450 | \r | |
451 | env["PYTHONIOENCODING"] = "ascii:replace"\r | |
452 | p = subprocess.Popen([sys.executable, "-c", 'print unichr(0xa2)'],\r | |
453 | stdout = subprocess.PIPE, env=env)\r | |
454 | out = p.communicate()[0].strip()\r | |
455 | self.assertEqual(out, '?')\r | |
456 | \r | |
457 | def test_call_tracing(self):\r | |
458 | self.assertEqual(sys.call_tracing(str, (2,)), "2")\r | |
459 | self.assertRaises(TypeError, sys.call_tracing, str, 2)\r | |
460 | \r | |
461 | def test_executable(self):\r | |
462 | # Issue #7774: Ensure that sys.executable is an empty string if argv[0]\r | |
463 | # has been set to an non existent program name and Python is unable to\r | |
464 | # retrieve the real program name\r | |
465 | import subprocess\r | |
466 | # For a normal installation, it should work without 'cwd'\r | |
467 | # argument. For test runs in the build directory, see #7774.\r | |
468 | python_dir = os.path.dirname(os.path.realpath(sys.executable))\r | |
469 | p = subprocess.Popen(\r | |
470 | ["nonexistent", "-c", 'import sys; print repr(sys.executable)'],\r | |
471 | executable=sys.executable, stdout=subprocess.PIPE, cwd=python_dir)\r | |
472 | executable = p.communicate()[0].strip()\r | |
473 | p.wait()\r | |
474 | self.assertIn(executable, ["''", repr(sys.executable)])\r | |
475 | \r | |
476 | class SizeofTest(unittest.TestCase):\r | |
477 | \r | |
478 | TPFLAGS_HAVE_GC = 1<<14\r | |
479 | TPFLAGS_HEAPTYPE = 1L<<9\r | |
480 | \r | |
481 | def setUp(self):\r | |
482 | self.c = len(struct.pack('c', ' '))\r | |
483 | self.H = len(struct.pack('H', 0))\r | |
484 | self.i = len(struct.pack('i', 0))\r | |
485 | self.l = len(struct.pack('l', 0))\r | |
486 | self.P = len(struct.pack('P', 0))\r | |
487 | # due to missing size_t information from struct, it is assumed that\r | |
488 | # sizeof(Py_ssize_t) = sizeof(void*)\r | |
489 | self.header = 'PP'\r | |
490 | self.vheader = self.header + 'P'\r | |
491 | if hasattr(sys, "gettotalrefcount"):\r | |
492 | self.header += '2P'\r | |
493 | self.vheader += '2P'\r | |
494 | self.longdigit = sys.long_info.sizeof_digit\r | |
495 | import _testcapi\r | |
496 | self.gc_headsize = _testcapi.SIZEOF_PYGC_HEAD\r | |
497 | self.file = open(test.test_support.TESTFN, 'wb')\r | |
498 | \r | |
499 | def tearDown(self):\r | |
500 | self.file.close()\r | |
501 | test.test_support.unlink(test.test_support.TESTFN)\r | |
502 | \r | |
503 | def check_sizeof(self, o, size):\r | |
504 | result = sys.getsizeof(o)\r | |
505 | if ((type(o) == type) and (o.__flags__ & self.TPFLAGS_HEAPTYPE) or\\r | |
506 | ((type(o) != type) and (type(o).__flags__ & self.TPFLAGS_HAVE_GC))):\r | |
507 | size += self.gc_headsize\r | |
508 | msg = 'wrong size for %s: got %d, expected %d' \\r | |
509 | % (type(o), result, size)\r | |
510 | self.assertEqual(result, size, msg)\r | |
511 | \r | |
512 | def calcsize(self, fmt):\r | |
513 | """Wrapper around struct.calcsize which enforces the alignment of the\r | |
514 | end of a structure to the alignment requirement of pointer.\r | |
515 | \r | |
516 | Note: This wrapper should only be used if a pointer member is included\r | |
517 | and no member with a size larger than a pointer exists.\r | |
518 | """\r | |
519 | return struct.calcsize(fmt + '0P')\r | |
520 | \r | |
521 | def test_gc_head_size(self):\r | |
522 | # Check that the gc header size is added to objects tracked by the gc.\r | |
523 | h = self.header\r | |
524 | size = self.calcsize\r | |
525 | gc_header_size = self.gc_headsize\r | |
526 | # bool objects are not gc tracked\r | |
527 | self.assertEqual(sys.getsizeof(True), size(h + 'l'))\r | |
528 | # but lists are\r | |
529 | self.assertEqual(sys.getsizeof([]), size(h + 'P PP') + gc_header_size)\r | |
530 | \r | |
531 | def test_default(self):\r | |
532 | h = self.header\r | |
533 | size = self.calcsize\r | |
534 | self.assertEqual(sys.getsizeof(True, -1), size(h + 'l'))\r | |
535 | \r | |
536 | def test_objecttypes(self):\r | |
537 | # check all types defined in Objects/\r | |
538 | h = self.header\r | |
539 | vh = self.vheader\r | |
540 | size = self.calcsize\r | |
541 | check = self.check_sizeof\r | |
542 | # bool\r | |
543 | check(True, size(h + 'l'))\r | |
544 | # buffer\r | |
545 | with test.test_support.check_py3k_warnings():\r | |
546 | check(buffer(''), size(h + '2P2Pil'))\r | |
547 | # builtin_function_or_method\r | |
548 | check(len, size(h + '3P'))\r | |
549 | # bytearray\r | |
550 | samples = ['', 'u'*100000]\r | |
551 | for sample in samples:\r | |
552 | x = bytearray(sample)\r | |
553 | check(x, size(vh + 'iPP') + x.__alloc__() * self.c)\r | |
554 | # bytearray_iterator\r | |
555 | check(iter(bytearray()), size(h + 'PP'))\r | |
556 | # cell\r | |
557 | def get_cell():\r | |
558 | x = 42\r | |
559 | def inner():\r | |
560 | return x\r | |
561 | return inner\r | |
562 | check(get_cell().func_closure[0], size(h + 'P'))\r | |
563 | # classobj (old-style class)\r | |
564 | class class_oldstyle():\r | |
565 | def method():\r | |
566 | pass\r | |
567 | check(class_oldstyle, size(h + '7P'))\r | |
568 | # instance (old-style class)\r | |
569 | check(class_oldstyle(), size(h + '3P'))\r | |
570 | # instancemethod (old-style class)\r | |
571 | check(class_oldstyle().method, size(h + '4P'))\r | |
572 | # complex\r | |
573 | check(complex(0,1), size(h + '2d'))\r | |
574 | # code\r | |
575 | check(get_cell().func_code, size(h + '4i8Pi3P'))\r | |
576 | # BaseException\r | |
577 | check(BaseException(), size(h + '3P'))\r | |
578 | # UnicodeEncodeError\r | |
579 | check(UnicodeEncodeError("", u"", 0, 0, ""), size(h + '5P2PP'))\r | |
580 | # UnicodeDecodeError\r | |
581 | check(UnicodeDecodeError("", "", 0, 0, ""), size(h + '5P2PP'))\r | |
582 | # UnicodeTranslateError\r | |
583 | check(UnicodeTranslateError(u"", 0, 1, ""), size(h + '5P2PP'))\r | |
584 | # method_descriptor (descriptor object)\r | |
585 | check(str.lower, size(h + '2PP'))\r | |
586 | # classmethod_descriptor (descriptor object)\r | |
587 | # XXX\r | |
588 | # member_descriptor (descriptor object)\r | |
589 | import datetime\r | |
590 | check(datetime.timedelta.days, size(h + '2PP'))\r | |
591 | # getset_descriptor (descriptor object)\r | |
592 | import __builtin__\r | |
593 | check(__builtin__.file.closed, size(h + '2PP'))\r | |
594 | # wrapper_descriptor (descriptor object)\r | |
595 | check(int.__add__, size(h + '2P2P'))\r | |
596 | # dictproxy\r | |
597 | class C(object): pass\r | |
598 | check(C.__dict__, size(h + 'P'))\r | |
599 | # method-wrapper (descriptor object)\r | |
600 | check({}.__iter__, size(h + '2P'))\r | |
601 | # dict\r | |
602 | check({}, size(h + '3P2P' + 8*'P2P'))\r | |
603 | x = {1:1, 2:2, 3:3, 4:4, 5:5, 6:6, 7:7, 8:8}\r | |
604 | check(x, size(h + '3P2P' + 8*'P2P') + 16*size('P2P'))\r | |
605 | # dictionary-keyiterator\r | |
606 | check({}.iterkeys(), size(h + 'P2PPP'))\r | |
607 | # dictionary-valueiterator\r | |
608 | check({}.itervalues(), size(h + 'P2PPP'))\r | |
609 | # dictionary-itemiterator\r | |
610 | check({}.iteritems(), size(h + 'P2PPP'))\r | |
611 | # ellipses\r | |
612 | check(Ellipsis, size(h + ''))\r | |
613 | # EncodingMap\r | |
614 | import codecs, encodings.iso8859_3\r | |
615 | x = codecs.charmap_build(encodings.iso8859_3.decoding_table)\r | |
616 | check(x, size(h + '32B2iB'))\r | |
617 | # enumerate\r | |
618 | check(enumerate([]), size(h + 'l3P'))\r | |
619 | # file\r | |
620 | check(self.file, size(h + '4P2i4P3i3P3i'))\r | |
621 | # float\r | |
622 | check(float(0), size(h + 'd'))\r | |
623 | # sys.floatinfo\r | |
624 | check(sys.float_info, size(vh) + self.P * len(sys.float_info))\r | |
625 | # frame\r | |
626 | import inspect\r | |
627 | CO_MAXBLOCKS = 20\r | |
628 | x = inspect.currentframe()\r | |
629 | ncells = len(x.f_code.co_cellvars)\r | |
630 | nfrees = len(x.f_code.co_freevars)\r | |
631 | extras = x.f_code.co_stacksize + x.f_code.co_nlocals +\\r | |
632 | ncells + nfrees - 1\r | |
633 | check(x, size(vh + '12P3i' + CO_MAXBLOCKS*'3i' + 'P' + extras*'P'))\r | |
634 | # function\r | |
635 | def func(): pass\r | |
636 | check(func, size(h + '9P'))\r | |
637 | class c():\r | |
638 | @staticmethod\r | |
639 | def foo():\r | |
640 | pass\r | |
641 | @classmethod\r | |
642 | def bar(cls):\r | |
643 | pass\r | |
644 | # staticmethod\r | |
645 | check(foo, size(h + 'P'))\r | |
646 | # classmethod\r | |
647 | check(bar, size(h + 'P'))\r | |
648 | # generator\r | |
649 | def get_gen(): yield 1\r | |
650 | check(get_gen(), size(h + 'Pi2P'))\r | |
651 | # integer\r | |
652 | check(1, size(h + 'l'))\r | |
653 | check(100, size(h + 'l'))\r | |
654 | # iterator\r | |
655 | check(iter('abc'), size(h + 'lP'))\r | |
656 | # callable-iterator\r | |
657 | import re\r | |
658 | check(re.finditer('',''), size(h + '2P'))\r | |
659 | # list\r | |
660 | samples = [[], [1,2,3], ['1', '2', '3']]\r | |
661 | for sample in samples:\r | |
662 | check(sample, size(vh + 'PP') + len(sample)*self.P)\r | |
663 | # sortwrapper (list)\r | |
664 | # XXX\r | |
665 | # cmpwrapper (list)\r | |
666 | # XXX\r | |
667 | # listiterator (list)\r | |
668 | check(iter([]), size(h + 'lP'))\r | |
669 | # listreverseiterator (list)\r | |
670 | check(reversed([]), size(h + 'lP'))\r | |
671 | # long\r | |
672 | check(0L, size(vh))\r | |
673 | check(1L, size(vh) + self.longdigit)\r | |
674 | check(-1L, size(vh) + self.longdigit)\r | |
675 | PyLong_BASE = 2**sys.long_info.bits_per_digit\r | |
676 | check(long(PyLong_BASE), size(vh) + 2*self.longdigit)\r | |
677 | check(long(PyLong_BASE**2-1), size(vh) + 2*self.longdigit)\r | |
678 | check(long(PyLong_BASE**2), size(vh) + 3*self.longdigit)\r | |
679 | # module\r | |
680 | check(unittest, size(h + 'P'))\r | |
681 | # None\r | |
682 | check(None, size(h + ''))\r | |
683 | # object\r | |
684 | check(object(), size(h + ''))\r | |
685 | # property (descriptor object)\r | |
686 | class C(object):\r | |
687 | def getx(self): return self.__x\r | |
688 | def setx(self, value): self.__x = value\r | |
689 | def delx(self): del self.__x\r | |
690 | x = property(getx, setx, delx, "")\r | |
691 | check(x, size(h + '4Pi'))\r | |
692 | # PyCObject\r | |
693 | # PyCapsule\r | |
694 | # XXX\r | |
695 | # rangeiterator\r | |
696 | check(iter(xrange(1)), size(h + '4l'))\r | |
697 | # reverse\r | |
698 | check(reversed(''), size(h + 'PP'))\r | |
699 | # set\r | |
700 | # frozenset\r | |
701 | PySet_MINSIZE = 8\r | |
702 | samples = [[], range(10), range(50)]\r | |
703 | s = size(h + '3P2P' + PySet_MINSIZE*'lP' + 'lP')\r | |
704 | for sample in samples:\r | |
705 | minused = len(sample)\r | |
706 | if minused == 0: tmp = 1\r | |
707 | # the computation of minused is actually a bit more complicated\r | |
708 | # but this suffices for the sizeof test\r | |
709 | minused = minused*2\r | |
710 | newsize = PySet_MINSIZE\r | |
711 | while newsize <= minused:\r | |
712 | newsize = newsize << 1\r | |
713 | if newsize <= 8:\r | |
714 | check(set(sample), s)\r | |
715 | check(frozenset(sample), s)\r | |
716 | else:\r | |
717 | check(set(sample), s + newsize*struct.calcsize('lP'))\r | |
718 | check(frozenset(sample), s + newsize*struct.calcsize('lP'))\r | |
719 | # setiterator\r | |
720 | check(iter(set()), size(h + 'P3P'))\r | |
721 | # slice\r | |
722 | check(slice(1), size(h + '3P'))\r | |
723 | # str\r | |
724 | check('', struct.calcsize(vh + 'li') + 1)\r | |
725 | check('abc', struct.calcsize(vh + 'li') + 1 + 3*self.c)\r | |
726 | # super\r | |
727 | check(super(int), size(h + '3P'))\r | |
728 | # tuple\r | |
729 | check((), size(vh))\r | |
730 | check((1,2,3), size(vh) + 3*self.P)\r | |
731 | # tupleiterator\r | |
732 | check(iter(()), size(h + 'lP'))\r | |
733 | # type\r | |
734 | # (PyTypeObject + PyNumberMethods + PyMappingMethods +\r | |
735 | # PySequenceMethods + PyBufferProcs)\r | |
736 | s = size(vh + 'P2P15Pl4PP9PP11PI') + size('41P 10P 3P 6P')\r | |
737 | class newstyleclass(object):\r | |
738 | pass\r | |
739 | check(newstyleclass, s)\r | |
740 | # builtin type\r | |
741 | check(int, s)\r | |
742 | # NotImplementedType\r | |
743 | import types\r | |
744 | check(types.NotImplementedType, s)\r | |
745 | # unicode\r | |
746 | usize = len(u'\0'.encode('unicode-internal'))\r | |
747 | samples = [u'', u'1'*100]\r | |
748 | # we need to test for both sizes, because we don't know if the string\r | |
749 | # has been cached\r | |
750 | for s in samples:\r | |
751 | check(s, size(h + 'PPlP') + usize * (len(s) + 1))\r | |
752 | # weakref\r | |
753 | import weakref\r | |
754 | check(weakref.ref(int), size(h + '2Pl2P'))\r | |
755 | # weakproxy\r | |
756 | # XXX\r | |
757 | # weakcallableproxy\r | |
758 | check(weakref.proxy(int), size(h + '2Pl2P'))\r | |
759 | # xrange\r | |
760 | check(xrange(1), size(h + '3l'))\r | |
761 | check(xrange(66000), size(h + '3l'))\r | |
762 | \r | |
763 | def test_pythontypes(self):\r | |
764 | # check all types defined in Python/\r | |
765 | h = self.header\r | |
766 | vh = self.vheader\r | |
767 | size = self.calcsize\r | |
768 | check = self.check_sizeof\r | |
769 | # _ast.AST\r | |
770 | import _ast\r | |
771 | check(_ast.AST(), size(h + ''))\r | |
772 | # imp.NullImporter\r | |
773 | import imp\r | |
774 | check(imp.NullImporter(self.file.name), size(h + ''))\r | |
775 | try:\r | |
776 | raise TypeError\r | |
777 | except TypeError:\r | |
778 | tb = sys.exc_info()[2]\r | |
779 | # traceback\r | |
780 | if tb != None:\r | |
781 | check(tb, size(h + '2P2i'))\r | |
782 | # symtable entry\r | |
783 | # XXX\r | |
784 | # sys.flags\r | |
785 | check(sys.flags, size(vh) + self.P * len(sys.flags))\r | |
786 | \r | |
787 | \r | |
788 | def test_main():\r | |
789 | test_classes = (SysModuleTest, SizeofTest)\r | |
790 | \r | |
791 | test.test_support.run_unittest(*test_classes)\r | |
792 | \r | |
793 | if __name__ == "__main__":\r | |
794 | test_main()\r |