Switch test runner to py.test