def test_decorator_attrs(self): def fxn(self, module): pass wrapped = util.module_for_loader(fxn) self.assertEqual(wrapped.__name__, fxn.__name__) self.assertEqual(wrapped.__qualname__, fxn.__qualname__)
def raise_exception(self, name): def to_wrap(self, module): raise ImportError fxn = util.module_for_loader(to_wrap) try: fxn(self, name) except ImportError: pass
def return_module(self, name): fxn = util.module_for_loader(lambda self, module: module) return fxn(self, name)
def module_for_loader(func): with warnings.catch_warnings(): warnings.simplefilter('ignore', PendingDeprecationWarning) return util.module_for_loader(func)
def test_warning(self): # Should raise a PendingDeprecationWarning when used. with warnings.catch_warnings(): warnings.simplefilter('error', PendingDeprecationWarning) with self.assertRaises(PendingDeprecationWarning): func = util.module_for_loader(lambda x: x)