TST: Test CategoricalIndex in test_is_categorical (pandas-dev#16243)

Follow-up to pandas-devgh-16237.
stangirala · Jun 11, 2017 · b6c0b91 · b6c0b91
1 parent 3bcbc90
commit b6c0b91
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/pandas/core/dtypes/common.py b/pandas/core/dtypes/common.py
@@ -205,13 +205,15 @@ def is_categorical(arr):
     >>> is_categorical([1, 2, 3])
     False
 
-    Categoricals and Series Categoricals will return True.
+    Categoricals, Series Categoricals, and CategoricalIndex will return True.
 
     >>> cat = pd.Categorical([1, 2, 3])
     >>> is_categorical(cat)
     True
     >>> is_categorical(pd.Series(cat))
     True
+    >>> is_categorical(pd.CategoricalIndex([1, 2, 3]))
+    True
     """
 
     return isinstance(arr, ABCCategorical) or is_categorical_dtype(arr)

diff --git a/pandas/tests/dtypes/test_common.py b/pandas/tests/dtypes/test_common.py
@@ -158,6 +158,7 @@ def test_is_categorical():
     cat = pd.Categorical([1, 2, 3])
     assert com.is_categorical(cat)
     assert com.is_categorical(pd.Series(cat))
+    assert com.is_categorical(pd.CategoricalIndex([1, 2, 3]))
 
     assert not com.is_categorical([1, 2, 3])