Switch to using py.test

Can you bring back `flake8` please?