Switch setup.py test to pytest