diff --git a/tests/test_account_statement.py b/tests/test_account_statement.py index 674ba1a490..0384bf6fd7 100644 --- a/tests/test_account_statement.py +++ b/tests/test_account_statement.py @@ -3,7 +3,8 @@ import unittest import doctest 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 doctest_setup, doctest_teardown class AccountStatementTestCase(unittest.TestCase): @@ -26,6 +27,6 @@ def suite(): suite.addTests(unittest.TestLoader().loadTestsFromTestCase( AccountStatementTestCase)) suite.addTests(doctest.DocFileSuite('scenario_account_statement.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