Switch tests to pytest
2 files changed