diff --git a/tests/__init__.py b/tests/__init__.py
index f817bd0..e69de29 100644
--- a/tests/__init__.py
+++ b/tests/__init__.py
@@ -1,7 +0,0 @@
-# This file is part of stock_lot_csv module for Tryton.
-# The COPYRIGHT file at the top level of this repository contains the full
-# copyright notices and license terms.
-from .test_stock_lot_csv import suite
-
-
-__all__ = ['suite']
diff --git a/tests/test_module.py b/tests/test_module.py
new file mode 100644
index 0000000..194a943
--- /dev/null
+++ b/tests/test_module.py
@@ -0,0 +1,14 @@
+
+# This file is part of Tryton.  The COPYRIGHT file at the top level of
+# this repository contains the full copyright notices and license terms.
+
+from trytond.modules.company.tests import CompanyTestMixin
+from trytond.tests.test_tryton import ModuleTestCase
+
+
+class StockLotCsvTestCase(CompanyTestMixin, ModuleTestCase):
+    'Test StockLotCsv module'
+    module = 'stock_lot_csv'
+
+
+del ModuleTestCase
diff --git a/tests/test_scenario.py b/tests/test_scenario.py
new file mode 100644
index 0000000..b5e24ad
--- /dev/null
+++ b/tests/test_scenario.py
@@ -0,0 +1,23 @@
+
+# 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
+
+
+def load_tests(loader, tests, pattern):
+    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
\ No newline at end of file
diff --git a/tests/test_stock_lot_csv.py b/tests/test_stock_lot_csv.py
deleted file mode 100644
index 8d3262f..0000000
--- a/tests/test_stock_lot_csv.py
+++ /dev/null
@@ -1,18 +0,0 @@
-# This file is part of stock_lot_csv module for Tryton.
-# The COPYRIGHT file at the top level of this repository contains the full
-# copyright notices and license terms.
-from trytond.tests.test_tryton import ModuleTestCase
-from trytond.tests.test_tryton import suite as test_suite
-import unittest
-
-
-class StockLotCsvTestCase(ModuleTestCase):
-    'Test Stock Lot Csv module'
-    module = 'stock_lot_csv'
-
-
-def suite():
-    suite = test_suite()
-    suite.addTests(unittest.TestLoader().loadTestsFromTestCase(
-            StockLotCsvTestCase))
-    return suite