Just adding a small line in setup.py so that you can just run tests by running 'python setup.py test'