1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
#!/usr/bin/env python3
"""Run doctests in extension modules of <pkg_name>
Collect extension modules in <pkg_name>
Run doctests in each extension module
Example:
%prog nipy
"""
import doctest
import os
import sys
from distutils.sysconfig import get_config_vars
from optparse import OptionParser
from os.path import abspath, dirname, relpath, sep, splitext
from os.path import join as pjoin
EXT_EXT = get_config_vars('SO')[0]
def get_ext_modules(pkg_name):
pkg = __import__(pkg_name, fromlist=[''])
pkg_dir = abspath(dirname(pkg.__file__))
# pkg_root = __import__(pkg_name)
ext_modules = []
for dirpath, dirnames, filenames in os.walk(pkg_dir):
reldir = relpath(dirpath, pkg_dir)
if reldir == '.':
reldir = ''
for filename in filenames:
froot, ext = splitext(filename)
if ext == EXT_EXT:
mod_path = pjoin(reldir, froot)
mod_uri = pkg_name + '.' + mod_path.replace(sep, '.')
# fromlist=[''] results in submodule being returned, rather than the
# top level module. See help(__import__)
mod = __import__(mod_uri, fromlist=[''])
ext_modules.append(mod)
return ext_modules
def main():
usage = "usage: %prog [options] <pkg_name>\n\n" + __doc__
parser = OptionParser(usage=usage)
opts, args = parser.parse_args()
if len(args) == 0:
parser.print_help()
sys.exit(1)
mod_name = args[0]
mods = get_ext_modules(mod_name)
for mod in mods:
print("Testing module: " + mod.__name__)
doctest.testmod(mod)
if __name__ == '__main__':
main()
|
