code: fix IndexError crash in getstatementrange_ast

Fix pytest-dev#11953.
flying-sheep · Apr 9, 2024 · a6a176b · a6a176b
1 parent 90c73d1
commit a6a176b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/changelog/11953.bugfix.rst b/changelog/11953.bugfix.rst
@@ -0,0 +1 @@
+Fix an ``IndexError`` crash raising from ``getstatementrange_ast``.
diff --git a/src/_pytest/_code/source.py b/src/_pytest/_code/source.py
@@ -197,7 +197,9 @@ def getstatementrange_ast(
         # by using the BlockFinder helper used which inspect.getsource() uses itself.
         block_finder = inspect.BlockFinder()
         # 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])
         try:
             for tok in tokenize.generate_tokens(lambda: next(it)):