Update travis to use py.test