Use pytest to run tests