Check Normality — Interview Query python problem