def test_import_all_attrs(self): math = import_module('math') # valid module attrs = import_all_attrs(['math']) self.assertIn('pi', attrs) self.assertEqual(attrs['pi'], math.pi) # two valid modules attrs = import_all_attrs(['math', 'os']) self.assertIn('pi', attrs) self.assertEqual(attrs['pi'], math.pi) self.assertNotIn('environ', attrs) # one invalid module attrs = import_all_attrs(['invalid_one', 'math']) self.assertIn('pi', attrs) self.assertEqual(attrs['pi'], math.pi) # invalid modules self.assertRaises(ImportError, import_all_attrs, ['invalid_one', 'invalid_two']) self.assertIsNone( import_all_attrs(['invalid_one', 'invalid_two'], raise_error=False))
def test_import_all_attrs(self): math = import_module('math') # valid module attrs = import_all_attrs(['math']) self.assertIn('pi', attrs) self.assertEqual(attrs['pi'], math.pi) # two valid modules attrs = import_all_attrs(['math', 'os']) self.assertIn('pi', attrs) self.assertEqual(attrs['pi'], math.pi) self.assertNotIn('environ', attrs) # one invalid module attrs = import_all_attrs(['invalid_one', 'math']) self.assertIn('pi', attrs) self.assertEqual(attrs['pi'], math.pi) # invalid modules self.assertRaises(ImportError, import_all_attrs, ['invalid_one', 'invalid_two']) self.assertIsNone(import_all_attrs(['invalid_one', 'invalid_two'], raise_error=False))
# -*- coding: utf-8 -*- import inspect from baph.utils.importing import import_all_attrs, import_attr globals().update(import_all_attrs(['django.test.client'])) Client = import_attr(['django.test.client'], 'Client') login = import_attr(['baph.auth'], 'login') StringIO = import_attr(['cStringIO', 'StringIO'], 'StringIO') # monkeypatching Client to use baph.auth exec inspect.getsource(Client)