Using py.test instead of setup.py test