Use unittest2 loader.discover for python2.6

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

1 files changed, 5 insertions, 1 deletions

diff --git a/setup.py b/setup.py
index b2fc16d..002cd44 100755
--- a/setup.py
+++ b/setup.py
@@ -41,7 +41,11 @@ class test(cmd.Command):
 
     def run(self):
         """Run the test suite."""
-        import unittest
+        try:
+            import unittest2 as unittest
+        except ImportError:
+            import unittest
+
         if self.verbose:
             verbosity=1
         else: