A minor change to setup.py to enable running test using python setup.py test

This little change will enable tests to be run using `python setup.py test`