mirror of
https://github.com/python/cpython.git
synced 2025-08-04 00:48:58 +00:00
[3.11] GH-95921: Fix positions for some chained comparisons (GH-96968) (GH-96973)
(cherry picked from commit dfc73b5724
)
Automerge-Triggered-By: GH:brandtbucher
This commit is contained in:
parent
32d80decbf
commit
0e61d2b557
3 changed files with 29 additions and 0 deletions
|
@ -1039,6 +1039,32 @@ if 1:
|
|||
for instr in dis.Bytecode(while_not_chained):
|
||||
self.assertNotEqual(instr.opname, "EXTENDED_ARG")
|
||||
|
||||
def test_compare_positions(self):
|
||||
for opname, op in [
|
||||
("COMPARE_OP", "<"),
|
||||
("COMPARE_OP", "<="),
|
||||
("COMPARE_OP", ">"),
|
||||
("COMPARE_OP", ">="),
|
||||
("CONTAINS_OP", "in"),
|
||||
("CONTAINS_OP", "not in"),
|
||||
("IS_OP", "is"),
|
||||
("IS_OP", "is not"),
|
||||
]:
|
||||
expr = f'a {op} b {op} c'
|
||||
expected_positions = 2 * [(2, 2, 0, len(expr))]
|
||||
for source in [
|
||||
f"\\\n{expr}", f'if \\\n{expr}: x', f"x if \\\n{expr} else y"
|
||||
]:
|
||||
code = compile(source, "<test>", "exec")
|
||||
actual_positions = [
|
||||
instruction.positions
|
||||
for instruction in dis.get_instructions(code)
|
||||
if instruction.opname == opname
|
||||
]
|
||||
with self.subTest(source):
|
||||
self.assertEqual(actual_positions, expected_positions)
|
||||
|
||||
|
||||
@requires_debug_ranges()
|
||||
class TestSourcePositions(unittest.TestCase):
|
||||
# Ensure that compiled code snippets have correct line and column numbers
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue