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

pandas-dev · Jan 25, 2018 · 83e95e0 · 83e95e0
1 parent b4662cd
commit 83e95e0
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
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
Expand Up		@@ -5,3 +5,4 @@
		FutureWarning, stacklevel=2)

		from pandas.core.arrays import Categorical # noqa
		from pandas.core.dtypes.dtypes import CategoricalDtype # noqa