Python Single Line If Statement