| Benjamin Peterson | ee8712c | 2008-05-20 21:35:26 +0000 | [diff] [blame] | 1 | import unittest, test.support | 
| Martin v. Löwis | 00709aa | 2008-06-04 14:18:43 +0000 | [diff] [blame] | 2 | import sys, io, os | 
| Robert Schuppenies | 4d45bfe | 2008-06-28 23:58:47 +0000 | [diff] [blame] | 3 | import struct | 
| Antoine Pitrou | 652e707 | 2009-03-13 19:25:20 +0000 | [diff] [blame] | 4 | import subprocess | 
 | 5 | import textwrap | 
| Antoine Pitrou | c3b0757 | 2009-11-13 22:19:19 +0000 | [diff] [blame] | 6 | import warnings | 
| Eric Smith | f7bb578 | 2010-01-27 00:44:57 +0000 | [diff] [blame] | 7 | import operator | 
| Victor Stinner | 85ca2af | 2010-08-19 11:23:47 +0000 | [diff] [blame] | 8 | import codecs | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 9 |  | 
| Georg Brandl | 0352655 | 2009-10-27 23:00:28 +0000 | [diff] [blame] | 10 | # count the number of test runs, used to create unique | 
 | 11 | # strings to intern in test_intern() | 
 | 12 | numruns = 0 | 
 | 13 |  | 
| Victor Stinner | 45df820 | 2010-04-28 22:31:17 +0000 | [diff] [blame] | 14 | try: | 
 | 15 |     import threading | 
 | 16 | except ImportError: | 
 | 17 |     threading = None | 
| Georg Brandl | 0352655 | 2009-10-27 23:00:28 +0000 | [diff] [blame] | 18 |  | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 19 | class SysModuleTest(unittest.TestCase): | 
 | 20 |  | 
| Christian Heimes | ef181a7 | 2007-11-07 16:14:12 +0000 | [diff] [blame] | 21 |     def setUp(self): | 
 | 22 |         self.orig_stdout = sys.stdout | 
 | 23 |         self.orig_stderr = sys.stderr | 
 | 24 |         self.orig_displayhook = sys.displayhook | 
 | 25 |  | 
 | 26 |     def tearDown(self): | 
 | 27 |         sys.stdout = self.orig_stdout | 
 | 28 |         sys.stderr = self.orig_stderr | 
 | 29 |         sys.displayhook = self.orig_displayhook | 
| Antoine Pitrou | 836f0e2 | 2009-10-27 19:25:57 +0000 | [diff] [blame] | 30 |         test.support.reap_children() | 
| Christian Heimes | ef181a7 | 2007-11-07 16:14:12 +0000 | [diff] [blame] | 31 |  | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 32 |     def test_original_displayhook(self): | 
| Georg Brandl | 1a3284e | 2007-12-02 09:40:06 +0000 | [diff] [blame] | 33 |         import builtins | 
| Guido van Rossum | 34d1928 | 2007-08-09 01:03:29 +0000 | [diff] [blame] | 34 |         out = io.StringIO() | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 35 |         sys.stdout = out | 
 | 36 |  | 
 | 37 |         dh = sys.__displayhook__ | 
 | 38 |  | 
 | 39 |         self.assertRaises(TypeError, dh) | 
| Georg Brandl | 1a3284e | 2007-12-02 09:40:06 +0000 | [diff] [blame] | 40 |         if hasattr(builtins, "_"): | 
 | 41 |             del builtins._ | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 42 |  | 
 | 43 |         dh(None) | 
 | 44 |         self.assertEqual(out.getvalue(), "") | 
| Benjamin Peterson | c9c0f20 | 2009-06-30 23:06:06 +0000 | [diff] [blame] | 45 |         self.assertTrue(not hasattr(builtins, "_")) | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 46 |         dh(42) | 
 | 47 |         self.assertEqual(out.getvalue(), "42\n") | 
| Georg Brandl | 1a3284e | 2007-12-02 09:40:06 +0000 | [diff] [blame] | 48 |         self.assertEqual(builtins._, 42) | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 49 |  | 
 | 50 |         del sys.stdout | 
 | 51 |         self.assertRaises(RuntimeError, dh, 42) | 
 | 52 |  | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 53 |     def test_lost_displayhook(self): | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 54 |         del sys.displayhook | 
 | 55 |         code = compile("42", "<string>", "single") | 
 | 56 |         self.assertRaises(RuntimeError, eval, code) | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 57 |  | 
 | 58 |     def test_custom_displayhook(self): | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 59 |         def baddisplayhook(obj): | 
 | 60 |             raise ValueError | 
 | 61 |         sys.displayhook = baddisplayhook | 
 | 62 |         code = compile("42", "<string>", "single") | 
 | 63 |         self.assertRaises(ValueError, eval, code) | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 64 |  | 
 | 65 |     def test_original_excepthook(self): | 
| Guido van Rossum | 34d1928 | 2007-08-09 01:03:29 +0000 | [diff] [blame] | 66 |         err = io.StringIO() | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 67 |         sys.stderr = err | 
 | 68 |  | 
 | 69 |         eh = sys.__excepthook__ | 
 | 70 |  | 
 | 71 |         self.assertRaises(TypeError, eh) | 
 | 72 |         try: | 
 | 73 |             raise ValueError(42) | 
| Guido van Rossum | b940e11 | 2007-01-10 16:19:56 +0000 | [diff] [blame] | 74 |         except ValueError as exc: | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 75 |             eh(*sys.exc_info()) | 
 | 76 |  | 
| Benjamin Peterson | c9c0f20 | 2009-06-30 23:06:06 +0000 | [diff] [blame] | 77 |         self.assertTrue(err.getvalue().endswith("ValueError: 42\n")) | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 78 |  | 
| Benjamin Peterson | bc74e5b | 2008-08-23 20:19:17 +0000 | [diff] [blame] | 79 |     def test_excepthook(self): | 
 | 80 |         with test.support.captured_output("stderr") as stderr: | 
 | 81 |             sys.excepthook(1, '1', 1) | 
| Benjamin Peterson | c9c0f20 | 2009-06-30 23:06:06 +0000 | [diff] [blame] | 82 |         self.assertTrue("TypeError: print_exception(): Exception expected for " \ | 
| Benjamin Peterson | bc74e5b | 2008-08-23 20:19:17 +0000 | [diff] [blame] | 83 |                          "value, str found" in stderr.getvalue()) | 
 | 84 |  | 
| Walter Dörwald | e7028ac | 2003-02-03 23:05:27 +0000 | [diff] [blame] | 85 |     # FIXME: testing the code for a lost or replaced excepthook in | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 86 |     # Python/pythonrun.c::PyErr_PrintEx() is tricky. | 
 | 87 |  | 
 | 88 |     def test_exit(self): | 
| Victor Stinner | e9fb319 | 2010-05-17 08:58:51 +0000 | [diff] [blame] | 89 |  | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 90 |         self.assertRaises(TypeError, sys.exit, 42, 42) | 
 | 91 |  | 
 | 92 |         # call without argument | 
 | 93 |         try: | 
 | 94 |             sys.exit(0) | 
| Guido van Rossum | b940e11 | 2007-01-10 16:19:56 +0000 | [diff] [blame] | 95 |         except SystemExit as exc: | 
| Ezio Melotti | b3aedd4 | 2010-11-20 19:04:17 +0000 | [diff] [blame] | 96 |             self.assertEqual(exc.code, 0) | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 97 |         except: | 
 | 98 |             self.fail("wrong exception") | 
 | 99 |         else: | 
 | 100 |             self.fail("no exception") | 
 | 101 |  | 
 | 102 |         # call with tuple argument with one entry | 
 | 103 |         # entry will be unpacked | 
 | 104 |         try: | 
 | 105 |             sys.exit(42) | 
| Guido van Rossum | b940e11 | 2007-01-10 16:19:56 +0000 | [diff] [blame] | 106 |         except SystemExit as exc: | 
| Ezio Melotti | b3aedd4 | 2010-11-20 19:04:17 +0000 | [diff] [blame] | 107 |             self.assertEqual(exc.code, 42) | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 108 |         except: | 
 | 109 |             self.fail("wrong exception") | 
 | 110 |         else: | 
 | 111 |             self.fail("no exception") | 
 | 112 |  | 
 | 113 |         # call with integer argument | 
 | 114 |         try: | 
 | 115 |             sys.exit((42,)) | 
| Guido van Rossum | b940e11 | 2007-01-10 16:19:56 +0000 | [diff] [blame] | 116 |         except SystemExit as exc: | 
| Ezio Melotti | b3aedd4 | 2010-11-20 19:04:17 +0000 | [diff] [blame] | 117 |             self.assertEqual(exc.code, 42) | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 118 |         except: | 
 | 119 |             self.fail("wrong exception") | 
 | 120 |         else: | 
 | 121 |             self.fail("no exception") | 
 | 122 |  | 
 | 123 |         # call with string argument | 
 | 124 |         try: | 
 | 125 |             sys.exit("exit") | 
| Guido van Rossum | b940e11 | 2007-01-10 16:19:56 +0000 | [diff] [blame] | 126 |         except SystemExit as exc: | 
| Ezio Melotti | b3aedd4 | 2010-11-20 19:04:17 +0000 | [diff] [blame] | 127 |             self.assertEqual(exc.code, "exit") | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 128 |         except: | 
 | 129 |             self.fail("wrong exception") | 
 | 130 |         else: | 
 | 131 |             self.fail("no exception") | 
 | 132 |  | 
 | 133 |         # call with tuple argument with two entries | 
 | 134 |         try: | 
 | 135 |             sys.exit((17, 23)) | 
| Guido van Rossum | b940e11 | 2007-01-10 16:19:56 +0000 | [diff] [blame] | 136 |         except SystemExit as exc: | 
| Ezio Melotti | b3aedd4 | 2010-11-20 19:04:17 +0000 | [diff] [blame] | 137 |             self.assertEqual(exc.code, (17, 23)) | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 138 |         except: | 
 | 139 |             self.fail("wrong exception") | 
 | 140 |         else: | 
 | 141 |             self.fail("no exception") | 
 | 142 |  | 
| Michael W. Hudson | f058858 | 2005-02-15 15:26:11 +0000 | [diff] [blame] | 143 |         # test that the exit machinery handles SystemExits properly | 
| Michael W. Hudson | f058858 | 2005-02-15 15:26:11 +0000 | [diff] [blame] | 144 |         rc = subprocess.call([sys.executable, "-c", | 
 | 145 |                               "raise SystemExit(47)"]) | 
 | 146 |         self.assertEqual(rc, 47) | 
| Tim Peters | f0db38d | 2005-02-15 21:50:12 +0000 | [diff] [blame] | 147 |  | 
| Victor Stinner | 7126dbc | 2010-05-21 23:45:42 +0000 | [diff] [blame] | 148 |         def check_exit_message(code, expected, env=None): | 
| Victor Stinner | e9fb319 | 2010-05-17 08:58:51 +0000 | [diff] [blame] | 149 |             process = subprocess.Popen([sys.executable, "-c", code], | 
| Victor Stinner | 7126dbc | 2010-05-21 23:45:42 +0000 | [diff] [blame] | 150 |                                        stderr=subprocess.PIPE, env=env) | 
| Victor Stinner | e9fb319 | 2010-05-17 08:58:51 +0000 | [diff] [blame] | 151 |             stdout, stderr = process.communicate() | 
 | 152 |             self.assertEqual(process.returncode, 1) | 
| Victor Stinner | 7cf7d72 | 2010-05-17 14:36:43 +0000 | [diff] [blame] | 153 |             self.assertTrue(stderr.startswith(expected), | 
| Victor Stinner | c098385 | 2010-05-17 19:57:40 +0000 | [diff] [blame] | 154 |                 "%s doesn't start with %s" % (ascii(stderr), ascii(expected))) | 
| Victor Stinner | e9fb319 | 2010-05-17 08:58:51 +0000 | [diff] [blame] | 155 |  | 
 | 156 |         # test that stderr buffer if flushed before the exit message is written | 
 | 157 |         # into stderr | 
 | 158 |         check_exit_message( | 
 | 159 |             r'import sys; sys.stderr.write("unflushed,"); sys.exit("message")', | 
 | 160 |             b"unflushed,message") | 
 | 161 |  | 
| Victor Stinner | 372ac5e | 2010-05-17 01:26:01 +0000 | [diff] [blame] | 162 |         # test that the exit message is written with backslashreplace error | 
 | 163 |         # handler to stderr | 
| Victor Stinner | e9fb319 | 2010-05-17 08:58:51 +0000 | [diff] [blame] | 164 |         check_exit_message( | 
 | 165 |             r'import sys; sys.exit("surrogates:\uDCFF")', | 
 | 166 |             b"surrogates:\\udcff") | 
| Victor Stinner | 372ac5e | 2010-05-17 01:26:01 +0000 | [diff] [blame] | 167 |  | 
| Victor Stinner | 7126dbc | 2010-05-21 23:45:42 +0000 | [diff] [blame] | 168 |         # test that the unicode message is encoded to the stderr encoding | 
 | 169 |         # instead of the default encoding (utf8) | 
 | 170 |         env = os.environ.copy() | 
 | 171 |         env['PYTHONIOENCODING'] = 'latin-1' | 
 | 172 |         check_exit_message( | 
 | 173 |             r'import sys; sys.exit("h\xe9")', | 
 | 174 |             b"h\xe9", env=env) | 
 | 175 |  | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 176 |     def test_getdefaultencoding(self): | 
| Walter Dörwald | 4aeaa96 | 2007-05-22 16:13:46 +0000 | [diff] [blame] | 177 |         self.assertRaises(TypeError, sys.getdefaultencoding, 42) | 
 | 178 |         # can't check more than the type, as the user might have changed it | 
| Ezio Melotti | e961593 | 2010-01-24 19:26:24 +0000 | [diff] [blame] | 179 |         self.assertIsInstance(sys.getdefaultencoding(), str) | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 180 |  | 
| Alexander Belopolsky | 07de165 | 2010-07-25 15:05:42 +0000 | [diff] [blame] | 181 |     # testing sys.settrace() is done in test_sys_settrace.py | 
 | 182 |     # testing sys.setprofile() is done in test_sys_setprofile.py | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 183 |  | 
 | 184 |     def test_setcheckinterval(self): | 
| Antoine Pitrou | c3b0757 | 2009-11-13 22:19:19 +0000 | [diff] [blame] | 185 |         with warnings.catch_warnings(): | 
 | 186 |             warnings.simplefilter("ignore") | 
 | 187 |             self.assertRaises(TypeError, sys.setcheckinterval) | 
 | 188 |             orig = sys.getcheckinterval() | 
 | 189 |             for n in 0, 100, 120, orig: # orig last to restore starting state | 
 | 190 |                 sys.setcheckinterval(n) | 
| Ezio Melotti | b3aedd4 | 2010-11-20 19:04:17 +0000 | [diff] [blame] | 191 |                 self.assertEqual(sys.getcheckinterval(), n) | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 192 |  | 
| Victor Stinner | 45df820 | 2010-04-28 22:31:17 +0000 | [diff] [blame] | 193 |     @unittest.skipUnless(threading, 'Threading required for this test.') | 
| Antoine Pitrou | 074e5ed | 2009-11-10 19:50:40 +0000 | [diff] [blame] | 194 |     def test_switchinterval(self): | 
 | 195 |         self.assertRaises(TypeError, sys.setswitchinterval) | 
 | 196 |         self.assertRaises(TypeError, sys.setswitchinterval, "a") | 
 | 197 |         self.assertRaises(ValueError, sys.setswitchinterval, -1.0) | 
 | 198 |         self.assertRaises(ValueError, sys.setswitchinterval, 0.0) | 
 | 199 |         orig = sys.getswitchinterval() | 
 | 200 |         # sanity check | 
 | 201 |         self.assertTrue(orig < 0.5, orig) | 
 | 202 |         try: | 
 | 203 |             for n in 0.00001, 0.05, 3.0, orig: | 
 | 204 |                 sys.setswitchinterval(n) | 
| Ezio Melotti | b3aedd4 | 2010-11-20 19:04:17 +0000 | [diff] [blame] | 205 |                 self.assertAlmostEqual(sys.getswitchinterval(), n) | 
| Antoine Pitrou | 074e5ed | 2009-11-10 19:50:40 +0000 | [diff] [blame] | 206 |         finally: | 
 | 207 |             sys.setswitchinterval(orig) | 
 | 208 |  | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 209 |     def test_recursionlimit(self): | 
| Tim Peters | f2715e0 | 2003-02-19 02:35:07 +0000 | [diff] [blame] | 210 |         self.assertRaises(TypeError, sys.getrecursionlimit, 42) | 
 | 211 |         oldlimit = sys.getrecursionlimit() | 
 | 212 |         self.assertRaises(TypeError, sys.setrecursionlimit) | 
 | 213 |         self.assertRaises(ValueError, sys.setrecursionlimit, -42) | 
 | 214 |         sys.setrecursionlimit(10000) | 
 | 215 |         self.assertEqual(sys.getrecursionlimit(), 10000) | 
 | 216 |         sys.setrecursionlimit(oldlimit) | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 217 |  | 
| Antoine Pitrou | 652e707 | 2009-03-13 19:25:20 +0000 | [diff] [blame] | 218 |     def test_recursionlimit_recovery(self): | 
 | 219 |         # NOTE: this test is slightly fragile in that it depends on the current | 
 | 220 |         # recursion count when executing the test being low enough so as to | 
 | 221 |         # trigger the recursion recovery detection in the _Py_MakeEndRecCheck | 
 | 222 |         # macro (see ceval.h). | 
 | 223 |         oldlimit = sys.getrecursionlimit() | 
 | 224 |         def f(): | 
 | 225 |             f() | 
 | 226 |         try: | 
 | 227 |             for i in (50, 1000): | 
 | 228 |                 # Issue #5392: stack overflow after hitting recursion limit twice | 
 | 229 |                 sys.setrecursionlimit(i) | 
 | 230 |                 self.assertRaises(RuntimeError, f) | 
 | 231 |                 self.assertRaises(RuntimeError, f) | 
 | 232 |         finally: | 
 | 233 |             sys.setrecursionlimit(oldlimit) | 
 | 234 |  | 
 | 235 |     def test_recursionlimit_fatalerror(self): | 
 | 236 |         # A fatal error occurs if a second recursion limit is hit when recovering | 
 | 237 |         # from a first one. | 
| Antoine Pitrou | 17dc446 | 2009-04-04 17:36:05 +0000 | [diff] [blame] | 238 |         if os.name == "nt": | 
 | 239 |             raise unittest.SkipTest( | 
 | 240 |                 "under Windows, test would generate a spurious crash dialog") | 
| Antoine Pitrou | 652e707 | 2009-03-13 19:25:20 +0000 | [diff] [blame] | 241 |         code = textwrap.dedent(""" | 
 | 242 |             import sys | 
 | 243 |  | 
 | 244 |             def f(): | 
 | 245 |                 try: | 
 | 246 |                     f() | 
 | 247 |                 except RuntimeError: | 
 | 248 |                     f() | 
 | 249 |  | 
 | 250 |             sys.setrecursionlimit(%d) | 
 | 251 |             f()""") | 
 | 252 |         for i in (50, 1000): | 
 | 253 |             sub = subprocess.Popen([sys.executable, '-c', code % i], | 
 | 254 |                 stderr=subprocess.PIPE) | 
 | 255 |             err = sub.communicate()[1] | 
 | 256 |             self.assertTrue(sub.returncode, sub.returncode) | 
 | 257 |             self.assertTrue( | 
 | 258 |                 b"Fatal Python error: Cannot recover from stack overflow" in err, | 
 | 259 |                 err) | 
 | 260 |  | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 261 |     def test_getwindowsversion(self): | 
| Eric Smith | 05f33744 | 2010-01-27 01:25:02 +0000 | [diff] [blame] | 262 |         # Raise SkipTest if sys doesn't have getwindowsversion attribute | 
| Eric Smith | d8aef76 | 2010-01-27 01:38:34 +0000 | [diff] [blame] | 263 |         test.support.get_attribute(sys, "getwindowsversion") | 
| Eric Smith | 05f33744 | 2010-01-27 01:25:02 +0000 | [diff] [blame] | 264 |         v = sys.getwindowsversion() | 
| Brian Curtin | 808e1ad | 2010-07-08 22:13:25 +0000 | [diff] [blame] | 265 |         self.assertEqual(len(v), 5) | 
| Eric Smith | 05f33744 | 2010-01-27 01:25:02 +0000 | [diff] [blame] | 266 |         self.assertIsInstance(v[0], int) | 
 | 267 |         self.assertIsInstance(v[1], int) | 
 | 268 |         self.assertIsInstance(v[2], int) | 
 | 269 |         self.assertIsInstance(v[3], int) | 
 | 270 |         self.assertIsInstance(v[4], str) | 
 | 271 |         self.assertRaises(IndexError, operator.getitem, v, 5) | 
 | 272 |         self.assertIsInstance(v.major, int) | 
 | 273 |         self.assertIsInstance(v.minor, int) | 
 | 274 |         self.assertIsInstance(v.build, int) | 
 | 275 |         self.assertIsInstance(v.platform, int) | 
 | 276 |         self.assertIsInstance(v.service_pack, str) | 
| Eric Smith | a1f20b9 | 2010-01-27 02:14:26 +0000 | [diff] [blame] | 277 |         self.assertIsInstance(v.service_pack_minor, int) | 
 | 278 |         self.assertIsInstance(v.service_pack_major, int) | 
 | 279 |         self.assertIsInstance(v.suite_mask, int) | 
 | 280 |         self.assertIsInstance(v.product_type, int) | 
| Eric Smith | 05f33744 | 2010-01-27 01:25:02 +0000 | [diff] [blame] | 281 |         self.assertEqual(v[0], v.major) | 
 | 282 |         self.assertEqual(v[1], v.minor) | 
 | 283 |         self.assertEqual(v[2], v.build) | 
 | 284 |         self.assertEqual(v[3], v.platform) | 
 | 285 |         self.assertEqual(v[4], v.service_pack) | 
| Eric Smith | f7bb578 | 2010-01-27 00:44:57 +0000 | [diff] [blame] | 286 |  | 
| Eric Smith | 05f33744 | 2010-01-27 01:25:02 +0000 | [diff] [blame] | 287 |         # This is how platform.py calls it. Make sure tuple | 
 | 288 |         #  still has 5 elements | 
 | 289 |         maj, min, buildno, plat, csd = sys.getwindowsversion() | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 290 |  | 
| Barry Warsaw | e94a37f | 2008-10-17 01:29:56 +0000 | [diff] [blame] | 291 |     def test_call_tracing(self): | 
 | 292 |         self.assertRaises(TypeError, sys.call_tracing, type, 2) | 
 | 293 |  | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 294 |     def test_dlopenflags(self): | 
 | 295 |         if hasattr(sys, "setdlopenflags"): | 
| Benjamin Peterson | c9c0f20 | 2009-06-30 23:06:06 +0000 | [diff] [blame] | 296 |             self.assertTrue(hasattr(sys, "getdlopenflags")) | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 297 |             self.assertRaises(TypeError, sys.getdlopenflags, 42) | 
 | 298 |             oldflags = sys.getdlopenflags() | 
 | 299 |             self.assertRaises(TypeError, sys.setdlopenflags) | 
 | 300 |             sys.setdlopenflags(oldflags+1) | 
 | 301 |             self.assertEqual(sys.getdlopenflags(), oldflags+1) | 
 | 302 |             sys.setdlopenflags(oldflags) | 
 | 303 |  | 
 | 304 |     def test_refcount(self): | 
| Benjamin Peterson | d23f822 | 2009-04-05 19:13:16 +0000 | [diff] [blame] | 305 |         # n here must be a global in order for this test to pass while | 
 | 306 |         # tracing with a python function.  Tracing calls PyFrame_FastToLocals | 
 | 307 |         # which will add a copy of any locals to the frame object, causing | 
 | 308 |         # the reference count to increase by 2 instead of 1. | 
 | 309 |         global n | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 310 |         self.assertRaises(TypeError, sys.getrefcount) | 
 | 311 |         c = sys.getrefcount(None) | 
 | 312 |         n = None | 
 | 313 |         self.assertEqual(sys.getrefcount(None), c+1) | 
 | 314 |         del n | 
 | 315 |         self.assertEqual(sys.getrefcount(None), c) | 
 | 316 |         if hasattr(sys, "gettotalrefcount"): | 
| Ezio Melotti | e961593 | 2010-01-24 19:26:24 +0000 | [diff] [blame] | 317 |             self.assertIsInstance(sys.gettotalrefcount(), int) | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 318 |  | 
 | 319 |     def test_getframe(self): | 
 | 320 |         self.assertRaises(TypeError, sys._getframe, 42, 42) | 
| Neal Norwitz | eb2a5ef | 2003-02-18 15:22:10 +0000 | [diff] [blame] | 321 |         self.assertRaises(ValueError, sys._getframe, 2000000000) | 
| Benjamin Peterson | c9c0f20 | 2009-06-30 23:06:06 +0000 | [diff] [blame] | 322 |         self.assertTrue( | 
| Christian Heimes | 4a22b5d | 2007-11-25 09:39:14 +0000 | [diff] [blame] | 323 |             SysModuleTest.test_getframe.__code__ \ | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 324 |             is sys._getframe().f_code | 
 | 325 |         ) | 
 | 326 |  | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 327 |     # sys._current_frames() is a CPython-only gimmick. | 
 | 328 |     def test_current_frames(self): | 
 | 329 |         have_threads = True | 
 | 330 |         try: | 
| Georg Brandl | 2067bfd | 2008-05-25 13:05:15 +0000 | [diff] [blame] | 331 |             import _thread | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 332 |         except ImportError: | 
 | 333 |             have_threads = False | 
 | 334 |  | 
 | 335 |         if have_threads: | 
 | 336 |             self.current_frames_with_threads() | 
 | 337 |         else: | 
 | 338 |             self.current_frames_without_threads() | 
 | 339 |  | 
 | 340 |     # Test sys._current_frames() in a WITH_THREADS build. | 
| Antoine Pitrou | 836f0e2 | 2009-10-27 19:25:57 +0000 | [diff] [blame] | 341 |     @test.support.reap_threads | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 342 |     def current_frames_with_threads(self): | 
| Georg Brandl | 2067bfd | 2008-05-25 13:05:15 +0000 | [diff] [blame] | 343 |         import threading, _thread | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 344 |         import traceback | 
 | 345 |  | 
 | 346 |         # Spawn a thread that blocks at a known place.  Then the main | 
 | 347 |         # thread does sys._current_frames(), and verifies that the frames | 
 | 348 |         # returned make sense. | 
 | 349 |         entered_g = threading.Event() | 
 | 350 |         leave_g = threading.Event() | 
 | 351 |         thread_info = []  # the thread's id | 
 | 352 |  | 
 | 353 |         def f123(): | 
 | 354 |             g456() | 
 | 355 |  | 
 | 356 |         def g456(): | 
| Georg Brandl | 2067bfd | 2008-05-25 13:05:15 +0000 | [diff] [blame] | 357 |             thread_info.append(_thread.get_ident()) | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 358 |             entered_g.set() | 
 | 359 |             leave_g.wait() | 
 | 360 |  | 
 | 361 |         t = threading.Thread(target=f123) | 
 | 362 |         t.start() | 
 | 363 |         entered_g.wait() | 
 | 364 |  | 
 | 365 |         # At this point, t has finished its entered_g.set(), although it's | 
 | 366 |         # impossible to guess whether it's still on that line or has moved on | 
 | 367 |         # to its leave_g.wait(). | 
 | 368 |         self.assertEqual(len(thread_info), 1) | 
 | 369 |         thread_id = thread_info[0] | 
 | 370 |  | 
 | 371 |         d = sys._current_frames() | 
 | 372 |  | 
| Georg Brandl | 2067bfd | 2008-05-25 13:05:15 +0000 | [diff] [blame] | 373 |         main_id = _thread.get_ident() | 
| Benjamin Peterson | 577473f | 2010-01-19 00:09:57 +0000 | [diff] [blame] | 374 |         self.assertIn(main_id, d) | 
 | 375 |         self.assertIn(thread_id, d) | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 376 |  | 
 | 377 |         # Verify that the captured main-thread frame is _this_ frame. | 
 | 378 |         frame = d.pop(main_id) | 
| Benjamin Peterson | c9c0f20 | 2009-06-30 23:06:06 +0000 | [diff] [blame] | 379 |         self.assertTrue(frame is sys._getframe()) | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 380 |  | 
 | 381 |         # Verify that the captured thread frame is blocked in g456, called | 
 | 382 |         # from f123.  This is a litte tricky, since various bits of | 
 | 383 |         # threading.py are also in the thread's call stack. | 
 | 384 |         frame = d.pop(thread_id) | 
 | 385 |         stack = traceback.extract_stack(frame) | 
 | 386 |         for i, (filename, lineno, funcname, sourceline) in enumerate(stack): | 
 | 387 |             if funcname == "f123": | 
 | 388 |                 break | 
 | 389 |         else: | 
 | 390 |             self.fail("didn't find f123() on thread's call stack") | 
 | 391 |  | 
 | 392 |         self.assertEqual(sourceline, "g456()") | 
 | 393 |  | 
 | 394 |         # And the next record must be for g456(). | 
 | 395 |         filename, lineno, funcname, sourceline = stack[i+1] | 
 | 396 |         self.assertEqual(funcname, "g456") | 
| Benjamin Peterson | 577473f | 2010-01-19 00:09:57 +0000 | [diff] [blame] | 397 |         self.assertIn(sourceline, ["leave_g.wait()", "entered_g.set()"]) | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 398 |  | 
 | 399 |         # Reap the spawned thread. | 
 | 400 |         leave_g.set() | 
 | 401 |         t.join() | 
 | 402 |  | 
 | 403 |     # Test sys._current_frames() when thread support doesn't exist. | 
 | 404 |     def current_frames_without_threads(self): | 
 | 405 |         # Not much happens here:  there is only one thread, with artificial | 
 | 406 |         # "thread id" 0. | 
 | 407 |         d = sys._current_frames() | 
 | 408 |         self.assertEqual(len(d), 1) | 
| Benjamin Peterson | 577473f | 2010-01-19 00:09:57 +0000 | [diff] [blame] | 409 |         self.assertIn(0, d) | 
| Benjamin Peterson | c9c0f20 | 2009-06-30 23:06:06 +0000 | [diff] [blame] | 410 |         self.assertTrue(d[0] is sys._getframe()) | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 411 |  | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 412 |     def test_attributes(self): | 
| Ezio Melotti | e961593 | 2010-01-24 19:26:24 +0000 | [diff] [blame] | 413 |         self.assertIsInstance(sys.api_version, int) | 
 | 414 |         self.assertIsInstance(sys.argv, list) | 
| Benjamin Peterson | 577473f | 2010-01-19 00:09:57 +0000 | [diff] [blame] | 415 |         self.assertIn(sys.byteorder, ("little", "big")) | 
| Ezio Melotti | e961593 | 2010-01-24 19:26:24 +0000 | [diff] [blame] | 416 |         self.assertIsInstance(sys.builtin_module_names, tuple) | 
 | 417 |         self.assertIsInstance(sys.copyright, str) | 
 | 418 |         self.assertIsInstance(sys.exec_prefix, str) | 
 | 419 |         self.assertIsInstance(sys.executable, str) | 
| Christian Heimes | 9385266 | 2007-12-01 12:22:32 +0000 | [diff] [blame] | 420 |         self.assertEqual(len(sys.float_info), 11) | 
| Christian Heimes | d32ed6f | 2008-01-14 18:49:24 +0000 | [diff] [blame] | 421 |         self.assertEqual(sys.float_info.radix, 2) | 
| Mark Dickinson | bd79264 | 2009-03-18 20:06:12 +0000 | [diff] [blame] | 422 |         self.assertEqual(len(sys.int_info), 2) | 
| Benjamin Peterson | c9c0f20 | 2009-06-30 23:06:06 +0000 | [diff] [blame] | 423 |         self.assertTrue(sys.int_info.bits_per_digit % 5 == 0) | 
 | 424 |         self.assertTrue(sys.int_info.sizeof_digit >= 1) | 
| Benjamin Peterson | ef3e4c2 | 2009-04-11 19:48:14 +0000 | [diff] [blame] | 425 |         self.assertEqual(type(sys.int_info.bits_per_digit), int) | 
 | 426 |         self.assertEqual(type(sys.int_info.sizeof_digit), int) | 
| Ezio Melotti | e961593 | 2010-01-24 19:26:24 +0000 | [diff] [blame] | 427 |         self.assertIsInstance(sys.hexversion, int) | 
| Mark Dickinson | dc787d2 | 2010-05-23 13:33:13 +0000 | [diff] [blame] | 428 |  | 
 | 429 |         self.assertEqual(len(sys.hash_info), 5) | 
 | 430 |         self.assertLess(sys.hash_info.modulus, 2**sys.hash_info.width) | 
 | 431 |         # sys.hash_info.modulus should be a prime; we do a quick | 
 | 432 |         # probable primality test (doesn't exclude the possibility of | 
 | 433 |         # a Carmichael number) | 
 | 434 |         for x in range(1, 100): | 
 | 435 |             self.assertEqual( | 
 | 436 |                 pow(x, sys.hash_info.modulus-1, sys.hash_info.modulus), | 
 | 437 |                 1, | 
 | 438 |                 "sys.hash_info.modulus {} is a non-prime".format( | 
 | 439 |                     sys.hash_info.modulus) | 
 | 440 |                 ) | 
 | 441 |         self.assertIsInstance(sys.hash_info.inf, int) | 
 | 442 |         self.assertIsInstance(sys.hash_info.nan, int) | 
 | 443 |         self.assertIsInstance(sys.hash_info.imag, int) | 
 | 444 |  | 
| Ezio Melotti | e961593 | 2010-01-24 19:26:24 +0000 | [diff] [blame] | 445 |         self.assertIsInstance(sys.maxsize, int) | 
 | 446 |         self.assertIsInstance(sys.maxunicode, int) | 
 | 447 |         self.assertIsInstance(sys.platform, str) | 
 | 448 |         self.assertIsInstance(sys.prefix, str) | 
 | 449 |         self.assertIsInstance(sys.version, str) | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 450 |         vi = sys.version_info | 
| Ezio Melotti | e961593 | 2010-01-24 19:26:24 +0000 | [diff] [blame] | 451 |         self.assertIsInstance(vi[:], tuple) | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 452 |         self.assertEqual(len(vi), 5) | 
| Ezio Melotti | e961593 | 2010-01-24 19:26:24 +0000 | [diff] [blame] | 453 |         self.assertIsInstance(vi[0], int) | 
 | 454 |         self.assertIsInstance(vi[1], int) | 
 | 455 |         self.assertIsInstance(vi[2], int) | 
| Benjamin Peterson | 577473f | 2010-01-19 00:09:57 +0000 | [diff] [blame] | 456 |         self.assertIn(vi[3], ("alpha", "beta", "candidate", "final")) | 
| Ezio Melotti | e961593 | 2010-01-24 19:26:24 +0000 | [diff] [blame] | 457 |         self.assertIsInstance(vi[4], int) | 
 | 458 |         self.assertIsInstance(vi.major, int) | 
 | 459 |         self.assertIsInstance(vi.minor, int) | 
 | 460 |         self.assertIsInstance(vi.micro, int) | 
| Ezio Melotti | b58e0bd | 2010-01-23 15:40:09 +0000 | [diff] [blame] | 461 |         self.assertIn(vi.releaselevel, ("alpha", "beta", "candidate", "final")) | 
| Ezio Melotti | e961593 | 2010-01-24 19:26:24 +0000 | [diff] [blame] | 462 |         self.assertIsInstance(vi.serial, int) | 
| Eric Smith | 0e5b562 | 2009-02-06 01:32:42 +0000 | [diff] [blame] | 463 |         self.assertEqual(vi[0], vi.major) | 
 | 464 |         self.assertEqual(vi[1], vi.minor) | 
 | 465 |         self.assertEqual(vi[2], vi.micro) | 
 | 466 |         self.assertEqual(vi[3], vi.releaselevel) | 
 | 467 |         self.assertEqual(vi[4], vi.serial) | 
| Benjamin Peterson | c9c0f20 | 2009-06-30 23:06:06 +0000 | [diff] [blame] | 468 |         self.assertTrue(vi > (1,0,0)) | 
| Mark Dickinson | 55d110c | 2009-10-24 14:04:48 +0000 | [diff] [blame] | 469 |         self.assertIsInstance(sys.float_repr_style, str) | 
| Benjamin Peterson | 577473f | 2010-01-19 00:09:57 +0000 | [diff] [blame] | 470 |         self.assertIn(sys.float_repr_style, ('short', 'legacy')) | 
| Barry Warsaw | 8cf4eae | 2010-10-16 01:04:07 +0000 | [diff] [blame] | 471 |         if not sys.platform.startswith('win'): | 
 | 472 |             self.assertIsInstance(sys.abiflags, str) | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 473 |  | 
| Thomas Wouters | 49fd7fa | 2006-04-21 10:40:58 +0000 | [diff] [blame] | 474 |     def test_43581(self): | 
| Georg Brandl | 0312494 | 2008-06-10 15:50:56 +0000 | [diff] [blame] | 475 |         # Can't use sys.stdout, as this is a StringIO object when | 
| Thomas Wouters | 49fd7fa | 2006-04-21 10:40:58 +0000 | [diff] [blame] | 476 |         # the test runs under regrtest. | 
| Guido van Rossum | e61fd5b | 2007-07-11 12:20:59 +0000 | [diff] [blame] | 477 |         self.assertEqual(sys.__stdout__.encoding, sys.__stderr__.encoding) | 
| Thomas Wouters | 49fd7fa | 2006-04-21 10:40:58 +0000 | [diff] [blame] | 478 |  | 
| Georg Brandl | 66a796e | 2006-12-19 20:50:34 +0000 | [diff] [blame] | 479 |     def test_intern(self): | 
| Georg Brandl | 0352655 | 2009-10-27 23:00:28 +0000 | [diff] [blame] | 480 |         global numruns | 
 | 481 |         numruns += 1 | 
| Georg Brandl | 66a796e | 2006-12-19 20:50:34 +0000 | [diff] [blame] | 482 |         self.assertRaises(TypeError, sys.intern) | 
| Georg Brandl | 0352655 | 2009-10-27 23:00:28 +0000 | [diff] [blame] | 483 |         s = "never interned before" + str(numruns) | 
| Benjamin Peterson | c9c0f20 | 2009-06-30 23:06:06 +0000 | [diff] [blame] | 484 |         self.assertTrue(sys.intern(s) is s) | 
| Georg Brandl | 66a796e | 2006-12-19 20:50:34 +0000 | [diff] [blame] | 485 |         s2 = s.swapcase().swapcase() | 
| Benjamin Peterson | c9c0f20 | 2009-06-30 23:06:06 +0000 | [diff] [blame] | 486 |         self.assertTrue(sys.intern(s2) is s) | 
| Georg Brandl | 66a796e | 2006-12-19 20:50:34 +0000 | [diff] [blame] | 487 |  | 
 | 488 |         # Subclasses of string can't be interned, because they | 
 | 489 |         # provide too much opportunity for insane things to happen. | 
 | 490 |         # We don't want them in the interned dict and if they aren't | 
 | 491 |         # actually interned, we don't want to create the appearance | 
 | 492 |         # that they are by allowing intern() to succeeed. | 
| Guido van Rossum | 98297ee | 2007-11-06 21:34:58 +0000 | [diff] [blame] | 493 |         class S(str): | 
| Georg Brandl | 66a796e | 2006-12-19 20:50:34 +0000 | [diff] [blame] | 494 |             def __hash__(self): | 
 | 495 |                 return 123 | 
 | 496 |  | 
| Guido van Rossum | 98297ee | 2007-11-06 21:34:58 +0000 | [diff] [blame] | 497 |         self.assertRaises(TypeError, sys.intern, S("abc")) | 
| Martin v. Löwis | 5b22213 | 2007-06-10 09:51:05 +0000 | [diff] [blame] | 498 |  | 
| Christian Heimes | d32ed6f | 2008-01-14 18:49:24 +0000 | [diff] [blame] | 499 |     def test_sys_flags(self): | 
| Benjamin Peterson | c9c0f20 | 2009-06-30 23:06:06 +0000 | [diff] [blame] | 500 |         self.assertTrue(sys.flags) | 
| Christian Heimes | d32ed6f | 2008-01-14 18:49:24 +0000 | [diff] [blame] | 501 |         attrs = ("debug", "division_warning", | 
 | 502 |                  "inspect", "interactive", "optimize", "dont_write_bytecode", | 
| Georg Brandl | c5e88d5 | 2008-06-06 09:02:07 +0000 | [diff] [blame] | 503 |                  "no_user_site", "no_site", "ignore_environment", "verbose", | 
| Benjamin Peterson | 699adb9 | 2008-05-08 22:27:58 +0000 | [diff] [blame] | 504 |                  "bytes_warning") | 
| Christian Heimes | d32ed6f | 2008-01-14 18:49:24 +0000 | [diff] [blame] | 505 |         for attr in attrs: | 
| Benjamin Peterson | c9c0f20 | 2009-06-30 23:06:06 +0000 | [diff] [blame] | 506 |             self.assertTrue(hasattr(sys.flags, attr), attr) | 
| Christian Heimes | d32ed6f | 2008-01-14 18:49:24 +0000 | [diff] [blame] | 507 |             self.assertEqual(type(getattr(sys.flags, attr)), int, attr) | 
| Benjamin Peterson | c9c0f20 | 2009-06-30 23:06:06 +0000 | [diff] [blame] | 508 |         self.assertTrue(repr(sys.flags)) | 
| Georg Brandl | c5e88d5 | 2008-06-06 09:02:07 +0000 | [diff] [blame] | 509 |         self.assertEqual(len(sys.flags), len(attrs)) | 
| Christian Heimes | d32ed6f | 2008-01-14 18:49:24 +0000 | [diff] [blame] | 510 |  | 
| Christian Heimes | 15ebc88 | 2008-02-04 18:48:49 +0000 | [diff] [blame] | 511 |     def test_clear_type_cache(self): | 
 | 512 |         sys._clear_type_cache() | 
 | 513 |  | 
| Martin v. Löwis | 0f59989 | 2008-06-02 11:13:03 +0000 | [diff] [blame] | 514 |     def test_ioencoding(self): | 
| Martin v. Löwis | 0f59989 | 2008-06-02 11:13:03 +0000 | [diff] [blame] | 515 |         env = dict(os.environ) | 
 | 516 |  | 
 | 517 |         # Test character: cent sign, encoded as 0x4A (ASCII J) in CP424, | 
 | 518 |         # not representable in ASCII. | 
 | 519 |  | 
 | 520 |         env["PYTHONIOENCODING"] = "cp424" | 
 | 521 |         p = subprocess.Popen([sys.executable, "-c", 'print(chr(0xa2))'], | 
 | 522 |                              stdout = subprocess.PIPE, env=env) | 
| Antoine Pitrou | 836f0e2 | 2009-10-27 19:25:57 +0000 | [diff] [blame] | 523 |         out = p.communicate()[0].strip() | 
| Martin v. Löwis | 0f59989 | 2008-06-02 11:13:03 +0000 | [diff] [blame] | 524 |         self.assertEqual(out, "\xa2\n".encode("cp424")) | 
 | 525 |  | 
 | 526 |         env["PYTHONIOENCODING"] = "ascii:replace" | 
 | 527 |         p = subprocess.Popen([sys.executable, "-c", 'print(chr(0xa2))'], | 
 | 528 |                              stdout = subprocess.PIPE, env=env) | 
| Antoine Pitrou | 836f0e2 | 2009-10-27 19:25:57 +0000 | [diff] [blame] | 529 |         out = p.communicate()[0].strip() | 
| Martin v. Löwis | 0f59989 | 2008-06-02 11:13:03 +0000 | [diff] [blame] | 530 |         self.assertEqual(out, b'?') | 
 | 531 |  | 
| Victor Stinner | 171ba05 | 2010-03-12 14:20:59 +0000 | [diff] [blame] | 532 |     def test_executable(self): | 
 | 533 |         # Issue #7774: Ensure that sys.executable is an empty string if argv[0] | 
 | 534 |         # has been set to an non existent program name and Python is unable to | 
 | 535 |         # retrieve the real program name | 
| Florent Xicluna | 02ea12b2 | 2010-07-28 16:39:41 +0000 | [diff] [blame] | 536 |  | 
| Victor Stinner | 171ba05 | 2010-03-12 14:20:59 +0000 | [diff] [blame] | 537 |         # For a normal installation, it should work without 'cwd' | 
 | 538 |         # argument. For test runs in the build directory, see #7774. | 
 | 539 |         python_dir = os.path.dirname(os.path.realpath(sys.executable)) | 
 | 540 |         p = subprocess.Popen( | 
 | 541 |             ["nonexistent", "-c", | 
 | 542 |              'import sys; print(sys.executable.encode("ascii", "backslashreplace"))'], | 
 | 543 |             executable=sys.executable, stdout=subprocess.PIPE, cwd=python_dir) | 
 | 544 |         stdout = p.communicate()[0] | 
 | 545 |         executable = stdout.strip().decode("ASCII") | 
 | 546 |         p.wait() | 
 | 547 |         self.assertIn(executable, ["b''", repr(sys.executable.encode("ascii", "backslashreplace"))]) | 
 | 548 |  | 
| Victor Stinner | 85ca2af | 2010-08-19 11:23:47 +0000 | [diff] [blame] | 549 |     def check_fsencoding(self, fs_encoding, expected=None): | 
 | 550 |         self.assertIsNotNone(fs_encoding) | 
 | 551 |         codecs.lookup(fs_encoding) | 
 | 552 |         if expected: | 
 | 553 |             self.assertEqual(fs_encoding, expected) | 
 | 554 |  | 
| Victor Stinner | c732e3b | 2010-08-18 22:44:15 +0000 | [diff] [blame] | 555 |     def test_getfilesystemencoding(self): | 
| Victor Stinner | c732e3b | 2010-08-18 22:44:15 +0000 | [diff] [blame] | 556 |         fs_encoding = sys.getfilesystemencoding() | 
| Victor Stinner | 85ca2af | 2010-08-19 11:23:47 +0000 | [diff] [blame] | 557 |         if sys.platform == 'darwin': | 
 | 558 |             expected = 'utf-8' | 
 | 559 |         elif sys.platform == 'win32': | 
 | 560 |             expected = 'mbcs' | 
 | 561 |         else: | 
 | 562 |             expected = None | 
| Victor Stinner | 9943524 | 2010-08-19 11:30:09 +0000 | [diff] [blame] | 563 |         self.check_fsencoding(fs_encoding, expected) | 
| Victor Stinner | c732e3b | 2010-08-18 22:44:15 +0000 | [diff] [blame] | 564 |  | 
| Martin v. Löwis | 0f59989 | 2008-06-02 11:13:03 +0000 | [diff] [blame] | 565 |  | 
| Martin v. Löwis | 00709aa | 2008-06-04 14:18:43 +0000 | [diff] [blame] | 566 | class SizeofTest(unittest.TestCase): | 
 | 567 |  | 
| Robert Schuppenies | fbe94c5 | 2008-07-14 10:13:31 +0000 | [diff] [blame] | 568 |     TPFLAGS_HAVE_GC = 1<<14 | 
 | 569 |     TPFLAGS_HEAPTYPE = 1<<9 | 
 | 570 |  | 
| Martin v. Löwis | 00709aa | 2008-06-04 14:18:43 +0000 | [diff] [blame] | 571 |     def setUp(self): | 
| Victor Stinner | baab9d0 | 2010-12-28 13:33:43 +0000 | [diff] [blame] | 572 |         self.c = len(struct.pack('c', b' ')) | 
| Robert Schuppenies | 4d45bfe | 2008-06-28 23:58:47 +0000 | [diff] [blame] | 573 |         self.H = len(struct.pack('H', 0)) | 
| Martin v. Löwis | 00709aa | 2008-06-04 14:18:43 +0000 | [diff] [blame] | 574 |         self.i = len(struct.pack('i', 0)) | 
 | 575 |         self.l = len(struct.pack('l', 0)) | 
| Robert Schuppenies | 4d45bfe | 2008-06-28 23:58:47 +0000 | [diff] [blame] | 576 |         self.P = len(struct.pack('P', 0)) | 
 | 577 |         # due to missing size_t information from struct, it is assumed that | 
 | 578 |         # sizeof(Py_ssize_t) = sizeof(void*) | 
 | 579 |         self.header = 'PP' | 
| Robert Schuppenies | fbe94c5 | 2008-07-14 10:13:31 +0000 | [diff] [blame] | 580 |         self.vheader = self.header + 'P' | 
| Martin v. Löwis | 00709aa | 2008-06-04 14:18:43 +0000 | [diff] [blame] | 581 |         if hasattr(sys, "gettotalrefcount"): | 
| Robert Schuppenies | 4d45bfe | 2008-06-28 23:58:47 +0000 | [diff] [blame] | 582 |             self.header += '2P' | 
| Robert Schuppenies | fbe94c5 | 2008-07-14 10:13:31 +0000 | [diff] [blame] | 583 |             self.vheader += '2P' | 
| Mark Dickinson | bd79264 | 2009-03-18 20:06:12 +0000 | [diff] [blame] | 584 |         self.longdigit = sys.int_info.sizeof_digit | 
| Robert Schuppenies | fbe94c5 | 2008-07-14 10:13:31 +0000 | [diff] [blame] | 585 |         import _testcapi | 
 | 586 |         self.gc_headsize = _testcapi.SIZEOF_PYGC_HEAD | 
| Martin v. Löwis | 00709aa | 2008-06-04 14:18:43 +0000 | [diff] [blame] | 587 |         self.file = open(test.support.TESTFN, 'wb') | 
 | 588 |  | 
 | 589 |     def tearDown(self): | 
 | 590 |         self.file.close() | 
 | 591 |         test.support.unlink(test.support.TESTFN) | 
 | 592 |  | 
| Robert Schuppenies | fbe94c5 | 2008-07-14 10:13:31 +0000 | [diff] [blame] | 593 |     def check_sizeof(self, o, size): | 
| Martin v. Löwis | 00709aa | 2008-06-04 14:18:43 +0000 | [diff] [blame] | 594 |         result = sys.getsizeof(o) | 
| Robert Schuppenies | fbe94c5 | 2008-07-14 10:13:31 +0000 | [diff] [blame] | 595 |         # add GC header size | 
 | 596 |         if ((type(o) == type) and (o.__flags__ & self.TPFLAGS_HEAPTYPE) or\ | 
 | 597 |            ((type(o) != type) and (type(o).__flags__ & self.TPFLAGS_HAVE_GC))): | 
 | 598 |             size += self.gc_headsize | 
 | 599 |         msg = 'wrong size for %s: got %d, expected %d' \ | 
 | 600 |               % (type(o), result, size) | 
 | 601 |         self.assertEqual(result, size, msg) | 
| Martin v. Löwis | 00709aa | 2008-06-04 14:18:43 +0000 | [diff] [blame] | 602 |  | 
| Robert Schuppenies | 4d45bfe | 2008-06-28 23:58:47 +0000 | [diff] [blame] | 603 |     def calcsize(self, fmt): | 
 | 604 |         """Wrapper around struct.calcsize which enforces the alignment of the | 
 | 605 |         end of a structure to the alignment requirement of pointer. | 
| Martin v. Löwis | 00709aa | 2008-06-04 14:18:43 +0000 | [diff] [blame] | 606 |  | 
| Robert Schuppenies | 4d45bfe | 2008-06-28 23:58:47 +0000 | [diff] [blame] | 607 |         Note: This wrapper should only be used if a pointer member is included | 
 | 608 |         and no member with a size larger than a pointer exists. | 
 | 609 |         """ | 
 | 610 |         return struct.calcsize(fmt + '0P') | 
| Martin v. Löwis | 00709aa | 2008-06-04 14:18:43 +0000 | [diff] [blame] | 611 |  | 
| Robert Schuppenies | fbe94c5 | 2008-07-14 10:13:31 +0000 | [diff] [blame] | 612 |     def test_gc_head_size(self): | 
 | 613 |         # Check that the gc header size is added to objects tracked by the gc. | 
| Robert Schuppenies | 4d45bfe | 2008-06-28 23:58:47 +0000 | [diff] [blame] | 614 |         h = self.header | 
| Robert Schuppenies | fbe94c5 | 2008-07-14 10:13:31 +0000 | [diff] [blame] | 615 |         vh = self.vheader | 
| Robert Schuppenies | 4d45bfe | 2008-06-28 23:58:47 +0000 | [diff] [blame] | 616 |         size = self.calcsize | 
| Robert Schuppenies | fbe94c5 | 2008-07-14 10:13:31 +0000 | [diff] [blame] | 617 |         gc_header_size = self.gc_headsize | 
 | 618 |         # bool objects are not gc tracked | 
| Mark Dickinson | bd79264 | 2009-03-18 20:06:12 +0000 | [diff] [blame] | 619 |         self.assertEqual(sys.getsizeof(True), size(vh) + self.longdigit) | 
| Robert Schuppenies | fbe94c5 | 2008-07-14 10:13:31 +0000 | [diff] [blame] | 620 |         # but lists are | 
 | 621 |         self.assertEqual(sys.getsizeof([]), size(vh + 'PP') + gc_header_size) | 
 | 622 |  | 
 | 623 |     def test_default(self): | 
 | 624 |         h = self.header | 
 | 625 |         vh = self.vheader | 
 | 626 |         size = self.calcsize | 
| Mark Dickinson | bd79264 | 2009-03-18 20:06:12 +0000 | [diff] [blame] | 627 |         self.assertEqual(sys.getsizeof(True), size(vh) + self.longdigit) | 
 | 628 |         self.assertEqual(sys.getsizeof(True, -1), size(vh) + self.longdigit) | 
| Robert Schuppenies | fbe94c5 | 2008-07-14 10:13:31 +0000 | [diff] [blame] | 629 |  | 
 | 630 |     def test_objecttypes(self): | 
 | 631 |         # check all types defined in Objects/ | 
 | 632 |         h = self.header | 
 | 633 |         vh = self.vheader | 
 | 634 |         size = self.calcsize | 
 | 635 |         check = self.check_sizeof | 
 | 636 |         # bool | 
| Mark Dickinson | bd79264 | 2009-03-18 20:06:12 +0000 | [diff] [blame] | 637 |         check(True, size(vh) + self.longdigit) | 
| Robert Schuppenies | fbe94c5 | 2008-07-14 10:13:31 +0000 | [diff] [blame] | 638 |         # buffer | 
 | 639 |         # XXX | 
 | 640 |         # builtin_function_or_method | 
 | 641 |         check(len, size(h + '3P')) | 
 | 642 |         # bytearray | 
 | 643 |         samples = [b'', b'u'*100000] | 
 | 644 |         for sample in samples: | 
 | 645 |             x = bytearray(sample) | 
 | 646 |             check(x, size(vh + 'iPP') + x.__alloc__() * self.c) | 
 | 647 |         # bytearray_iterator | 
 | 648 |         check(iter(bytearray()), size(h + 'PP')) | 
| Martin v. Löwis | 00709aa | 2008-06-04 14:18:43 +0000 | [diff] [blame] | 649 |         # cell | 
 | 650 |         def get_cell(): | 
 | 651 |             x = 42 | 
 | 652 |             def inner(): | 
 | 653 |                 return x | 
 | 654 |             return inner | 
| Robert Schuppenies | fbe94c5 | 2008-07-14 10:13:31 +0000 | [diff] [blame] | 655 |         check(get_cell().__closure__[0], size(h + 'P')) | 
| Robert Schuppenies | 4efb518 | 2008-06-06 11:49:51 +0000 | [diff] [blame] | 656 |         # code | 
| Collin Winter | 4222e9c | 2010-03-18 22:46:40 +0000 | [diff] [blame] | 657 |         check(get_cell().__code__, size(h + '5i8Pi3P')) | 
| Martin v. Löwis | 00709aa | 2008-06-04 14:18:43 +0000 | [diff] [blame] | 658 |         # complex | 
| Robert Schuppenies | fbe94c5 | 2008-07-14 10:13:31 +0000 | [diff] [blame] | 659 |         check(complex(0,1), size(h + '2d')) | 
 | 660 |         # method_descriptor (descriptor object) | 
 | 661 |         check(str.lower, size(h + '2PP')) | 
 | 662 |         # classmethod_descriptor (descriptor object) | 
 | 663 |         # XXX | 
 | 664 |         # member_descriptor (descriptor object) | 
 | 665 |         import datetime | 
 | 666 |         check(datetime.timedelta.days, size(h + '2PP')) | 
 | 667 |         # getset_descriptor (descriptor object) | 
 | 668 |         import collections | 
 | 669 |         check(collections.defaultdict.default_factory, size(h + '2PP')) | 
 | 670 |         # wrapper_descriptor (descriptor object) | 
 | 671 |         check(int.__add__, size(h + '2P2P')) | 
 | 672 |         # method-wrapper (descriptor object) | 
 | 673 |         check({}.__iter__, size(h + '2P')) | 
 | 674 |         # dict | 
 | 675 |         check({}, size(h + '3P2P' + 8*'P2P')) | 
 | 676 |         longdict = {1:1, 2:2, 3:3, 4:4, 5:5, 6:6, 7:7, 8:8} | 
 | 677 |         check(longdict, size(h + '3P2P' + 8*'P2P') + 16*size('P2P')) | 
 | 678 |         # dictionary-keyiterator | 
 | 679 |         check({}.keys(), size(h + 'P')) | 
 | 680 |         # dictionary-valueiterator | 
 | 681 |         check({}.values(), size(h + 'P')) | 
 | 682 |         # dictionary-itemiterator | 
 | 683 |         check({}.items(), size(h + 'P')) | 
 | 684 |         # dictproxy | 
 | 685 |         class C(object): pass | 
 | 686 |         check(C.__dict__, size(h + 'P')) | 
 | 687 |         # BaseException | 
 | 688 |         check(BaseException(), size(h + '5P')) | 
 | 689 |         # UnicodeEncodeError | 
 | 690 |         check(UnicodeEncodeError("", "", 0, 0, ""), size(h + '5P 2P2PP')) | 
 | 691 |         # UnicodeDecodeError | 
 | 692 |         # XXX | 
 | 693 | #        check(UnicodeDecodeError("", "", 0, 0, ""), size(h + '5P2PP')) | 
 | 694 |         # UnicodeTranslateError | 
 | 695 |         check(UnicodeTranslateError("", 0, 1, ""), size(h + '5P 2P2PP')) | 
 | 696 |         # ellipses | 
 | 697 |         check(Ellipsis, size(h + '')) | 
 | 698 |         # EncodingMap | 
 | 699 |         import codecs, encodings.iso8859_3 | 
 | 700 |         x = codecs.charmap_build(encodings.iso8859_3.decoding_table) | 
 | 701 |         check(x, size(h + '32B2iB')) | 
| Martin v. Löwis | 00709aa | 2008-06-04 14:18:43 +0000 | [diff] [blame] | 702 |         # enumerate | 
| Robert Schuppenies | fbe94c5 | 2008-07-14 10:13:31 +0000 | [diff] [blame] | 703 |         check(enumerate([]), size(h + 'l3P')) | 
| Martin v. Löwis | 00709aa | 2008-06-04 14:18:43 +0000 | [diff] [blame] | 704 |         # reverse | 
| Robert Schuppenies | fbe94c5 | 2008-07-14 10:13:31 +0000 | [diff] [blame] | 705 |         check(reversed(''), size(h + 'PP')) | 
| Martin v. Löwis | 00709aa | 2008-06-04 14:18:43 +0000 | [diff] [blame] | 706 |         # float | 
| Robert Schuppenies | fbe94c5 | 2008-07-14 10:13:31 +0000 | [diff] [blame] | 707 |         check(float(0), size(h + 'd')) | 
 | 708 |         # sys.floatinfo | 
 | 709 |         check(sys.float_info, size(vh) + self.P * len(sys.float_info)) | 
 | 710 |         # frame | 
 | 711 |         import inspect | 
 | 712 |         CO_MAXBLOCKS = 20 | 
 | 713 |         x = inspect.currentframe() | 
 | 714 |         ncells = len(x.f_code.co_cellvars) | 
 | 715 |         nfrees = len(x.f_code.co_freevars) | 
 | 716 |         extras = x.f_code.co_stacksize + x.f_code.co_nlocals +\ | 
 | 717 |                   ncells + nfrees - 1 | 
 | 718 |         check(x, size(vh + '12P3i' + CO_MAXBLOCKS*'3i' + 'P' + extras*'P')) | 
| Martin v. Löwis | 00709aa | 2008-06-04 14:18:43 +0000 | [diff] [blame] | 719 |         # function | 
 | 720 |         def func(): pass | 
| Robert Schuppenies | fbe94c5 | 2008-07-14 10:13:31 +0000 | [diff] [blame] | 721 |         check(func, size(h + '11P')) | 
| Martin v. Löwis | 00709aa | 2008-06-04 14:18:43 +0000 | [diff] [blame] | 722 |         class c(): | 
 | 723 |             @staticmethod | 
 | 724 |             def foo(): | 
 | 725 |                 pass | 
 | 726 |             @classmethod | 
 | 727 |             def bar(cls): | 
 | 728 |                 pass | 
 | 729 |             # staticmethod | 
| Robert Schuppenies | fbe94c5 | 2008-07-14 10:13:31 +0000 | [diff] [blame] | 730 |             check(foo, size(h + 'P')) | 
| Martin v. Löwis | 00709aa | 2008-06-04 14:18:43 +0000 | [diff] [blame] | 731 |             # classmethod | 
| Robert Schuppenies | fbe94c5 | 2008-07-14 10:13:31 +0000 | [diff] [blame] | 732 |             check(bar, size(h + 'P')) | 
| Martin v. Löwis | 00709aa | 2008-06-04 14:18:43 +0000 | [diff] [blame] | 733 |         # generator | 
 | 734 |         def get_gen(): yield 1 | 
| Robert Schuppenies | fbe94c5 | 2008-07-14 10:13:31 +0000 | [diff] [blame] | 735 |         check(get_gen(), size(h + 'Pi2P')) | 
 | 736 |         # iterator | 
 | 737 |         check(iter('abc'), size(h + 'lP')) | 
 | 738 |         # callable-iterator | 
 | 739 |         import re | 
 | 740 |         check(re.finditer('',''), size(h + '2P')) | 
 | 741 |         # list | 
 | 742 |         samples = [[], [1,2,3], ['1', '2', '3']] | 
 | 743 |         for sample in samples: | 
 | 744 |             check(sample, size(vh + 'PP') + len(sample)*self.P) | 
 | 745 |         # sortwrapper (list) | 
 | 746 |         # XXX | 
 | 747 |         # cmpwrapper (list) | 
 | 748 |         # XXX | 
 | 749 |         # listiterator (list) | 
 | 750 |         check(iter([]), size(h + 'lP')) | 
 | 751 |         # listreverseiterator (list) | 
 | 752 |         check(reversed([]), size(h + 'lP')) | 
 | 753 |         # long | 
 | 754 |         check(0, size(vh)) | 
| Mark Dickinson | bd79264 | 2009-03-18 20:06:12 +0000 | [diff] [blame] | 755 |         check(1, size(vh) + self.longdigit) | 
 | 756 |         check(-1, size(vh) + self.longdigit) | 
 | 757 |         PyLong_BASE = 2**sys.int_info.bits_per_digit | 
| Benjamin Peterson | ef3e4c2 | 2009-04-11 19:48:14 +0000 | [diff] [blame] | 758 |         check(int(PyLong_BASE), size(vh) + 2*self.longdigit) | 
 | 759 |         check(int(PyLong_BASE**2-1), size(vh) + 2*self.longdigit) | 
 | 760 |         check(int(PyLong_BASE**2), size(vh) + 3*self.longdigit) | 
| Robert Schuppenies | fbe94c5 | 2008-07-14 10:13:31 +0000 | [diff] [blame] | 761 |         # memory | 
| Antoine Pitrou | aeb6cee | 2010-11-04 20:30:33 +0000 | [diff] [blame] | 762 |         check(memoryview(b''), size(h + 'PP2P2i7P')) | 
| Martin v. Löwis | 00709aa | 2008-06-04 14:18:43 +0000 | [diff] [blame] | 763 |         # module | 
| Robert Schuppenies | fbe94c5 | 2008-07-14 10:13:31 +0000 | [diff] [blame] | 764 |         check(unittest, size(h + '3P')) | 
 | 765 |         # None | 
 | 766 |         check(None, size(h + '')) | 
 | 767 |         # NotImplementedType | 
 | 768 |         check(NotImplemented, size(h)) | 
 | 769 |         # object | 
 | 770 |         check(object(), size(h + '')) | 
 | 771 |         # property (descriptor object) | 
 | 772 |         class C(object): | 
 | 773 |             def getx(self): return self.__x | 
 | 774 |             def setx(self, value): self.__x = value | 
 | 775 |             def delx(self): del self.__x | 
 | 776 |             x = property(getx, setx, delx, "") | 
 | 777 |             check(x, size(h + '4Pi')) | 
| Benjamin Peterson | b173f78 | 2009-05-05 22:31:58 +0000 | [diff] [blame] | 778 |         # PyCapsule | 
| Robert Schuppenies | fbe94c5 | 2008-07-14 10:13:31 +0000 | [diff] [blame] | 779 |         # XXX | 
 | 780 |         # rangeiterator | 
 | 781 |         check(iter(range(1)), size(h + '4l')) | 
 | 782 |         # reverse | 
 | 783 |         check(reversed(''), size(h + 'PP')) | 
| Martin v. Löwis | 00709aa | 2008-06-04 14:18:43 +0000 | [diff] [blame] | 784 |         # range | 
| Nick Coghlan | 37ee850 | 2010-12-03 14:26:13 +0000 | [diff] [blame] | 785 |         check(range(1), size(h + '4P')) | 
 | 786 |         check(range(66000), size(h + '4P')) | 
| Robert Schuppenies | fbe94c5 | 2008-07-14 10:13:31 +0000 | [diff] [blame] | 787 |         # set | 
 | 788 |         # frozenset | 
 | 789 |         PySet_MINSIZE = 8 | 
 | 790 |         samples = [[], range(10), range(50)] | 
 | 791 |         s = size(h + '3P2P' + PySet_MINSIZE*'lP' + 'lP') | 
 | 792 |         for sample in samples: | 
 | 793 |             minused = len(sample) | 
 | 794 |             if minused == 0: tmp = 1 | 
 | 795 |             # the computation of minused is actually a bit more complicated | 
 | 796 |             # but this suffices for the sizeof test | 
 | 797 |             minused = minused*2 | 
 | 798 |             newsize = PySet_MINSIZE | 
 | 799 |             while newsize <= minused: | 
 | 800 |                 newsize = newsize << 1 | 
 | 801 |             if newsize <= 8: | 
 | 802 |                 check(set(sample), s) | 
 | 803 |                 check(frozenset(sample), s) | 
 | 804 |             else: | 
 | 805 |                 check(set(sample), s + newsize*struct.calcsize('lP')) | 
 | 806 |                 check(frozenset(sample), s + newsize*struct.calcsize('lP')) | 
 | 807 |         # setiterator | 
 | 808 |         check(iter(set()), size(h + 'P3P')) | 
| Martin v. Löwis | 00709aa | 2008-06-04 14:18:43 +0000 | [diff] [blame] | 809 |         # slice | 
| Robert Schuppenies | fbe94c5 | 2008-07-14 10:13:31 +0000 | [diff] [blame] | 810 |         check(slice(0), size(h + '3P')) | 
 | 811 |         # super | 
 | 812 |         check(super(int), size(h + '3P')) | 
 | 813 |         # tuple | 
 | 814 |         check((), size(vh)) | 
 | 815 |         check((1,2,3), size(vh) + 3*self.P) | 
 | 816 |         # type | 
 | 817 |         # (PyTypeObject + PyNumberMethods + PyMappingMethods + | 
 | 818 |         #  PySequenceMethods + PyBufferProcs) | 
 | 819 |         s = size(vh + 'P2P15Pl4PP9PP11PI') + size('16Pi17P 3P 10P 2P 2P') | 
 | 820 |         check(int, s) | 
 | 821 |         # class | 
 | 822 |         class newstyleclass(object): pass | 
 | 823 |         check(newstyleclass, s) | 
| Georg Brandl | c28e1fa | 2008-06-10 19:20:26 +0000 | [diff] [blame] | 824 |         # unicode | 
| Benjamin Peterson | b49d994 | 2008-06-17 12:44:04 +0000 | [diff] [blame] | 825 |         usize = len('\0'.encode('unicode-internal')) | 
| Georg Brandl | c28e1fa | 2008-06-10 19:20:26 +0000 | [diff] [blame] | 826 |         samples = ['', '1'*100] | 
 | 827 |         # we need to test for both sizes, because we don't know if the string | 
 | 828 |         # has been cached | 
 | 829 |         for s in samples: | 
| Antoine Pitrou | fbb1c61 | 2010-10-23 17:37:54 +0000 | [diff] [blame] | 830 |             basicsize =  size(h + 'PPPiP') + usize * (len(s) + 1) | 
| Robert Schuppenies | fbe94c5 | 2008-07-14 10:13:31 +0000 | [diff] [blame] | 831 |             check(s, basicsize) | 
 | 832 |         # weakref | 
 | 833 |         import weakref | 
 | 834 |         check(weakref.ref(int), size(h + '2Pl2P')) | 
 | 835 |         # weakproxy | 
 | 836 |         # XXX | 
 | 837 |         # weakcallableproxy | 
 | 838 |         check(weakref.proxy(int), size(h + '2Pl2P')) | 
| Martin v. Löwis | 00709aa | 2008-06-04 14:18:43 +0000 | [diff] [blame] | 839 |  | 
| Robert Schuppenies | fbe94c5 | 2008-07-14 10:13:31 +0000 | [diff] [blame] | 840 |     def test_pythontypes(self): | 
 | 841 |         # check all types defined in Python/ | 
 | 842 |         h = self.header | 
 | 843 |         vh = self.vheader | 
 | 844 |         size = self.calcsize | 
 | 845 |         check = self.check_sizeof | 
 | 846 |         # _ast.AST | 
 | 847 |         import _ast | 
 | 848 |         check(_ast.AST(), size(h + '')) | 
 | 849 |         # imp.NullImporter | 
 | 850 |         import imp | 
 | 851 |         check(imp.NullImporter(self.file.name), size(h + '')) | 
 | 852 |         try: | 
 | 853 |             raise TypeError | 
 | 854 |         except TypeError: | 
 | 855 |             tb = sys.exc_info()[2] | 
 | 856 |             # traceback | 
 | 857 |             if tb != None: | 
 | 858 |                 check(tb, size(h + '2P2i')) | 
 | 859 |         # symtable entry | 
 | 860 |         # XXX | 
 | 861 |         # sys.flags | 
 | 862 |         check(sys.flags, size(vh) + self.P * len(sys.flags)) | 
| Martin v. Löwis | 00709aa | 2008-06-04 14:18:43 +0000 | [diff] [blame] | 863 |  | 
 | 864 |  | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 865 | def test_main(): | 
| Martin v. Löwis | 00709aa | 2008-06-04 14:18:43 +0000 | [diff] [blame] | 866 |     test.support.run_unittest(SysModuleTest, SizeofTest) | 
| Walter Dörwald | c350246 | 2003-02-03 23:03:49 +0000 | [diff] [blame] | 867 |  | 
 | 868 | if __name__ == "__main__": | 
 | 869 |     test_main() |