Added support for setup.py test