diff options
author | Philipp A <flying-sheep@web.de> | 2019-06-08 12:05:46 (GMT) |
---|---|---|
committer | Miss Islington (bot) <31488909+miss-islington@users.noreply.github.com> | 2019-06-08 12:05:46 (GMT) |
commit | d407d2a7265f6102e51a1d62b3fd28b4f7a78d16 (patch) | |
tree | 6e52402cf0cdff805922a75e13822d54a0acb3f0 /Lib | |
parent | 65e5860fcc8ffe66f3c325d5484112f3b6540e8c (diff) | |
download | cpython-d407d2a7265f6102e51a1d62b3fd28b4f7a78d16.zip cpython-d407d2a7265f6102e51a1d62b3fd28b4f7a78d16.tar.gz cpython-d407d2a7265f6102e51a1d62b3fd28b4f7a78d16.tar.bz2 |
bpo-37173: Show passed class in inspect.getfile error (GH-13861)
Currently, inspect.getfile(str) will report nonsense:
```pytb
>>> inspect.getfile(str)
TypeError: <module 'builtins' (built-in)> is a built-in class
```
This fixes that
https://bugs.python.org/issue37173
Diffstat (limited to 'Lib')
-rw-r--r-- | Lib/inspect.py | 6 | ||||
-rw-r--r-- | Lib/test/test_inspect.py | 18 |
2 files changed, 21 insertions, 3 deletions
diff --git a/Lib/inspect.py b/Lib/inspect.py index 91d209d..99a580b 100644 --- a/Lib/inspect.py +++ b/Lib/inspect.py @@ -659,9 +659,9 @@ def getfile(object): raise TypeError('{!r} is a built-in module'.format(object)) if isclass(object): if hasattr(object, '__module__'): - object = sys.modules.get(object.__module__) - if getattr(object, '__file__', None): - return object.__file__ + module = sys.modules.get(object.__module__) + if getattr(module, '__file__', None): + return module.__file__ raise TypeError('{!r} is a built-in class'.format(object)) if ismethod(object): object = object.__func__ diff --git a/Lib/test/test_inspect.py b/Lib/test/test_inspect.py index 83a5f7e..1cd4ea2 100644 --- a/Lib/test/test_inspect.py +++ b/Lib/test/test_inspect.py @@ -513,6 +513,24 @@ class TestRetrievingSourceCode(GetSourceBase): def test_getfile(self): self.assertEqual(inspect.getfile(mod.StupidGit), mod.__file__) + def test_getfile_builtin_module(self): + with self.assertRaises(TypeError) as e: + inspect.getfile(sys) + self.assertTrue(str(e.exception).startswith('<module')) + + def test_getfile_builtin_class(self): + with self.assertRaises(TypeError) as e: + inspect.getfile(int) + self.assertTrue(str(e.exception).startswith('<class')) + + def test_getfile_builtin_function_or_method(self): + with self.assertRaises(TypeError) as e_abs: + inspect.getfile(abs) + self.assertIn('expected, got', str(e_abs.exception)) + with self.assertRaises(TypeError) as e_append: + inspect.getfile(list.append) + self.assertIn('expected, got', str(e_append.exception)) + def test_getfile_class_without_module(self): class CM(type): @property |