[3.11] GH-104405: Add missing PEP 523 checks (GH-104441)

This commit is contained in:
Brandt Bucher 2023-05-12 16:03:47 -07:00 committed by GitHub
parent 4ade7c3369
commit fbb6def08a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 35 additions and 17 deletions

View file

@ -1378,28 +1378,39 @@ SUFFICIENT_TO_DEOPT_AND_SPECIALIZE = 100
class Test_Pep523API(unittest.TestCase):
def do_test(self, func):
calls = []
def do_test(self, func, names):
actual_calls = []
start = SUFFICIENT_TO_DEOPT_AND_SPECIALIZE
count = start + SUFFICIENT_TO_DEOPT_AND_SPECIALIZE
for i in range(count):
if i == start:
_testinternalcapi.set_eval_frame_record(calls)
func()
_testinternalcapi.set_eval_frame_default()
self.assertEqual(len(calls), SUFFICIENT_TO_DEOPT_AND_SPECIALIZE)
for name in calls:
self.assertEqual(name, func.__name__)
try:
for i in range(count):
if i == start:
_testinternalcapi.set_eval_frame_record(actual_calls)
func()
finally:
_testinternalcapi.set_eval_frame_default()
expected_calls = names * SUFFICIENT_TO_DEOPT_AND_SPECIALIZE
self.assertEqual(len(expected_calls), len(actual_calls))
for expected, actual in zip(expected_calls, actual_calls, strict=True):
self.assertEqual(expected, actual)
def test_pep523_with_specialization_simple(self):
def func1():
pass
self.do_test(func1)
def test_inlined_binary_subscr(self):
class C:
def __getitem__(self, other):
return None
def func():
C()[42]
names = ["func", "__getitem__"]
self.do_test(func, names)
def test_pep523_with_specialization_with_default(self):
def func2(x=None):
def test_inlined_call(self):
def inner(x=42):
pass
self.do_test(func2)
def func():
inner()
inner(42)
names = ["func", "inner", "inner"]
self.do_test(func, names)
if __name__ == "__main__":