Update travis config to use py.test