Switch to pytest

For better cross-python testing