|
| 1 | +""" |
| 2 | +Tests on PyConfig API (PEP 587). |
| 3 | +""" |
| 4 | +import os |
| 5 | +import sys |
| 6 | +import unittest |
| 7 | +try: |
| 8 | + import _testcapi |
| 9 | +except ImportError: |
| 10 | + _testcapi = None |
| 11 | + |
| 12 | + |
| 13 | +@unittest.skipIf(_testcapi is None, 'need _testcapi') |
| 14 | +class CAPITests(unittest.TestCase): |
| 15 | + def check_get_func(self, get_func, wrong_type_key: str): |
| 16 | + with self.assertRaises(TypeError): |
| 17 | + get_func(wrong_type_key) |
| 18 | + |
| 19 | + with self.assertRaisesRegex(ValueError, |
| 20 | + 'unknown config option: NONEXISTENT_KEY'): |
| 21 | + get_func('NONEXISTENT_KEY') |
| 22 | + |
| 23 | + def test_config_getint(self): |
| 24 | + config_getint = _testcapi.config_getint |
| 25 | + # PyConfig_MEMBER_INT type |
| 26 | + self.assertEqual(config_getint('verbose'), sys.flags.verbose) |
| 27 | + |
| 28 | + # PyConfig_MEMBER_UINT type |
| 29 | + self.assertEqual(config_getint('isolated'), sys.flags.isolated) |
| 30 | + |
| 31 | + # PyConfig_MEMBER_ULONG type |
| 32 | + hash_seed = config_getint('hash_seed') |
| 33 | + self.assertIsInstance(hash_seed, int) |
| 34 | + self.assertGreaterEqual(hash_seed, 0) |
| 35 | + |
| 36 | + # platlibdir is a str |
| 37 | + self.check_get_func(config_getint, 'platlibdir') |
| 38 | + |
| 39 | + def test_config_getstr(self): |
| 40 | + config_getstr = _testcapi.config_getstr |
| 41 | + # PyConfig_MEMBER_WSTR type |
| 42 | + self.assertEqual(config_getstr('platlibdir'), sys.platlibdir) |
| 43 | + if 'PYTHONDUMPREFSFILE' not in os.environ: |
| 44 | + # PyConfig_MEMBER_WSTR_OPT type |
| 45 | + self.assertIsNone(config_getstr('dump_refs_file')) |
| 46 | + |
| 47 | + # verbose is an int |
| 48 | + self.check_get_func(config_getstr, 'verbose') |
| 49 | + |
| 50 | + def test_config_getstrlist(self): |
| 51 | + config_getstrlist = _testcapi.config_getstrlist |
| 52 | + self.assertEqual(config_getstrlist('orig_argv'), sys.orig_argv) |
| 53 | + |
| 54 | + # verbose is an int |
| 55 | + self.check_get_func(config_getstrlist, 'verbose') |
| 56 | + |
| 57 | + |
| 58 | +if __name__ == "__main__": |
| 59 | + unittest.main() |
0 commit comments