diff -r 36b9e48666ce -r 9352acb9366b src/pyams_zmq/tests/test_utilsdocs.py --- a/src/pyams_zmq/tests/test_utilsdocs.py Fri Jan 11 13:52:29 2019 +0100 +++ b/src/pyams_zmq/tests/test_utilsdocs.py Thu Nov 28 19:19:25 2019 +0100 @@ -1,5 +1,7 @@ +# -*- coding: utf-8 -*- ###################################################### +############################################################################## # -# Copyright (c) 2008-2015 Thierry Florac +# Copyright (c) 2008-2010 Thierry Florac # All Rights Reserved. # # This software is subject to the provisions of the Zope Public License, @@ -9,21 +11,25 @@ # WARRANTIES OF TITLE, MERCHANTABILITY, AGAINST INFRINGEMENT, AND FITNESS # FOR A PARTICULAR PURPOSE. # +############################################################################## """ -Generic Test case for pyams_zmq doctest +Generic test case for pyams_zmq doctests """ + __docformat__ = 'restructuredtext' +import doctest +import os import unittest -import doctest -import sys -import os + +from pyams_utils.tests import get_package_dir -current_dir = os.path.dirname(__file__) +CURRENT_DIR = os.path.abspath(os.path.dirname(__file__)) -def doc_suite(test_dir, setUp=None, tearDown=None, globs=None): + +def doc_suite(test_dir, setUp=None, tearDown=None, globs=None): # pylint: disable=invalid-name """Returns a test suite, based on doctests found in /doctest.""" suite = [] if globs is None: @@ -32,15 +38,12 @@ flags = (doctest.ELLIPSIS | doctest.NORMALIZE_WHITESPACE | doctest.REPORT_ONLY_FIRST_FAILURE) - package_dir = os.path.split(test_dir)[0] - if package_dir not in sys.path: - sys.path.append(package_dir) - + package_dir = get_package_dir(test_dir) doctest_dir = os.path.join(package_dir, 'doctests') # filtering files on extension docs = [os.path.join(doctest_dir, doc) for doc in - os.listdir(doctest_dir) if doc.endswith('.txt')] + os.listdir(doctest_dir) if doc.endswith('.txt') or doc.endswith('.rst')] for test in docs: suite.append(doctest.DocFileSuite(test, optionflags=flags, @@ -50,10 +53,11 @@ return unittest.TestSuite(suite) + def test_suite(): """returns the test suite""" - return doc_suite(current_dir) + return doc_suite(CURRENT_DIR) + if __name__ == '__main__': unittest.main(defaultTest='test_suite') -