python assignment in while