Pytest Function To Count Set