Add pytest requirement