inspect.getfile: Don't crash on classes without '__module__' attribute #20372

Some classes defined in C may not have the '__module__' attribute, so
we now handle this case to avoid having unexepected AttributeError.
This commit is contained in:
Yury Selivanov 2014-01-27 13:24:56 -05:00
parent 32970b8dec
commit 2eed8b7da0
2 changed files with 14 additions and 3 deletions

View file

@ -319,6 +319,16 @@ class TestRetrievingSourceCode(GetSourceBase):
def test_getfile(self):
self.assertEqual(inspect.getfile(mod.StupidGit), mod.__file__)
def test_getfile_class_without_module(self):
class CM(type):
@property
def __module__(cls):
raise AttributeError
class C(metaclass=CM):
pass
with self.assertRaises(TypeError):
inspect.getfile(C)
def test_getmodule_recursion(self):
from types import ModuleType
name = '__inspect_dummy'