unittest discovery

Quasars · Oct 18, 2024 · aca79bb · aca79bb
1 parent b3394fe
commit aca79bb
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 0 deletions.
diff --git a/orangecontrib/snom/tests/__init__.py b/orangecontrib/snom/tests/__init__.py
@@ -0,0 +1,17 @@
+import os
+import unittest
+
+
+def load_tests(loader, tests, pattern):
+    tests_dir = os.path.dirname(__file__)
+
+    if loader is None:
+        loader = unittest.TestLoader()
+    if pattern is None:
+        pattern = 'test*.py'
+
+    all_tests = [
+        loader.discover(tests_dir, pattern, tests_dir),
+    ]
+
+    return unittest.TestSuite(all_tests)
diff --git a/orangecontrib/snom/widgets/tests/__init__.py b/orangecontrib/snom/widgets/tests/__init__.py
@@ -0,0 +1,17 @@
+import os
+import unittest
+
+
+def load_tests(loader, tests, pattern):
+    tests_dir = os.path.dirname(__file__)
+
+    if loader is None:
+        loader = unittest.TestLoader()
+    if pattern is None:
+        pattern = 'test*.py'
+
+    all_tests = [
+        loader.discover(tests_dir, pattern, tests_dir),
+    ]
+
+    return unittest.TestSuite(all_tests)