test/test_moduleimport.py

   1 #!/usr/bin/env python
   2 import os
   3 import imp
   4 import unittest
   5
   6 import testlib
   7
   8 class ImportTest(testlib.TestCase):
   9     pass
  10
  11 def gen_class(module):
  12     def import_test(self):
  13         modinfo = None
  14         for idx, path in enumerate(module.split('.')):
  15             if idx == 0:
  16                 modinfo = imp.find_module(path)
  17                 mod = imp.load_module(module, *modinfo)
  18             else:
  19                 modinfo = imp.find_module('cola/'+path, modinfo[0])
  20                 mod = imp.load_module(path, *modinfo)
  21             self.failUnless(mod)
  22     return import_test
  23
  24 def __create_tests():
  25     for module in """
  26         cola.git
  27         cola.model
  28         cola.observer
  29         cola.exception
  30         cola.defaults
  31         cola.settings
  32     """.split():
  33         setattr(ImportTest,
  34                 "test" + module.title().replace('.', ''),
  35                 gen_class(module))
  36 __create_tests()
  37
  38 if __name__ == '__main__':
  39     unittest.main()