Switched test runner to py.test