move is_categorical_dtype()

Author: kotamatsuoka

pandas/core/internals/blocks.py

@@ -1010,10 +1010,8 @@ def coerce_to_target_dtype(self, other):
         if is_dtype_equal(self.dtype, dtype):
             return self
 
-        if (
-            is_extension_array_dtype(self.dtype)
-            and not is_categorical_dtype(self.dtype)
-            and not self.is_datetime
+        if is_extension_array_dtype(self.dtype) and not is_categorical_dtype(
+            self.dtype
         ):
             dtype = self.dtype
 
@@ -1060,6 +1058,8 @@ def coerce_to_target_dtype(self, other):
             raise AssertionError(
                 f"possible recursion in coerce_to_target_dtype: {self} {other}"
             )
+        if is_categorical_dtype(dtype) or self.is_datetime:
+            return self.astype(object)
 
         try:
             return self.astype(dtype)

pandas/tests/series/methods/test_replace.py

@@ -3,6 +3,7 @@
 
 import pandas as pd
 import pandas._testing as tm
+from pandas.core.arrays import IntervalArray
 
 
 class TestSeriesReplace:
@@ -414,13 +415,13 @@ def test_replace_only_one_dictlike_arg(self):
             (pd.Series([1, 2], dtype="int64"), {1: 10, 2: 20}, "int64"),
             (pd.Series([True, False], dtype="bool"), {True: False}, "bool"),
             (
-                pd.Series(pd.interval_range(0, 5), dtype=pd.IntervalDtype("int64")),
-                {pd.Interval(0, 1): pd.Interval(10, 20)},
+                pd.Series(IntervalArray.from_breaks([1, 2, 3, 4]), dtype=pd.IntervalDtype("int64")),
+                {pd.Interval(1, 2): pd.Interval(10, 20)},
                 "interval[int64]",
             ),
             (
-                pd.Series(pd.interval_range(0, 1), dtype=pd.IntervalDtype("float64")),
-                {pd.Interval(0.0, 1.0): pd.Interval(0.2, 0.3)},
+                pd.Series(IntervalArray.from_breaks([1, 2, 3, 4]), dtype=pd.IntervalDtype("float64")),
+                {pd.Interval(1, 2): pd.Interval(0.2, 0.3)},
                 "interval[float64]",
             ),
             (
@@ -449,5 +450,5 @@ def test_replace_only_one_dictlike_arg(self):
     )
     def test_replace_dtype(self, series, to_replace, expected):
         # GH 33484
-        result = str(series.replace(to_replace).dtype)
+        result = series.replace(to_replace).dtype
         assert expected == result