Switch to pytest