diff --git a/tests/test_scenario.py b/tests/test_scenario.py index d541941c69..23f5380d62 100644 --- a/tests/test_scenario.py +++ b/tests/test_scenario.py @@ -1,26 +1,13 @@ # This file is part of Tryton. The COPYRIGHT file at the top level of # this repository contains the full copyright notices and license terms. -import doctest -import glob import os -from trytond.tests.test_tryton import doctest_checker, doctest_teardown +from trytond.tests.test_tryton import load_doc_tests - -def load_tests(loader, tests, pattern): - if (os.getenv('UPS_USER_ID') - and os.getenv('UPS_PASSWORD') - and os.getenv('UPS_LICENSE') - and os.getenv('UPS_ACCOUNT_NUMBER')): - cwd = os.getcwd() - try: - os.chdir(os.path.dirname(__file__)) - for scenario in glob.glob('*.rst'): - tests.addTests(doctest.DocFileSuite( - scenario, tearDown=doctest_teardown, encoding='utf-8', - checker=doctest_checker, - optionflags=doctest.REPORT_ONLY_FIRST_FAILURE)) - finally: - os.chdir(cwd) - return tests +if (os.getenv('UPS_USER_ID') + and os.getenv('UPS_PASSWORD') + and os.getenv('UPS_LICENSE') + and os.getenv('UPS_ACCOUNT_NUMBER')): + def load_tests(*args, **kwargs): + return load_doc_tests(__name__, __file__, *args, **kwargs)