gh-98442: Fix source location of return in a with block

[iritkatriel]

Fixes #98442.

• Issue: Source location of return instruction in a with block is incorrect #98442

[markshannon]

This looks like it ought to be correct. PEP 626 says that return should be traced, so the RETURN_VALUE should have the location of the return statement.

[iritkatriel]

This looks like it ought to be correct. PEP 626 says that return should be traced, so the RETURN_VALUE should have the location of the return statement.

Do you mean that the current output is correct? It shows the whole with-block as the location of the RETURN_VALUE.

[iritkatriel]

This issue seems to have been created semi-deliberately here: https://github.com/python/cpython/pull/26513/files#diff-ebc983d9f91e5bcf73500e377ac65e85863c4f77fd5b6b6caf4fcdf7c0f0b057L1868

The location of the return AST node is overwritten in the compiler_unwind_fblock function.