Add py.test requirement