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