From ef3ff02eaa80b3b6513a4f323fc2574906b43e28 Mon Sep 17 00:00:00 2001
From: Denis Bilenko <denis.bilenko@gmail.com>
Date: Wed, 15 Sep 2010 18:44:18 +0700
Subject: [PATCH] test__doctests.py: use walk_modules() function
---
greentest/test__doctests.py | 13 +++----------
1 file changed, 3 insertions(+), 10 deletions(-)
diff --git a/greentest/test__doctests.py b/greentest/test__doctests.py
index 57394dd1..5e049e44 100644
--- a/greentest/test__doctests.py
+++ b/greentest/test__doctests.py
@@ -6,6 +6,7 @@ import unittest
from os.path import join, dirname
import gevent
from gevent import socket
+from greentest import walk_modules
# Ignore tracebacks: ZeroDivisionError
@@ -30,16 +31,8 @@ if __name__ == '__main__':
return
modules.add((name, path))
- for path, dirs, files in os.walk(base):
- package = 'gevent' + path.replace(base, '').replace('/', '.')
- add_module(package, join(path, '__init__.py'))
- for f in files:
- module = None
- if f.endswith('.py'):
- module = f[:-3]
- if module:
- add_module(package + '.' + module, join(path, f))
-
+ for path, module in walk_modules():
+ add_module(module, path)
add_module('setup', 'setup.py')
if not modules:
--
2.30.9