Added support for "python setup.py test"