wip

Author: MarcoGorelli

pandas/tests/frame/indexing/test_where.py

@@ -512,28 +512,10 @@ def test_where_axis_with_upcast(self):
         result = df.where(mask, ser, axis="index")
         tm.assert_frame_equal(result, expected)
 
-        result = df.copy()
-        with tm.assert_produces_warning(FutureWarning, match="incompatible dtype"):
-            return_value = result.where(mask, ser, axis="index", inplace=True)
-        assert return_value is None
-        tm.assert_frame_equal(result, expected)
-
         expected = DataFrame([[0, np.nan], [0, np.nan]])
         result = df.where(mask, ser, axis="columns")
         tm.assert_frame_equal(result, expected)
 
-        expected = DataFrame(
-            {
-                0: np.array([0, 0], dtype="int64"),
-                1: np.array([np.nan, np.nan], dtype="float64"),
-            }
-        )
-        result = df.copy()
-        with tm.assert_produces_warning(FutureWarning, match="incompatible dtype"):
-            return_value = result.where(mask, ser, axis="columns", inplace=True)
-        assert return_value is None
-        tm.assert_frame_equal(result, expected)
-
     def test_where_axis_multiple_dtypes(self):
         # Multiple dtypes (=> multiple Blocks)
         df = pd.concat(
@@ -583,16 +565,6 @@ def test_where_axis_multiple_dtypes(self):
         tm.assert_frame_equal(result, expected)
         result = df.where(mask, d1, axis="index")
         tm.assert_frame_equal(result, expected)
-        result = df.copy()
-        with tm.assert_produces_warning(FutureWarning, match="incompatible dtype"):
-            return_value = result.where(mask, d1, inplace=True)
-        assert return_value is None
-        tm.assert_frame_equal(result, expected)
-        result = df.copy()
-        with tm.assert_produces_warning(FutureWarning, match="incompatible dtype"):
-            return_value = result.where(mask, d1, inplace=True, axis="index")
-        assert return_value is None
-        tm.assert_frame_equal(result, expected)
 
         d2 = df.copy().drop(1, axis=1)
         expected = df.copy()
@@ -732,19 +704,6 @@ def test_where_interval_noop(self):
         res = ser.where(ser.notna())
         tm.assert_series_equal(res, ser)
 
-    def test_where_interval_fullop_downcast(self, frame_or_series):
-        # GH#45768
-        obj = frame_or_series([pd.Interval(0, 0)] * 2)
-        other = frame_or_series([1.0, 2.0], dtype=object)
-        res = obj.where(~obj.notna(), other)
-        tm.assert_equal(res, other)
-
-        with tm.assert_produces_warning(
-            FutureWarning, match="Setting an item of incompatible dtype"
-        ):
-            obj.mask(obj.notna(), other, inplace=True)
-        tm.assert_equal(obj, other.astype(object))
-
     @pytest.mark.parametrize(
         "dtype",
         [
@@ -773,14 +732,6 @@ def test_where_datetimelike_noop(self, dtype):
 
         res4 = df.mask(mask2, "foo")
         tm.assert_frame_equal(res4, df)
-        expected = DataFrame(4, index=df.index, columns=df.columns)
-
-        # unlike where, Block.putmask does not downcast
-        with tm.assert_produces_warning(
-            FutureWarning, match="Setting an item of incompatible dtype"
-        ):
-            df.mask(~mask2, 4, inplace=True)
-        tm.assert_frame_equal(df, expected.astype(object))
 
 
 def test_where_int_downcasting_deprecated():
@@ -934,12 +885,6 @@ def test_where_period_invalid_na(frame_or_series, as_cat, request):
         result = obj.mask(mask, tdnat)
         tm.assert_equal(result, expected)
 
-        with tm.assert_produces_warning(
-            FutureWarning, match="Setting an item of incompatible dtype"
-        ):
-            obj.mask(mask, tdnat, inplace=True)
-        tm.assert_equal(obj, expected)
-
 
 def test_where_nullable_invalid_na(frame_or_series, any_numeric_ea_dtype):
     # GH#44697
@@ -981,56 +926,6 @@ def test_where_downcast_to_td64():
     tm.assert_series_equal(res2, expected2)
 
 
-def _check_where_equivalences(df, mask, other, expected):
-    # similar to tests.series.indexing.test_setitem.SetitemCastingEquivalences
-    #  but with DataFrame in mind and less fleshed-out
-    res = df.where(mask, other)
-    tm.assert_frame_equal(res, expected)
-
-    res = df.mask(~mask, other)
-    tm.assert_frame_equal(res, expected)
-
-    # Note: frame.mask(~mask, other, inplace=True) takes some more work bc
-    #  Block.putmask does *not* downcast.  The change to 'expected' here
-    #  is specific to the cases in test_where_dt64_2d.
-    df = df.copy()
-    df.mask(~mask, other, inplace=True)
-    if not mask.all():
-        # with mask.all(), Block.putmask is a no-op, so does not downcast
-        expected = expected.copy()
-        expected["A"] = expected["A"].astype(object)
-    tm.assert_frame_equal(df, expected)
-
-
-def test_where_dt64_2d():
-    dti = date_range("2016-01-01", periods=6)
-    dta = dti._data.reshape(3, 2)
-    other = dta - dta[0, 0]
-
-    df = DataFrame(dta, columns=["A", "B"])
-
-    mask = np.asarray(df.isna()).copy()
-    mask[:, 1] = True
-
-    # setting part of one column, none of the other
-    mask[1, 0] = True
-    expected = DataFrame(
-        {
-            "A": np.array([other[0, 0], dta[1, 0], other[2, 0]], dtype=object),
-            "B": dta[:, 1],
-        }
-    )
-    with tm.assert_produces_warning(
-        FutureWarning, match="Setting an item of incompatible dtype"
-    ):
-        _check_where_equivalences(df, mask, other, expected)
-
-    # setting nothing in either column
-    mask[:] = True
-    expected = df
-    _check_where_equivalences(df, mask, other, expected)
-
-
 def test_where_producing_ea_cond_for_np_dtype():
     # GH#44014
     df = DataFrame({"a": Series([1, pd.NA, 2], dtype="Int64"), "b": [1, 2, 3]})