Generator as function argument Generator as function argument python python