Add py.test command