diff --git a/tests/test_purchase.py b/tests/test_purchase.py index f466068bfc..01f2b25f09 100644 --- a/tests/test_purchase.py +++ b/tests/test_purchase.py @@ -5,8 +5,9 @@ import doctest from decimal import Decimal import trytond.tests.test_tryton -from trytond.tests.test_tryton import test_view, test_depends, doctest_dropdb +from trytond.tests.test_tryton import test_view, test_depends from trytond.tests.test_tryton import POOL, DB_NAME, USER, CONTEXT +from trytond.tests.test_tryton import doctest_setup, doctest_teardown from trytond.transaction import Transaction @@ -124,6 +125,6 @@ def suite(): suite.addTests(unittest.TestLoader().loadTestsFromTestCase( PurchaseTestCase)) suite.addTests(doctest.DocFileSuite('scenario_purchase.rst', - setUp=doctest_dropdb, tearDown=doctest_dropdb, encoding='UTF-8', + setUp=doctest_setup, tearDown=doctest_teardown, encoding='UTF-8', optionflags=doctest.REPORT_ONLY_FIRST_FAILURE)) return suite