From 0cabe97af256c5b4ec72fc19a4ded7ac0f399c72 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Rapha=C3=ABl=20Barrois?= <raphael.barrois@polytechnique.org>
Date: Wed, 21 Dec 2011 01:01:59 +0100
Subject: Add a 'test' command to the setup.py script.
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: Raphaël Barrois <raphael.barrois@polytechnique.org>
---
 setup.py | 34 +++++++++++++++++++++++++++++++++-
 1 file changed, 33 insertions(+), 1 deletion(-)

(limited to 'setup.py')

diff --git a/setup.py b/setup.py
index 0933245..6f0dfed 100644
--- a/setup.py
+++ b/setup.py
@@ -2,10 +2,41 @@
 # -*- coding: utf-8 -*-
 
 from distutils.core import setup
+from distutils import cmd
 
 # Remember to change in factory/__init__.py as well!
 VERSION = '1.0.4'
 
+
+class test(cmd.Command):
+    """Run the tests for this package."""
+    command_name = 'test'
+    description = 'run the tests associated with the package'
+
+    user_options = [
+        ('test-suite=', None, "A test suite to run (defaults to 'tests')"),
+    ]
+
+    def initialize_options(self):
+        self.test_runner = None
+        self.test_suite = None
+
+    def finalize_options(self):
+        self.ensure_string('test_suite', 'tests')
+
+    def run(self):
+        """Run the test suite."""
+        import unittest
+        if self.verbose:
+            verbosity=1
+        else:
+            verbosity=0
+
+        suite = unittest.TestLoader().loadTestsFromName(self.test_suite)
+
+        unittest.TextTestRunner(verbosity=verbosity).run(suite)
+
+
 setup(
     name='factory_boy_rbarrois',
     version=VERSION,
@@ -25,5 +56,6 @@ setup(
         'Programming Language :: Python',
         'Topic :: Software Development :: Testing',
         'Topic :: Software Development :: Libraries :: Python Modules'
-    ]
+    ],
+    cmdclass={'test': test},
 )
-- 
cgit v1.2.3