update indexing tests

jorisvandenbossche · Dec 22, 2024 · 65f8b54 · 65f8b54
1 parent cc7ebe4
commit 65f8b54
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/pandas/tests/indexing/test_coercion.py b/pandas/tests/indexing/test_coercion.py
@@ -856,7 +856,7 @@ def test_replace_series(self, how, to_key, from_key, replacer, using_infer_strin
         else:
             exp = pd.Series(self.rep[to_key], index=index, name="yyy")
 
-        if using_infer_string and exp.dtype == "string" and obj.dtype == object:
+        if using_infer_string and exp.dtype == "string":
             # with infer_string, we disable the deprecated downcasting behavior
             exp = exp.astype(object)
 
@@ -889,8 +889,9 @@ def test_replace_series_datetime_tz(
         assert obj.dtype == from_key
 
         exp = pd.Series(self.rep[to_key], index=index, name="yyy")
-        if using_infer_string and to_key == "object":
-            assert exp.dtype == "string"
+        if using_infer_string and exp.dtype == "string":
+            # with infer_string, we disable the deprecated downcasting behavior
+            exp = exp.astype(object)
         else:
             assert exp.dtype == to_key