https://github.com/python/cpython/commit/4defb58d3812a80d556410077e3391bb81a1c0b7
commit: 4defb58d3812a80d556410077e3391bb81a1c0b7
branch: main
author: larryhastings <[email protected]>
committer: larryhastings <[email protected]>
date: 2024-09-25T17:01:45-07:00
summary:
Add "annotate" SET_FUNCTION_ATTRIBUTE bit to dis. (#124566)
files:
M Lib/dis.py
M Lib/test/test_dis.py
diff --git a/Lib/dis.py b/Lib/dis.py
index f8832b30497822..e87e6a78469ab0 100644
--- a/Lib/dis.py
+++ b/Lib/dis.py
@@ -32,7 +32,7 @@
CONVERT_VALUE = opmap['CONVERT_VALUE']
SET_FUNCTION_ATTRIBUTE = opmap['SET_FUNCTION_ATTRIBUTE']
-FUNCTION_ATTR_FLAGS = ('defaults', 'kwdefaults', 'annotations', 'closure')
+FUNCTION_ATTR_FLAGS = ('defaults', 'kwdefaults', 'annotations', 'closure',
'annotate')
ENTER_EXECUTOR = opmap['ENTER_EXECUTOR']
LOAD_CONST = opmap['LOAD_CONST']
diff --git a/Lib/test/test_dis.py b/Lib/test/test_dis.py
index bccd2182412577..1ee0fbe98914be 100644
--- a/Lib/test/test_dis.py
+++ b/Lib/test/test_dis.py
@@ -380,6 +380,23 @@ def wrap_func_w_kwargs():
RETURN_CONST 3 (None)
"""
+fn_with_annotate_str = """
+def foo(a: int, b: str) -> str:
+ return a * b
+"""
+
+dis_fn_with_annotate_str = """\
+ 0 RESUME 0
+
+ 2 LOAD_CONST 0 (<code object __annotate__ at 0x...,
file "<dis>", line 2>)
+ MAKE_FUNCTION
+ LOAD_CONST 1 (<code object foo at 0x..., file
"<dis>", line 2>)
+ MAKE_FUNCTION
+ SET_FUNCTION_ATTRIBUTE 16 (annotate)
+ STORE_NAME 0 (foo)
+ RETURN_CONST 2 (None)
+"""
+
compound_stmt_str = """\
x = 0
while 1:
@@ -1098,6 +1115,7 @@ def test_disassemble_str(self):
self.do_disassembly_test(expr_str, dis_expr_str)
self.do_disassembly_test(simple_stmt_str, dis_simple_stmt_str)
self.do_disassembly_test(annot_stmt_str, dis_annot_stmt_str)
+ self.do_disassembly_test(fn_with_annotate_str,
dis_fn_with_annotate_str)
self.do_disassembly_test(compound_stmt_str, dis_compound_stmt_str)
def test_disassemble_bytes(self):
_______________________________________________
Python-checkins mailing list -- [email protected]
To unsubscribe send an email to [email protected]
https://mail.python.org/mailman3/lists/python-checkins.python.org/
Member address: [email protected]
