p53209761
/
test_ok2
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #11996 from bluetech/getstatementrange-index-error 

code: fix `IndexError` crash in `getstatementrange_ast`
This commit is contained in:
Ran Benita 2024-02-17 18:59:10 +02:00 committed by GitHub
parent 984478109f 9f13d41d7b
commit aebeb33137
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
changelog/11953.bugfix.rst Normal file
View File

@ -0,0 +1 @@
Fix an ``IndexError`` crash raising from ``getstatementrange_ast``.

4
src/_pytest/_code/source.py
View File

@ -197,7 +197,9 @@ def getstatementrange_ast(
# by using the BlockFinder helper used which inspect.getsource() uses itself. # by using the BlockFinder helper used which inspect.getsource() uses itself.
block_finder = inspect.BlockFinder() block_finder = inspect.BlockFinder()
# If we start with an indented line, put blockfinder to "started" mode. # If we start with an indented line, put blockfinder to "started" mode.
block_finder.started = source.lines[start][0].isspace() block_finder.started = (
bool(source.lines[start]) and source.lines[start][0].isspace()
)
it = ((x + "\n") for x in source.lines[start:end]) it = ((x + "\n") for x in source.lines[start:end])
try: try:
for tok in tokenize.generate_tokens(lambda: next(it)): for tok in tokenize.generate_tokens(lambda: next(it)):