Pandas Categorical Data Test