Set travis to use py.test