Add CategoricalDtype to deprecated core.categorical shim (#19373)

jorisvandenbossche · jreback · commit 83e95e09a08b · 2018-01-24T20:07:53.000-05:00
diff --git a/pandas/core/categorical.py b/pandas/core/categorical.py
@@ -5,3 +5,4 @@
               FutureWarning, stacklevel=2)
 
 from pandas.core.arrays import Categorical  # noqa
+from pandas.core.dtypes.dtypes import CategoricalDtype  # noqa
diff --git a/pandas/tests/api/test_api.py b/pandas/tests/api/test_api.py
@@ -259,3 +259,8 @@ def test_categorical_move(self):
 
         with tm.assert_produces_warning(FutureWarning):
             from pandas.core.categorical import Categorical  # noqa
+
+        sys.modules.pop("pandas.core.categorical", None)
+
+        with tm.assert_produces_warning(FutureWarning):
+            from pandas.core.categorical import CategoricalDtype  # noqa

Original file line number	Diff line number	Diff line change
`@@ -5,3 +5,4 @@`
`5`	`5`	`FutureWarning, stacklevel=2)`
`6`	`6`
`7`	`7`	`from pandas.core.arrays import Categorical # noqa`
	`8`	`+from pandas.core.dtypes.dtypes import CategoricalDtype # noqa`