update yaml to use py.test