diff options
| author | Ned Batchelder <ned@nedbatchelder.com> | 2022-02-04 18:50:30 -0500 |
|---|---|---|
| committer | Ned Batchelder <ned@nedbatchelder.com> | 2022-02-05 12:49:11 -0500 |
| commit | bf99bb663940f203bee6d373f6b6f30840831fab (patch) | |
| tree | 6bc321937dacc79fb1490e20add177543486a601 /lab | |
| parent | c9c1efcf752f0deee4b06c0b5fdf4dfc41c0bb98 (diff) | |
| download | python-coveragepy-git-bf99bb663940f203bee6d373f6b6f30840831fab.tar.gz | |
debug: a little easier to get code samples from tests
Diffstat (limited to 'lab')
| -rw-r--r-- | lab/extract_code.py | 5 |
1 files changed, 4 insertions, 1 deletions
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: |