From bf99bb663940f203bee6d373f6b6f30840831fab Mon Sep 17 00:00:00 2001 From: Ned Batchelder Date: Fri, 4 Feb 2022 18:50:30 -0500 Subject: debug: a little easier to get code samples from tests --- lab/extract_code.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) (limited to 'lab/extract_code.py') diff --git a/lab/extract_code.py b/lab/extract_code.py index ec077af7..c5a9df81 100644 --- a/lab/extract_code.py +++ b/lab/extract_code.py @@ -50,7 +50,10 @@ then you can do:: import sys import textwrap -fname, lineno = sys.argv[1:] +if len(sys.argv) == 2: + fname, lineno = sys.argv[1].split(":") +else: + fname, lineno = sys.argv[1:] lineno = int(lineno) with open(fname) as code_file: -- cgit v1.2.1