Move to py.test