Better setup.py test integration

python setup.py test will now work, instead of the
user manually needing to execute py.test