[3.12] gh-84489: C API: Add tests for Py_BuildValue() (GH-110596) (GH-110680)

(cherry picked from commit 5c6e85480a)

Co-authored-by: Serhiy Storchaka <storchaka@gmail.com>
This commit is contained in:
Miss Islington (bot) 2023-10-11 10:12:39 +02:00 committed by GitHub
parent 9ffef4d797
commit ca971d12ed
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 117 additions and 0 deletions

View file

@ -298,6 +298,86 @@ class CAPITest(unittest.TestCase):
# test _Py_CheckFunctionResult() instead.
self.assertIn('returned a result with an exception set', err)
def test_buildvalue(self):
# Test Py_BuildValue() with object arguments
buildvalue = _testcapi.py_buildvalue
self.assertEqual(buildvalue(''), None)
self.assertEqual(buildvalue('()'), ())
self.assertEqual(buildvalue('[]'), [])
self.assertEqual(buildvalue('{}'), {})
self.assertEqual(buildvalue('()[]{}'), ((), [], {}))
self.assertEqual(buildvalue('O', 1), 1)
self.assertEqual(buildvalue('(O)', 1), (1,))
self.assertEqual(buildvalue('[O]', 1), [1])
self.assertRaises(SystemError, buildvalue, '{O}', 1)
self.assertEqual(buildvalue('OO', 1, 2), (1, 2))
self.assertEqual(buildvalue('(OO)', 1, 2), (1, 2))
self.assertEqual(buildvalue('[OO]', 1, 2), [1, 2])
self.assertEqual(buildvalue('{OO}', 1, 2), {1: 2})
self.assertEqual(buildvalue('{OOOO}', 1, 2, 3, 4), {1: 2, 3: 4})
self.assertEqual(buildvalue('((O))', 1), ((1,),))
self.assertEqual(buildvalue('((OO))', 1, 2), ((1, 2),))
self.assertEqual(buildvalue(' \t,:'), None)
self.assertEqual(buildvalue(' O ', 1), 1)
self.assertEqual(buildvalue('\tO\t', 1), 1)
self.assertEqual(buildvalue('O,O', 1, 2), (1, 2))
self.assertEqual(buildvalue('O, O', 1, 2), (1, 2))
self.assertEqual(buildvalue('O,\tO', 1, 2), (1, 2))
self.assertEqual(buildvalue('O O', 1, 2), (1, 2))
self.assertEqual(buildvalue('O\tO', 1, 2), (1, 2))
self.assertEqual(buildvalue('(O,O)', 1, 2), (1, 2))
self.assertEqual(buildvalue('(O, O)', 1, 2), (1, 2))
self.assertEqual(buildvalue(' ( O O) ', 1, 2), (1, 2))
self.assertEqual(buildvalue('\t(\tO\tO)\t', 1, 2), (1, 2))
self.assertEqual(buildvalue('[O,O]', 1, 2), [1, 2])
self.assertEqual(buildvalue('[O, O]', 1, 2), [1, 2])
self.assertEqual(buildvalue(' [ O O] ', 1, 2), [1, 2])
self.assertEqual(buildvalue('{O:O}', 1, 2), {1: 2})
self.assertEqual(buildvalue('{O:O,O:O}', 1, 2, 3, 4), {1: 2, 3: 4})
self.assertEqual(buildvalue('{O: O, O: O}', 1, 2, 3, 4), {1: 2, 3: 4})
self.assertEqual(buildvalue(' { O O O O} ', 1, 2, 3, 4), {1: 2, 3: 4})
self.assertEqual(buildvalue('\t{\tO\tO\tO\tO}\t', 1, 2, 3, 4), {1: 2, 3: 4})
self.assertRaises(SystemError, buildvalue, 'O', NULL)
self.assertRaises(SystemError, buildvalue, '(O)', NULL)
self.assertRaises(SystemError, buildvalue, '[O]', NULL)
self.assertRaises(SystemError, buildvalue, '{O}', NULL)
self.assertRaises(SystemError, buildvalue, 'OO', 1, NULL)
self.assertRaises(SystemError, buildvalue, 'OO', NULL, 2)
self.assertRaises(SystemError, buildvalue, '(OO)', 1, NULL)
self.assertRaises(SystemError, buildvalue, '(OO)', NULL, 2)
self.assertRaises(SystemError, buildvalue, '[OO]', 1, NULL)
self.assertRaises(SystemError, buildvalue, '[OO]', NULL, 2)
self.assertRaises(SystemError, buildvalue, '{OO}', 1, NULL)
self.assertRaises(SystemError, buildvalue, '{OO}', NULL, 2)
def test_buildvalue_ints(self):
# Test Py_BuildValue() with integer arguments
buildvalue = _testcapi.py_buildvalue_ints
from _testcapi import SHRT_MIN, SHRT_MAX, USHRT_MAX, INT_MIN, INT_MAX, UINT_MAX
self.assertEqual(buildvalue('i', INT_MAX), INT_MAX)
self.assertEqual(buildvalue('i', INT_MIN), INT_MIN)
self.assertEqual(buildvalue('I', UINT_MAX), UINT_MAX)
self.assertEqual(buildvalue('h', SHRT_MAX), SHRT_MAX)
self.assertEqual(buildvalue('h', SHRT_MIN), SHRT_MIN)
self.assertEqual(buildvalue('H', USHRT_MAX), USHRT_MAX)
self.assertEqual(buildvalue('b', 127), 127)
self.assertEqual(buildvalue('b', -128), -128)
self.assertEqual(buildvalue('B', 255), 255)
self.assertEqual(buildvalue('c', ord('A')), b'A')
self.assertEqual(buildvalue('c', 255), b'\xff')
self.assertEqual(buildvalue('c', 256), b'\x00')
self.assertEqual(buildvalue('c', -1), b'\xff')
self.assertEqual(buildvalue('C', 255), chr(255))
self.assertEqual(buildvalue('C', 256), chr(256))
self.assertEqual(buildvalue('C', sys.maxunicode), chr(sys.maxunicode))
self.assertRaises(ValueError, buildvalue, 'C', -1)
self.assertRaises(ValueError, buildvalue, 'C', sys.maxunicode+1)
def test_buildvalue_N(self):
_testcapi.test_buildvalue_N()