gh-94808: Cover PyFunction_GetCode, PyFunction_GetGlobals, PyFunction_GetModule (#98158)

This commit is contained in:
Nikita Sobolev 2022-10-16 07:18:59 +03:00 committed by GitHub
parent 660f10248b
commit 7b48d02933
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 74 additions and 0 deletions

View file

@ -895,6 +895,41 @@ class CAPITest(unittest.TestCase):
self.assertEqual(_testcapi.eval_get_func_name(sum), "sum") # c function
self.assertEqual(_testcapi.eval_get_func_name(A), "type")
def test_function_get_code(self):
import types
def some():
pass
code = _testcapi.function_get_code(some)
self.assertIsInstance(code, types.CodeType)
self.assertEqual(code, some.__code__)
with self.assertRaises(SystemError):
_testcapi.function_get_code(None) # not a function
def test_function_get_globals(self):
def some():
pass
globals_ = _testcapi.function_get_globals(some)
self.assertIsInstance(globals_, dict)
self.assertEqual(globals_, some.__globals__)
with self.assertRaises(SystemError):
_testcapi.function_get_globals(None) # not a function
def test_function_get_module(self):
def some():
pass
module = _testcapi.function_get_module(some)
self.assertIsInstance(module, str)
self.assertEqual(module, some.__module__)
with self.assertRaises(SystemError):
_testcapi.function_get_module(None) # not a function
class TestPendingCalls(unittest.TestCase):