]> git.proxmox.com Git - mirror_edk2.git/blame - AppPkg/Applications/Python/Python-2.7.2/Lib/test/test_sys.py
EmbeddedPkg: Extend NvVarStoreFormattedLib LIBRARY_CLASS
[mirror_edk2.git] / AppPkg / Applications / Python / Python-2.7.2 / Lib / test / test_sys.py
CommitLineData
4710c53d 1# -*- coding: iso-8859-1 -*-\r
2import unittest, test.test_support\r
3import sys, os, cStringIO\r
4import struct\r
5import operator\r
6\r
7class 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
476class 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
788def test_main():\r
789 test_classes = (SysModuleTest, SizeofTest)\r
790\r
791 test.test_support.run_unittest(*test_classes)\r
792\r
793if __name__ == "__main__":\r
794 test_main()\r