14710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm# -*- coding: iso-8859-1 -*- 24710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylmimport unittest, test.test_support 34710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylmimport sys, os, cStringIO 44710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylmimport struct 54710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylmimport operator 64710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 74710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylmclass SysModuleTest(unittest.TestCase): 84710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 94710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def tearDown(self): 104710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm test.test_support.reap_children() 114710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 124710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def test_original_displayhook(self): 134710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm import __builtin__ 144710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm savestdout = sys.stdout 154710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm out = cStringIO.StringIO() 164710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm sys.stdout = out 174710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 184710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm dh = sys.__displayhook__ 194710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 204710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertRaises(TypeError, dh) 214710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm if hasattr(__builtin__, "_"): 224710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm del __builtin__._ 234710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 244710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm dh(None) 254710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(out.getvalue(), "") 264710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertTrue(not hasattr(__builtin__, "_")) 274710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm dh(42) 284710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(out.getvalue(), "42\n") 294710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(__builtin__._, 42) 304710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 314710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm del sys.stdout 324710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertRaises(RuntimeError, dh, 42) 334710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 344710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm sys.stdout = savestdout 354710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 364710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def test_lost_displayhook(self): 374710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm olddisplayhook = sys.displayhook 384710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm del sys.displayhook 394710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm code = compile("42", "<string>", "single") 404710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertRaises(RuntimeError, eval, code) 414710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm sys.displayhook = olddisplayhook 424710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 434710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def test_custom_displayhook(self): 444710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm olddisplayhook = sys.displayhook 454710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def baddisplayhook(obj): 464710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm raise ValueError 474710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm sys.displayhook = baddisplayhook 484710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm code = compile("42", "<string>", "single") 494710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertRaises(ValueError, eval, code) 504710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm sys.displayhook = olddisplayhook 514710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 524710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def test_original_excepthook(self): 534710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm savestderr = sys.stderr 544710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm err = cStringIO.StringIO() 554710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm sys.stderr = err 564710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 574710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm eh = sys.__excepthook__ 584710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 594710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertRaises(TypeError, eh) 604710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm try: 614710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm raise ValueError(42) 624710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm except ValueError, exc: 634710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm eh(*sys.exc_info()) 644710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 654710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm sys.stderr = savestderr 664710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertTrue(err.getvalue().endswith("ValueError: 42\n")) 674710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 684710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # FIXME: testing the code for a lost or replaced excepthook in 694710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # Python/pythonrun.c::PyErr_PrintEx() is tricky. 704710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 714710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def test_exc_clear(self): 724710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertRaises(TypeError, sys.exc_clear, 42) 734710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 744710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # Verify that exc_info is present and matches exc, then clear it, and 754710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # check that it worked. 764710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def clear_check(exc): 774710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm typ, value, traceback = sys.exc_info() 784710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertTrue(typ is not None) 794710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertTrue(value is exc) 804710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertTrue(traceback is not None) 814710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 824710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm with test.test_support.check_py3k_warnings(): 834710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm sys.exc_clear() 844710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 854710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm typ, value, traceback = sys.exc_info() 864710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertTrue(typ is None) 874710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertTrue(value is None) 884710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertTrue(traceback is None) 894710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 904710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def clear(): 914710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm try: 924710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm raise ValueError, 42 934710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm except ValueError, exc: 944710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm clear_check(exc) 954710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 964710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # Raise an exception and check that it can be cleared 974710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm clear() 984710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 994710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # Verify that a frame currently handling an exception is 1004710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # unaffected by calling exc_clear in a nested frame. 1014710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm try: 1024710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm raise ValueError, 13 1034710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm except ValueError, exc: 1044710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm typ1, value1, traceback1 = sys.exc_info() 1054710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm clear() 1064710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm typ2, value2, traceback2 = sys.exc_info() 1074710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 1084710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertTrue(typ1 is typ2) 1094710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertTrue(value1 is exc) 1104710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertTrue(value1 is value2) 1114710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertTrue(traceback1 is traceback2) 1124710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 1134710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # Check that an exception can be cleared outside of an except block 1144710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm clear_check(exc) 1154710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 1164710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def test_exit(self): 1174710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertRaises(TypeError, sys.exit, 42, 42) 1184710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 1194710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # call without argument 1204710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm try: 1214710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm sys.exit(0) 1224710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm except SystemExit, exc: 1234710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(exc.code, 0) 1244710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm except: 1254710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.fail("wrong exception") 1264710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm else: 1274710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.fail("no exception") 1284710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 1294710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # call with tuple argument with one entry 1304710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # entry will be unpacked 1314710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm try: 1324710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm sys.exit(42) 1334710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm except SystemExit, exc: 1344710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(exc.code, 42) 1354710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm except: 1364710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.fail("wrong exception") 1374710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm else: 1384710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.fail("no exception") 1394710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 1404710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # call with integer argument 1414710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm try: 1424710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm sys.exit((42,)) 1434710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm except SystemExit, exc: 1444710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(exc.code, 42) 1454710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm except: 1464710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.fail("wrong exception") 1474710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm else: 1484710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.fail("no exception") 1494710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 1504710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # call with string argument 1514710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm try: 1524710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm sys.exit("exit") 1534710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm except SystemExit, exc: 1544710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(exc.code, "exit") 1554710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm except: 1564710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.fail("wrong exception") 1574710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm else: 1584710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.fail("no exception") 1594710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 1604710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # call with tuple argument with two entries 1614710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm try: 1624710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm sys.exit((17, 23)) 1634710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm except SystemExit, exc: 1644710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(exc.code, (17, 23)) 1654710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm except: 1664710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.fail("wrong exception") 1674710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm else: 1684710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.fail("no exception") 1694710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 1704710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # test that the exit machinery handles SystemExits properly 1714710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm import subprocess 1724710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # both unnormalized... 1734710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm rc = subprocess.call([sys.executable, "-c", 1744710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm "raise SystemExit, 46"]) 1754710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(rc, 46) 1764710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # ... and normalized 1774710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm rc = subprocess.call([sys.executable, "-c", 1784710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm "raise SystemExit(47)"]) 1794710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(rc, 47) 1804710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 1814710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def check_exit_message(code, expected, env=None): 1824710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm process = subprocess.Popen([sys.executable, "-c", code], 1834710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm stderr=subprocess.PIPE, env=env) 1844710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm stdout, stderr = process.communicate() 1854710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(process.returncode, 1) 1864710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertTrue(stderr.startswith(expected), 1874710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm "%s doesn't start with %s" % (repr(stderr), repr(expected))) 1884710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 1894710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # test that stderr buffer if flushed before the exit message is written 1904710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # into stderr 1914710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check_exit_message( 1924710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm r'import sys; sys.stderr.write("unflushed,"); sys.exit("message")', 1934710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm b"unflushed,message") 1944710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 1954710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # test that the unicode message is encoded to the stderr encoding 1964710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm env = os.environ.copy() 1974710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm env['PYTHONIOENCODING'] = 'latin-1' 1984710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check_exit_message( 1994710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm r'import sys; sys.exit(u"h\xe9")', 2004710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm b"h\xe9", env=env) 2014710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 2024710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def test_getdefaultencoding(self): 2034710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm if test.test_support.have_unicode: 2044710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertRaises(TypeError, sys.getdefaultencoding, 42) 2054710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # can't check more than the type, as the user might have changed it 2064710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIsInstance(sys.getdefaultencoding(), str) 2074710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 2084710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # testing sys.settrace() is done in test_sys_settrace.py 2094710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # testing sys.setprofile() is done in test_sys_setprofile.py 2104710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 2114710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def test_setcheckinterval(self): 2124710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertRaises(TypeError, sys.setcheckinterval) 2134710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm orig = sys.getcheckinterval() 2144710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm for n in 0, 100, 120, orig: # orig last to restore starting state 2154710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm sys.setcheckinterval(n) 2164710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(sys.getcheckinterval(), n) 2174710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 2184710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def test_recursionlimit(self): 2194710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertRaises(TypeError, sys.getrecursionlimit, 42) 2204710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm oldlimit = sys.getrecursionlimit() 2214710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertRaises(TypeError, sys.setrecursionlimit) 2224710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertRaises(ValueError, sys.setrecursionlimit, -42) 2234710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm sys.setrecursionlimit(10000) 2244710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(sys.getrecursionlimit(), 10000) 2254710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm sys.setrecursionlimit(oldlimit) 2264710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 2274710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def test_getwindowsversion(self): 2284710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # Raise SkipTest if sys doesn't have getwindowsversion attribute 2294710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm test.test_support.get_attribute(sys, "getwindowsversion") 2304710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm v = sys.getwindowsversion() 2314710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(len(v), 5) 2324710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIsInstance(v[0], int) 2334710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIsInstance(v[1], int) 2344710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIsInstance(v[2], int) 2354710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIsInstance(v[3], int) 2364710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIsInstance(v[4], str) 2374710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertRaises(IndexError, operator.getitem, v, 5) 2384710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIsInstance(v.major, int) 2394710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIsInstance(v.minor, int) 2404710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIsInstance(v.build, int) 2414710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIsInstance(v.platform, int) 2424710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIsInstance(v.service_pack, str) 2434710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIsInstance(v.service_pack_minor, int) 2444710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIsInstance(v.service_pack_major, int) 2454710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIsInstance(v.suite_mask, int) 2464710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIsInstance(v.product_type, int) 2474710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(v[0], v.major) 2484710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(v[1], v.minor) 2494710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(v[2], v.build) 2504710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(v[3], v.platform) 2514710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(v[4], v.service_pack) 2524710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 2534710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # This is how platform.py calls it. Make sure tuple 2544710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # still has 5 elements 2554710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm maj, min, buildno, plat, csd = sys.getwindowsversion() 2564710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 2574710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def test_dlopenflags(self): 2584710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm if hasattr(sys, "setdlopenflags"): 2594710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertTrue(hasattr(sys, "getdlopenflags")) 2604710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertRaises(TypeError, sys.getdlopenflags, 42) 2614710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm oldflags = sys.getdlopenflags() 2624710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertRaises(TypeError, sys.setdlopenflags) 2634710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm sys.setdlopenflags(oldflags+1) 2644710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(sys.getdlopenflags(), oldflags+1) 2654710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm sys.setdlopenflags(oldflags) 2664710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 2674710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def test_refcount(self): 2684710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # n here must be a global in order for this test to pass while 2694710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # tracing with a python function. Tracing calls PyFrame_FastToLocals 2704710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # which will add a copy of any locals to the frame object, causing 2714710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # the reference count to increase by 2 instead of 1. 2724710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm global n 2734710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertRaises(TypeError, sys.getrefcount) 2744710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm c = sys.getrefcount(None) 2754710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm n = None 2764710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(sys.getrefcount(None), c+1) 2774710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm del n 2784710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(sys.getrefcount(None), c) 2794710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm if hasattr(sys, "gettotalrefcount"): 2804710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIsInstance(sys.gettotalrefcount(), int) 2814710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 2824710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def test_getframe(self): 2834710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertRaises(TypeError, sys._getframe, 42, 42) 2844710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertRaises(ValueError, sys._getframe, 2000000000) 2854710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertTrue( 2864710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm SysModuleTest.test_getframe.im_func.func_code \ 2874710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm is sys._getframe().f_code 2884710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm ) 2894710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 2904710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # sys._current_frames() is a CPython-only gimmick. 2914710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def test_current_frames(self): 2924710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm have_threads = True 2934710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm try: 2944710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm import thread 2954710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm except ImportError: 2964710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm have_threads = False 2974710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 2984710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm if have_threads: 2994710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.current_frames_with_threads() 3004710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm else: 3014710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.current_frames_without_threads() 3024710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 3034710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # Test sys._current_frames() in a WITH_THREADS build. 3044710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm @test.test_support.reap_threads 3054710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def current_frames_with_threads(self): 3064710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm import threading, thread 3074710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm import traceback 3084710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 3094710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # Spawn a thread that blocks at a known place. Then the main 3104710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # thread does sys._current_frames(), and verifies that the frames 3114710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # returned make sense. 3124710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm entered_g = threading.Event() 3134710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm leave_g = threading.Event() 3144710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm thread_info = [] # the thread's id 3154710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 3164710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def f123(): 3174710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm g456() 3184710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 3194710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def g456(): 3204710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm thread_info.append(thread.get_ident()) 3214710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm entered_g.set() 3224710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm leave_g.wait() 3234710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 3244710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm t = threading.Thread(target=f123) 3254710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm t.start() 3264710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm entered_g.wait() 3274710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 3284710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # At this point, t has finished its entered_g.set(), although it's 3294710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # impossible to guess whether it's still on that line or has moved on 3304710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # to its leave_g.wait(). 3314710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(len(thread_info), 1) 3324710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm thread_id = thread_info[0] 3334710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 3344710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm d = sys._current_frames() 3354710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 3364710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm main_id = thread.get_ident() 3374710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIn(main_id, d) 3384710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIn(thread_id, d) 3394710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 3404710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # Verify that the captured main-thread frame is _this_ frame. 3414710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm frame = d.pop(main_id) 3424710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertTrue(frame is sys._getframe()) 3434710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 3444710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # Verify that the captured thread frame is blocked in g456, called 3454710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # from f123. This is a litte tricky, since various bits of 3464710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # threading.py are also in the thread's call stack. 3474710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm frame = d.pop(thread_id) 3484710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm stack = traceback.extract_stack(frame) 3494710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm for i, (filename, lineno, funcname, sourceline) in enumerate(stack): 3504710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm if funcname == "f123": 3514710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm break 3524710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm else: 3534710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.fail("didn't find f123() on thread's call stack") 3544710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 3554710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(sourceline, "g456()") 3564710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 3574710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # And the next record must be for g456(). 3584710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm filename, lineno, funcname, sourceline = stack[i+1] 3594710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(funcname, "g456") 3604710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIn(sourceline, ["leave_g.wait()", "entered_g.set()"]) 3614710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 3624710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # Reap the spawned thread. 3634710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm leave_g.set() 3644710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm t.join() 3654710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 3664710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # Test sys._current_frames() when thread support doesn't exist. 3674710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def current_frames_without_threads(self): 3684710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # Not much happens here: there is only one thread, with artificial 3694710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # "thread id" 0. 3704710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm d = sys._current_frames() 3714710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(len(d), 1) 3724710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIn(0, d) 3734710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertTrue(d[0] is sys._getframe()) 3744710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 3754710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def test_attributes(self): 3764710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIsInstance(sys.api_version, int) 3774710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIsInstance(sys.argv, list) 3784710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIn(sys.byteorder, ("little", "big")) 3794710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIsInstance(sys.builtin_module_names, tuple) 3804710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIsInstance(sys.copyright, basestring) 3814710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIsInstance(sys.exec_prefix, basestring) 3824710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIsInstance(sys.executable, basestring) 3834710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(len(sys.float_info), 11) 3844710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(sys.float_info.radix, 2) 3854710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(len(sys.long_info), 2) 3864710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertTrue(sys.long_info.bits_per_digit % 5 == 0) 3874710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertTrue(sys.long_info.sizeof_digit >= 1) 3884710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(type(sys.long_info.bits_per_digit), int) 3894710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(type(sys.long_info.sizeof_digit), int) 3904710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIsInstance(sys.hexversion, int) 3914710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIsInstance(sys.maxint, int) 3924710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm if test.test_support.have_unicode: 3934710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIsInstance(sys.maxunicode, int) 3944710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIsInstance(sys.platform, basestring) 3954710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIsInstance(sys.prefix, basestring) 3964710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIsInstance(sys.version, basestring) 3974710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm vi = sys.version_info 3984710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIsInstance(vi[:], tuple) 3994710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(len(vi), 5) 4004710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIsInstance(vi[0], int) 4014710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIsInstance(vi[1], int) 4024710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIsInstance(vi[2], int) 4034710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIn(vi[3], ("alpha", "beta", "candidate", "final")) 4044710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIsInstance(vi[4], int) 4054710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIsInstance(vi.major, int) 4064710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIsInstance(vi.minor, int) 4074710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIsInstance(vi.micro, int) 4084710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIn(vi.releaselevel, ("alpha", "beta", "candidate", "final")) 4094710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIsInstance(vi.serial, int) 4104710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(vi[0], vi.major) 4114710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(vi[1], vi.minor) 4124710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(vi[2], vi.micro) 4134710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(vi[3], vi.releaselevel) 4144710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(vi[4], vi.serial) 4154710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertTrue(vi > (1,0,0)) 4164710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIsInstance(sys.float_repr_style, str) 4174710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIn(sys.float_repr_style, ('short', 'legacy')) 4184710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 4194710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def test_43581(self): 4204710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # Can't use sys.stdout, as this is a cStringIO object when 4214710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # the test runs under regrtest. 4224710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertTrue(sys.__stdout__.encoding == sys.__stderr__.encoding) 4234710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 4244710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def test_sys_flags(self): 4254710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertTrue(sys.flags) 4264710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm attrs = ("debug", "py3k_warning", "division_warning", "division_new", 4274710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm "inspect", "interactive", "optimize", "dont_write_bytecode", 4284710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm "no_site", "ignore_environment", "tabcheck", "verbose", 4294710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm "unicode", "bytes_warning") 4304710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm for attr in attrs: 4314710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertTrue(hasattr(sys.flags, attr), attr) 4324710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(type(getattr(sys.flags, attr)), int, attr) 4334710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertTrue(repr(sys.flags)) 4344710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 4354710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def test_clear_type_cache(self): 4364710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm sys._clear_type_cache() 4374710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 4384710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def test_ioencoding(self): 4394710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm import subprocess 4404710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm env = dict(os.environ) 4414710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 4424710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # Test character: cent sign, encoded as 0x4A (ASCII J) in CP424, 4434710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # not representable in ASCII. 4444710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 4454710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm env["PYTHONIOENCODING"] = "cp424" 4464710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm p = subprocess.Popen([sys.executable, "-c", 'print unichr(0xa2)'], 4474710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm stdout = subprocess.PIPE, env=env) 4484710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm out = p.communicate()[0].strip() 4494710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(out, unichr(0xa2).encode("cp424")) 4504710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 4514710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm env["PYTHONIOENCODING"] = "ascii:replace" 4524710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm p = subprocess.Popen([sys.executable, "-c", 'print unichr(0xa2)'], 4534710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm stdout = subprocess.PIPE, env=env) 4544710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm out = p.communicate()[0].strip() 4554710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(out, '?') 4564710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 4574710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def test_call_tracing(self): 4584710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(sys.call_tracing(str, (2,)), "2") 4594710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertRaises(TypeError, sys.call_tracing, str, 2) 4604710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 4614710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def test_executable(self): 4624710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # Issue #7774: Ensure that sys.executable is an empty string if argv[0] 4634710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # has been set to an non existent program name and Python is unable to 4644710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # retrieve the real program name 4654710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm import subprocess 4664710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # For a normal installation, it should work without 'cwd' 4674710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # argument. For test runs in the build directory, see #7774. 4684710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm python_dir = os.path.dirname(os.path.realpath(sys.executable)) 4694710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm p = subprocess.Popen( 4704710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm ["nonexistent", "-c", 'import sys; print repr(sys.executable)'], 4714710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm executable=sys.executable, stdout=subprocess.PIPE, cwd=python_dir) 4724710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm executable = p.communicate()[0].strip() 4734710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm p.wait() 4744710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertIn(executable, ["''", repr(sys.executable)]) 4754710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 4764710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylmclass SizeofTest(unittest.TestCase): 4774710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 4784710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm TPFLAGS_HAVE_GC = 1<<14 4794710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm TPFLAGS_HEAPTYPE = 1L<<9 4804710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 4814710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def setUp(self): 4824710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.c = len(struct.pack('c', ' ')) 4834710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.H = len(struct.pack('H', 0)) 4844710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.i = len(struct.pack('i', 0)) 4854710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.l = len(struct.pack('l', 0)) 4864710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.P = len(struct.pack('P', 0)) 4874710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # due to missing size_t information from struct, it is assumed that 4884710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # sizeof(Py_ssize_t) = sizeof(void*) 4894710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.header = 'PP' 4904710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.vheader = self.header + 'P' 4914710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm if hasattr(sys, "gettotalrefcount"): 4924710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.header += '2P' 4934710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.vheader += '2P' 4944710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.longdigit = sys.long_info.sizeof_digit 4954710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm import _testcapi 4964710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.gc_headsize = _testcapi.SIZEOF_PYGC_HEAD 4974710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.file = open(test.test_support.TESTFN, 'wb') 4984710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 4994710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def tearDown(self): 5004710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.file.close() 5014710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm test.test_support.unlink(test.test_support.TESTFN) 5024710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 5034710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def check_sizeof(self, o, size): 5044710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm result = sys.getsizeof(o) 5054710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm if ((type(o) == type) and (o.__flags__ & self.TPFLAGS_HEAPTYPE) or\ 5064710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm ((type(o) != type) and (type(o).__flags__ & self.TPFLAGS_HAVE_GC))): 5074710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm size += self.gc_headsize 5084710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm msg = 'wrong size for %s: got %d, expected %d' \ 5094710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm % (type(o), result, size) 5104710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(result, size, msg) 5114710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 5124710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def calcsize(self, fmt): 5134710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm """Wrapper around struct.calcsize which enforces the alignment of the 5144710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm end of a structure to the alignment requirement of pointer. 5154710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 5164710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm Note: This wrapper should only be used if a pointer member is included 5174710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm and no member with a size larger than a pointer exists. 5184710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm """ 5194710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm return struct.calcsize(fmt + '0P') 5204710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 5214710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def test_gc_head_size(self): 5224710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # Check that the gc header size is added to objects tracked by the gc. 5234710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm h = self.header 5244710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm size = self.calcsize 5254710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm gc_header_size = self.gc_headsize 5264710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # bool objects are not gc tracked 5274710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(sys.getsizeof(True), size(h + 'l')) 5284710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # but lists are 5294710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(sys.getsizeof([]), size(h + 'P PP') + gc_header_size) 5304710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 5314710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def test_default(self): 5324710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm h = self.header 5334710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm size = self.calcsize 5344710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(sys.getsizeof(True, -1), size(h + 'l')) 5354710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 5364710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def test_objecttypes(self): 5374710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # check all types defined in Objects/ 5384710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm h = self.header 5394710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm vh = self.vheader 5404710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm size = self.calcsize 5414710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check = self.check_sizeof 5424710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # bool 5434710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(True, size(h + 'l')) 5444710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # buffer 5454710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm with test.test_support.check_py3k_warnings(): 5464710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(buffer(''), size(h + '2P2Pil')) 5474710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # builtin_function_or_method 5484710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(len, size(h + '3P')) 5494710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # bytearray 5504710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm samples = ['', 'u'*100000] 5514710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm for sample in samples: 5524710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm x = bytearray(sample) 5534710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(x, size(vh + 'iPP') + x.__alloc__() * self.c) 5544710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # bytearray_iterator 5554710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(iter(bytearray()), size(h + 'PP')) 5564710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # cell 5574710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def get_cell(): 5584710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm x = 42 5594710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def inner(): 5604710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm return x 5614710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm return inner 5624710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(get_cell().func_closure[0], size(h + 'P')) 5634710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # classobj (old-style class) 5644710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm class class_oldstyle(): 5654710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def method(): 5664710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm pass 5674710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(class_oldstyle, size(h + '7P')) 5684710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # instance (old-style class) 5694710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(class_oldstyle(), size(h + '3P')) 5704710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # instancemethod (old-style class) 5714710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(class_oldstyle().method, size(h + '4P')) 5724710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # complex 5734710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(complex(0,1), size(h + '2d')) 5744710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # code 5754710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(get_cell().func_code, size(h + '4i8Pi3P')) 5764710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # BaseException 5774710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(BaseException(), size(h + '3P')) 5784710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # UnicodeEncodeError 5794710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(UnicodeEncodeError("", u"", 0, 0, ""), size(h + '5P2PP')) 5804710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # UnicodeDecodeError 5814710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(UnicodeDecodeError("", "", 0, 0, ""), size(h + '5P2PP')) 5824710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # UnicodeTranslateError 5834710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(UnicodeTranslateError(u"", 0, 1, ""), size(h + '5P2PP')) 5844710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # method_descriptor (descriptor object) 5854710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(str.lower, size(h + '2PP')) 5864710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # classmethod_descriptor (descriptor object) 5874710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # XXX 5884710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # member_descriptor (descriptor object) 5894710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm import datetime 5904710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(datetime.timedelta.days, size(h + '2PP')) 5914710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # getset_descriptor (descriptor object) 5924710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm import __builtin__ 5934710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(__builtin__.file.closed, size(h + '2PP')) 5944710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # wrapper_descriptor (descriptor object) 5954710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(int.__add__, size(h + '2P2P')) 5964710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # dictproxy 5974710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm class C(object): pass 5984710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(C.__dict__, size(h + 'P')) 5994710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # method-wrapper (descriptor object) 6004710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check({}.__iter__, size(h + '2P')) 6014710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # dict 6024710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check({}, size(h + '3P2P' + 8*'P2P')) 6034710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm x = {1:1, 2:2, 3:3, 4:4, 5:5, 6:6, 7:7, 8:8} 6044710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(x, size(h + '3P2P' + 8*'P2P') + 16*size('P2P')) 6054710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # dictionary-keyiterator 6064710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check({}.iterkeys(), size(h + 'P2PPP')) 6074710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # dictionary-valueiterator 6084710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check({}.itervalues(), size(h + 'P2PPP')) 6094710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # dictionary-itemiterator 6104710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check({}.iteritems(), size(h + 'P2PPP')) 6114710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # ellipses 6124710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(Ellipsis, size(h + '')) 6134710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # EncodingMap 6144710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm import codecs, encodings.iso8859_3 6154710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm x = codecs.charmap_build(encodings.iso8859_3.decoding_table) 6164710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(x, size(h + '32B2iB')) 6174710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # enumerate 6184710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(enumerate([]), size(h + 'l3P')) 6194710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # file 6204710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(self.file, size(h + '4P2i4P3i3P3i')) 6214710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # float 6224710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(float(0), size(h + 'd')) 6234710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # sys.floatinfo 6244710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(sys.float_info, size(vh) + self.P * len(sys.float_info)) 6254710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # frame 6264710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm import inspect 6274710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm CO_MAXBLOCKS = 20 6284710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm x = inspect.currentframe() 6294710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm ncells = len(x.f_code.co_cellvars) 6304710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm nfrees = len(x.f_code.co_freevars) 6314710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm extras = x.f_code.co_stacksize + x.f_code.co_nlocals +\ 6324710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm ncells + nfrees - 1 6334710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(x, size(vh + '12P3i' + CO_MAXBLOCKS*'3i' + 'P' + extras*'P')) 6344710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # function 6354710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def func(): pass 6364710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(func, size(h + '9P')) 6374710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm class c(): 6384710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm @staticmethod 6394710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def foo(): 6404710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm pass 6414710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm @classmethod 6424710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def bar(cls): 6434710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm pass 6444710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # staticmethod 6454710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(foo, size(h + 'P')) 6464710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # classmethod 6474710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(bar, size(h + 'P')) 6484710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # generator 6494710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def get_gen(): yield 1 6504710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(get_gen(), size(h + 'Pi2P')) 6514710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # integer 6524710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(1, size(h + 'l')) 6534710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(100, size(h + 'l')) 6544710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # iterator 6554710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(iter('abc'), size(h + 'lP')) 6564710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # callable-iterator 6574710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm import re 6584710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(re.finditer('',''), size(h + '2P')) 6594710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # list 6604710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm samples = [[], [1,2,3], ['1', '2', '3']] 6614710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm for sample in samples: 6624710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(sample, size(vh + 'PP') + len(sample)*self.P) 6634710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # sortwrapper (list) 6644710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # XXX 6654710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # cmpwrapper (list) 6664710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # XXX 6674710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # listiterator (list) 6684710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(iter([]), size(h + 'lP')) 6694710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # listreverseiterator (list) 6704710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(reversed([]), size(h + 'lP')) 6714710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # long 6724710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(0L, size(vh)) 6734710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(1L, size(vh) + self.longdigit) 6744710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(-1L, size(vh) + self.longdigit) 6754710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm PyLong_BASE = 2**sys.long_info.bits_per_digit 6764710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(long(PyLong_BASE), size(vh) + 2*self.longdigit) 6774710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(long(PyLong_BASE**2-1), size(vh) + 2*self.longdigit) 6784710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(long(PyLong_BASE**2), size(vh) + 3*self.longdigit) 6794710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # module 6804710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(unittest, size(h + 'P')) 6814710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # None 6824710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(None, size(h + '')) 6834710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # object 6844710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(object(), size(h + '')) 6854710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # property (descriptor object) 6864710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm class C(object): 6874710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def getx(self): return self.__x 6884710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def setx(self, value): self.__x = value 6894710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def delx(self): del self.__x 6904710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm x = property(getx, setx, delx, "") 6914710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(x, size(h + '4Pi')) 6924710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # PyCObject 6934710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # PyCapsule 6944710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # XXX 6954710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # rangeiterator 6964710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(iter(xrange(1)), size(h + '4l')) 6974710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # reverse 6984710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(reversed(''), size(h + 'PP')) 6994710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # set 7004710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # frozenset 7014710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm PySet_MINSIZE = 8 7024710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm samples = [[], range(10), range(50)] 7034710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm s = size(h + '3P2P' + PySet_MINSIZE*'lP' + 'lP') 7044710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm for sample in samples: 7054710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm minused = len(sample) 7064710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm if minused == 0: tmp = 1 7074710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # the computation of minused is actually a bit more complicated 7084710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # but this suffices for the sizeof test 7094710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm minused = minused*2 7104710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm newsize = PySet_MINSIZE 7114710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm while newsize <= minused: 7124710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm newsize = newsize << 1 7134710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm if newsize <= 8: 7144710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(set(sample), s) 7154710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(frozenset(sample), s) 7164710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm else: 7174710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(set(sample), s + newsize*struct.calcsize('lP')) 7184710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(frozenset(sample), s + newsize*struct.calcsize('lP')) 7194710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # setiterator 7204710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(iter(set()), size(h + 'P3P')) 7214710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # slice 7224710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(slice(1), size(h + '3P')) 7234710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # str 7244710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check('', struct.calcsize(vh + 'li') + 1) 7254710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check('abc', struct.calcsize(vh + 'li') + 1 + 3*self.c) 7264710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # super 7274710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(super(int), size(h + '3P')) 7284710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # tuple 7294710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check((), size(vh)) 7304710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check((1,2,3), size(vh) + 3*self.P) 7314710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # tupleiterator 7324710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(iter(()), size(h + 'lP')) 7334710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # type 7344710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # (PyTypeObject + PyNumberMethods + PyMappingMethods + 7354710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # PySequenceMethods + PyBufferProcs) 7364710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm s = size(vh + 'P2P15Pl4PP9PP11PI') + size('41P 10P 3P 6P') 7374710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm class newstyleclass(object): 7384710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm pass 7394710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(newstyleclass, s) 7404710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # builtin type 7414710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(int, s) 7424710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # NotImplementedType 7434710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm import types 7444710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(types.NotImplementedType, s) 7454710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # unicode 7464710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm usize = len(u'\0'.encode('unicode-internal')) 7474710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm samples = [u'', u'1'*100] 7484710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # we need to test for both sizes, because we don't know if the string 7494710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # has been cached 7504710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm for s in samples: 7514710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(s, size(h + 'PPlP') + usize * (len(s) + 1)) 7524710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # weakref 7534710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm import weakref 7544710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(weakref.ref(int), size(h + '2Pl2P')) 7554710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # weakproxy 7564710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # XXX 7574710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # weakcallableproxy 7584710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(weakref.proxy(int), size(h + '2Pl2P')) 7594710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # xrange 7604710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(xrange(1), size(h + '3l')) 7614710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(xrange(66000), size(h + '3l')) 7624710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 7634710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def test_pythontypes(self): 7644710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # check all types defined in Python/ 7654710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm h = self.header 7664710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm vh = self.vheader 7674710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm size = self.calcsize 7684710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check = self.check_sizeof 7694710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # _ast.AST 7704710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm import _ast 7714710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(_ast.AST(), size(h + '')) 7724710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # imp.NullImporter 7734710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm import imp 7744710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(imp.NullImporter(self.file.name), size(h + '')) 7754710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm try: 7764710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm raise TypeError 7774710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm except TypeError: 7784710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm tb = sys.exc_info()[2] 7794710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # traceback 7804710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm if tb != None: 7814710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(tb, size(h + '2P2i')) 7824710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # symtable entry 7834710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # XXX 7844710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # sys.flags 7854710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm check(sys.flags, size(vh) + self.P * len(sys.flags)) 7864710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 7874710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 7884710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylmdef test_main(): 7894710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm test_classes = (SysModuleTest, SizeofTest) 7904710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 7914710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm test.test_support.run_unittest(*test_classes) 7924710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 7934710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylmif __name__ == "__main__": 7944710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm test_main() 795