Using py.test instead of nosetest