diff --git a/tests/__init__.py b/tests/__init__.py
index 9b5b97d..e69de29 100644
--- a/tests/__init__.py
+++ b/tests/__init__.py
@@ -1,7 +0,0 @@
-# This file is part stock_move_warehouse module for Tryton.
-# The COPYRIGHT file at the top level of this repository contains
-# the full copyright notices and license terms.
-
-from .test_stock_move_warehouse import suite
-
-__all__ = ['suite']
diff --git a/tests/test_module.py b/tests/test_module.py
new file mode 100644
index 0000000..d3c1fbd
--- /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 StockMoveWarehouseTestCase(CompanyTestMixin, ModuleTestCase):
+    'Test StockMoveWarehouse module'
+    module = 'stock_move_warehouse'
+
+
+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_move_warehouse.py b/tests/test_stock_move_warehouse.py
deleted file mode 100644
index b38f49e..0000000
--- a/tests/test_stock_move_warehouse.py
+++ /dev/null
@@ -1,18 +0,0 @@
-# This file is part of the stock_move_warehouse module for Tryton.
-# The COPYRIGHT file at the top level of this repository contains the full
-# copyright notices and license terms.
-import unittest
-import trytond.tests.test_tryton
-from trytond.tests.test_tryton import ModuleTestCase
-
-
-class StockMoveWarehouseTestCase(ModuleTestCase):
-    'Test Stock Move Warehouse module'
-    module = 'stock_move_warehouse'
-
-
-def suite():
-    suite = trytond.tests.test_tryton.suite()
-    suite.addTests(unittest.TestLoader().loadTestsFromTestCase(
-        StockMoveWarehouseTestCase))
-    return suite