Change test runner to pytest