Improve testing helpers.

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

2 files changed, 8 insertions, 2 deletions

diff --git a/Makefile b/Makefile
index 144c2e9..f9c27c0 100644
--- a/Makefile
+++ b/Makefile
@@ -17,7 +17,6 @@ coverage:
 	coverage report "--include=factory/*.py,tests/*.py"
 	coverage html "--include=factory/*.py,tests/*.py"
 
-
 doc:
 	$(MAKE) -C docs html
 
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():