diff options
author | Ezio Melotti <ezio.melotti@gmail.com> | 2013-03-30 05:17:24 +0200 |
---|---|---|
committer | Ezio Melotti <ezio.melotti@gmail.com> | 2013-03-30 05:17:24 +0200 |
commit | 1b145927d725b136b987df220dff0ec7529b6f29 (patch) | |
tree | a597728e07ca0af44f923c7f056aec396f61e784 /Lib | |
parent | 9098472299a18471f29b3ed471376be2d29178be (diff) | |
download | cpython-git-1b145927d725b136b987df220dff0ec7529b6f29.tar.gz |
#17526: fix an IndexError raised while passing code without filename to inspect.findsource(). Initial patch by Tyler Doyle.
Diffstat (limited to 'Lib')
-rw-r--r-- | Lib/inspect.py | 2 | ||||
-rw-r--r-- | Lib/test/test_inspect.py | 6 |
2 files changed, 7 insertions, 1 deletions
diff --git a/Lib/inspect.py b/Lib/inspect.py index 7a7bb91b1b..7834d12ea7 100644 --- a/Lib/inspect.py +++ b/Lib/inspect.py @@ -550,7 +550,7 @@ def findsource(object): file = getfile(object) sourcefile = getsourcefile(object) - if not sourcefile and file[0] + file[-1] != '<>': + if not sourcefile and file[:1] + file[-1:] != '<>': raise IOError('source code not available') file = sourcefile if sourcefile else file diff --git a/Lib/test/test_inspect.py b/Lib/test/test_inspect.py index 6e3f04e68a..9f5e93b0c7 100644 --- a/Lib/test/test_inspect.py +++ b/Lib/test/test_inspect.py @@ -416,6 +416,12 @@ class TestBuggyCases(GetSourceBase): finally: del linecache.cache[co.co_filename] + def test_findsource_without_filename(self): + for fname in ['', '<string>']: + co = compile('x=1', fname, "exec") + self.assertRaises(IOError, inspect.findsource, co) + self.assertRaises(IOError, inspect.getsource, co) + class TestNoEOL(GetSourceBase): def __init__(self, *args, **kwargs): self.tempdir = TESTFN + '_dir' |