Commit bfb7e3b9 authored by Denis Bilenko's avatar Denis Bilenko

test__doctests.py: make sure setup.py can be imported regardless of PYTHONPATH

parent 2a723bf6
import sys
import os
import re
import doctest
......@@ -6,21 +7,22 @@ from os.path import join, dirname
import gevent
from gevent import socket
modules = set()
modules.add(('setup', 'setup.py'))
def myfunction(*args, **kwargs):
pass
if __name__ == '__main__':
cwd = os.getcwd()
saved_sys_path = sys.path[:]
try:
sys.path.append('.')
base = dirname(gevent.__file__)
print base
os.chdir(join(base, '..'))
os.chdir('..')
globs = {'myfunction': myfunction, 'gevent': gevent, 'socket': socket}
modules = set()
for path, dirs, files in os.walk(base):
package = 'gevent' + path.replace(base, '').replace('/', '.')
modules.add((package, join(path, '__init__.py')))
......@@ -31,6 +33,8 @@ if __name__ == '__main__':
if module:
modules.add((package + '.' + module, join(path, f)))
modules.add(('setup', 'setup.py'))
suite = unittest.TestSuite()
tests_count = 0
modules_count = 0
......@@ -46,5 +50,4 @@ if __name__ == '__main__':
runner.run(suite)
finally:
os.chdir(cwd)
sys.path = saved_sys_path
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment