Add pytest support to setup.py