diff --git a/tests/test_stock_lot.py b/tests/test_stock_lot.py index a380170d8e..23895166d4 100644 --- a/tests/test_stock_lot.py +++ b/tests/test_stock_lot.py @@ -13,10 +13,11 @@ from trytond.tests.test_tryton import doctest_checker from trytond.transaction import Transaction from trytond.pool import Pool -from trytond.modules.company.tests import create_company, set_company +from trytond.modules.company.tests import ( + create_company, set_company, CompanyTestMixin) -class StockLotTestCase(ModuleTestCase): +class StockLotTestCase(CompanyTestMixin, ModuleTestCase): 'Test Stock Lot module' module = 'stock_lot'