python assignment in if