Adding the support for 'python setup.py test'