Improve testing helpers.

Signed-off-by: Raphaël Barrois <raphael.barrois@polytechnique.org>

1 files changed, 8 insertions, 1 deletions

diff --git a/setup.py b/setup.py
index 57e701e..ef3da96 100755
--- a/setup.py
+++ b/setup.py
@@ -51,7 +51,14 @@ class test(cmd.Command):
         else:
             verbosity=0
 
-        suite = unittest.TestLoader().loadTestsFromName(self.test_suite)
+        loader = unittest.TestLoader()
+        suite = unittest.TestSuite()
+
+        if self.test_suite == 'tests':
+            for test_module in loader.discover('.'):
+                suite.addTest(test_module)
+        else:
+            suite.addTest(loader.loadTestsFromName(self.test_suite))
 
         result = unittest.TextTestRunner(verbosity=verbosity).run(suite)
         if not result.wasSuccessful():