Added NumpyTestCase.rundocs method to support running doc-tests via unittest machinery - needed for running tests via testoob.

1 files changed, 15 insertions, 0 deletions

diff --git a/numpy/testing/numpytest.py b/numpy/testing/numpytest.py
index 55e4e0fc8..211a1b581 100644
--- a/numpy/testing/numpytest.py
+++ b/numpy/testing/numpytest.py
@@ -168,6 +168,21 @@ class NumpyTestCase (unittest.TestCase):
         print>>sys.stdout, message
         sys.stdout.flush()
 
+    def rundocs(self, filename=None):
+        """ Run doc string tests found in filename.
+        """
+        import doctest
+        if filename is None:
+            f = get_frame(1)
+            filename = f.f_globals['__file__']
+        name = os.path.splitext(os.path.basename(filename))[0]
+        m = imp.load_module(name, open(filename), filename,('.py','U',1))
+        tests = doctest.DocTestFinder().find(m)
+        runner = doctest.DocTestRunner(verbose=False)
+        for test in tests:
+            runner.run(test)
+        return
+
 ScipyTestCase = NumpyTestCase
 
 def _get_all_method_names(cls):