Skip to content

Commit e64c747

Browse files
committed
Fix edge case where node.end_lineno is None
1 parent d3b1874 commit e64c747

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

src/tap/utils.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -264,6 +264,11 @@ def get_subsequent_assign_lines(source_cls: str) -> Set[int]:
264264
assign_lines = set()
265265
for node in cls_body.body:
266266
if isinstance(node, (ast.Assign, ast.AnnAssign)):
267+
# Check if the end line number is found
268+
if node.end_lineno is None:
269+
warnings.warn(parse_warning)
270+
return set()
271+
267272
# Get line number of assign statement excluding the first line (and minus 1 for the if statement)
268273
assign_lines |= set(range(node.lineno, node.end_lineno))
269274

0 commit comments

Comments
 (0)