Re-add make_suite() that was accidentally removed

wbond · Aug 23, 2023 · 3f8655c · 3f8655c
1 parent d845bb7
commit 3f8655c
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/tests/__init__.py b/tests/__init__.py
@@ -114,6 +114,23 @@ def _import_from(mod, path, mod_dir=None):
  return None
 
 
+def make_suite():
+ """
+ Constructs a unittest.TestSuite() of all tests for the package. For use
+ with setuptools.
+
+ :return:
+ A unittest.TestSuite() object
+ """
+
+ loader = unittest.TestLoader()
+ suite = unittest.TestSuite()
+ for test_class in test_classes():
+ tests = loader.loadTestsFromTestCase(test_class)
+ suite.addTests(tests)
+ return suite
+
+
 def test_classes():
  """
  Returns a list of unittest.TestCase classes for the package