#!/usr/bin/env python # encoding=utf-8 from distutils.core import setup from distutils.cmd import Command class TestClass(Command): # Must be present for "setup.py test -h" to work user_options = [("suite=", "s", "run test suite for a specific module [default: run all tests]")] def __init__(self, dist): Command.__init__(self, dist) setup( # The name string in the dict is used by "setup.py --help-commands", # but *NOT* by "setup.py test -h" cmdclass = { "test" : TestClass }, )