|
| 1 | +""" |
| 2 | +Tests on PyConfig API (PEP 587). |
| 3 | +""" |
| 4 | +import os |
| 5 | +import sys |
| 6 | +import unittest |
| 7 | +from test import support |
| 8 | +try: |
| 9 | + import _testcapi |
| 10 | +except ImportError: |
| 11 | + _testcapi = None |
| 12 | + |
| 13 | + |
| 14 | +@unittest.skipIf(_testcapi is None, 'need _testcapi') |
| 15 | +class CAPITests(unittest.TestCase): |
| 16 | + def check_config_get(self, get_func): |
| 17 | + # write_bytecode is read from sys.dont_write_bytecode as int |
| 18 | + with support.swap_attr(sys, "dont_write_bytecode", 0): |
| 19 | + self.assertEqual(get_func('write_bytecode'), 1) |
| 20 | + with support.swap_attr(sys, "dont_write_bytecode", "yes"): |
| 21 | + self.assertEqual(get_func('write_bytecode'), 0) |
| 22 | + with support.swap_attr(sys, "dont_write_bytecode", []): |
| 23 | + self.assertEqual(get_func('write_bytecode'), 1) |
| 24 | + |
| 25 | + # non-existent config option name |
| 26 | + NONEXISTENT_KEY = 'NONEXISTENT_KEY' |
| 27 | + err_msg = f'unknown config option name: {NONEXISTENT_KEY}' |
| 28 | + with self.assertRaisesRegex(ValueError, err_msg): |
| 29 | + get_func('NONEXISTENT_KEY') |
| 30 | + |
| 31 | + def test_config_get(self): |
| 32 | + config_get = _testcapi.config_get |
| 33 | + |
| 34 | + self.check_config_get(config_get) |
| 35 | + |
| 36 | + for name, config_type, expected in ( |
| 37 | + ('verbose', int, sys.flags.verbose), # PyConfig_MEMBER_INT |
| 38 | + ('isolated', int, sys.flags.isolated), # PyConfig_MEMBER_UINT |
| 39 | + ('platlibdir', str, sys.platlibdir), # PyConfig_MEMBER_WSTR |
| 40 | + ('argv', list, sys.argv), # PyConfig_MEMBER_WSTR_LIST |
| 41 | + ('xoptions', dict, sys._xoptions), # xoptions dict |
| 42 | + ): |
| 43 | + with self.subTest(name=name): |
| 44 | + value = config_get(name) |
| 45 | + self.assertEqual(type(value), config_type) |
| 46 | + self.assertEqual(value, expected) |
| 47 | + |
| 48 | + # PyConfig_MEMBER_ULONG type |
| 49 | + hash_seed = config_get('hash_seed') |
| 50 | + self.assertIsInstance(hash_seed, int) |
| 51 | + self.assertGreaterEqual(hash_seed, 0) |
| 52 | + |
| 53 | + # PyConfig_MEMBER_WSTR_OPT type |
| 54 | + if 'PYTHONDUMPREFSFILE' not in os.environ: |
| 55 | + self.assertIsNone(config_get('dump_refs_file')) |
| 56 | + |
| 57 | + # attributes read from sys |
| 58 | + value_str = "TEST_MARKER_STR" |
| 59 | + value_list = ["TEST_MARKER_STRLIST"] |
| 60 | + value_dict = {"x": "value", "y": True} |
| 61 | + for name, sys_name, value in ( |
| 62 | + ("base_exec_prefix", None, value_str), |
| 63 | + ("base_prefix", None, value_str), |
| 64 | + ("exec_prefix", None, value_str), |
| 65 | + ("executable", None, value_str), |
| 66 | + ("platlibdir", None, value_str), |
| 67 | + ("prefix", None, value_str), |
| 68 | + ("pycache_prefix", None, value_str), |
| 69 | + ("base_executable", "_base_executable", value_str), |
| 70 | + ("stdlib_dir", "_stdlib_dir", value_str), |
| 71 | + ("argv", None, value_list), |
| 72 | + ("orig_argv", None, value_list), |
| 73 | + ("warnoptions", None, value_list), |
| 74 | + ("module_search_paths", "path", value_list), |
| 75 | + ("xoptions", "_xoptions", value_dict), |
| 76 | + ): |
| 77 | + with self.subTest(name=name): |
| 78 | + if sys_name is None: |
| 79 | + sys_name = name |
| 80 | + with support.swap_attr(sys, sys_name, value): |
| 81 | + self.assertEqual(config_get(name), value) |
| 82 | + |
| 83 | + def test_config_getint(self): |
| 84 | + config_getint = _testcapi.config_getint |
| 85 | + |
| 86 | + self.check_config_get(config_getint) |
| 87 | + |
| 88 | + # PyConfig_MEMBER_INT type |
| 89 | + self.assertEqual(config_getint('verbose'), sys.flags.verbose) |
| 90 | + |
| 91 | + # PyConfig_MEMBER_UINT type |
| 92 | + self.assertEqual(config_getint('isolated'), sys.flags.isolated) |
| 93 | + |
| 94 | + # PyConfig_MEMBER_ULONG type |
| 95 | + hash_seed = config_getint('hash_seed') |
| 96 | + self.assertIsInstance(hash_seed, int) |
| 97 | + self.assertGreaterEqual(hash_seed, 0) |
| 98 | + |
| 99 | + # platlibdir is a str |
| 100 | + with self.assertRaises(TypeError): |
| 101 | + config_getint('platlibdir') |
| 102 | + |
| 103 | + |
| 104 | +if __name__ == "__main__": |
| 105 | + unittest.main() |
0 commit comments