20 import sys

    20 import sys

    21 import os

    21 import os

    22 

    22 

    23 

    23 

    24 current_dir = os.path.dirname(__file__)

    24 current_dir = os.path.dirname(__file__)

    25 

    26 

    26 def doc_suite(test_dir, setUp=None, tearDown=None, globs=None):

    27 def doc_suite(test_dir, setUp=None, tearDown=None, globs=None):

    27     """Returns a test suite, based on doctests found in /doctest."""

    28     """Returns a test suite, based on doctests found in /doctest."""

    28     suite = []

    29     suite = []

    29     if globs is None:

    30     if globs is None:

    48                                           tearDown=tearDown,

    49                                           tearDown=tearDown,

    49                                           module_relative=False))

    50                                           module_relative=False))

    50 

    51 

    51     return unittest.TestSuite(suite)

    52     return unittest.TestSuite(suite)

    52 

    53 

    53 def test_suite():

    55 def test_suite():

    54     """returns the test suite"""

    56     """returns the test suite"""

    55     return doc_suite(current_dir)

    57     return doc_suite(current_dir)

    56 

    58 

    57 if __name__ == '__main__':

    60 if __name__ == '__main__':

    58     unittest.main(defaultTest='test_suite')

    61     unittest.main(defaultTest='test_suite')

    59 

    62