testmodule.py 文件源码

def get_submodules(self,module=None,nr=0):
    print('get_submodules(%s)'%module)
    try:
      module = module or self.module
      if fn.isString(module):
        m = self.load_module(module)
      else:
        m,module = module,module.__name__
      result = set()
      l = getattr(m,'__test__',dir(m))
      print m,l
      l = list(l)
      for o in l:
        o = o.split('.')[-1]
        n = getattr(m,o)
        if self.is_module(n) and module == n.__package__:
          o = module+'.'+o
          result.add(o)
          if nr<10:
            result = result.union(self.get_submodules(n,nr=nr+1))
    except Exception,e:
      print('get_submodules(%s)'%module)
      traceback.print_exc()
      raise e
    return result