Add pytest sample