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)
