0
|
1 from distutils.core import setup, Command
|
|
2 from unittest import TextTestRunner
|
|
3
|
|
4
|
|
5 class test(Command):
|
|
6 description = "Runs the unit tests"
|
|
7 user_options = [('test-suite=', 's', "Name of the unittest suite to run")]
|
|
8
|
|
9 def initialize_options(self):
|
|
10 self.test_suite = None
|
|
11
|
|
12 def finalize_options(self):
|
|
13 pass
|
|
14
|
|
15 def run(self):
|
|
16 print 'Hey yo'
|
|
17 print self.test_suite
|
|
18 suite = __import__(self.test_suite, locals(), globals())
|
|
19 runner = unittest.TextTestRunner()
|
|
20 TextTestRunner.run(suite)
|
|
21
|
|
22
|
|
23 setup(name='bitten', version='1.0',
|
|
24 packages=['bitten', 'bitten.general', 'bitten.python'],
|
|
25 cmdclass={'test': test})
|