Update setup.py

Add py.test runner