From 3944f9f952ae594cc3cc6486f467703c1152bfa3 Mon Sep 17 00:00:00 2001 From: Joris Van den Bossche Date: Wed, 4 Aug 2021 14:49:24 +0200 Subject: [PATCH] TST: remove chained assignment outside indexing tests --- pandas/tests/frame/methods/test_dropna.py | 2 +- .../tests/frame/methods/test_interpolate.py | 26 +++++++++---------- pandas/tests/frame/methods/test_isin.py | 6 ++--- 3 files changed, 17 insertions(+), 17 deletions(-) diff --git a/pandas/tests/frame/methods/test_dropna.py b/pandas/tests/frame/methods/test_dropna.py index 76a6f3aa25362..bc2b48d3312d7 100644 --- a/pandas/tests/frame/methods/test_dropna.py +++ b/pandas/tests/frame/methods/test_dropna.py @@ -66,7 +66,7 @@ def test_dropIncompleteRows(self, float_frame): def test_dropna(self): df = DataFrame(np.random.randn(6, 4)) - df[2][:2] = np.nan + df.iloc[:2, 2] = np.nan dropped = df.dropna(axis=1) expected = df.loc[:, [0, 1, 3]] diff --git a/pandas/tests/frame/methods/test_interpolate.py b/pandas/tests/frame/methods/test_interpolate.py index d0551ffd5cffe..7a749f3705e35 100644 --- a/pandas/tests/frame/methods/test_interpolate.py +++ b/pandas/tests/frame/methods/test_interpolate.py @@ -102,34 +102,34 @@ def test_interp_various(self): expected = df.copy() result = df.interpolate(method="polynomial", order=1) - expected.A.loc[3] = 2.66666667 - expected.A.loc[13] = 5.76923076 + expected.loc[3, "A"] = 2.66666667 + expected.loc[13, "A"] = 5.76923076 tm.assert_frame_equal(result, expected) result = df.interpolate(method="cubic") # GH #15662. - expected.A.loc[3] = 2.81547781 - expected.A.loc[13] = 5.52964175 + expected.loc[3, "A"] = 2.81547781 + expected.loc[13, "A"] = 5.52964175 tm.assert_frame_equal(result, expected) result = df.interpolate(method="nearest") - expected.A.loc[3] = 2 - expected.A.loc[13] = 5 + expected.loc[3, "A"] = 2 + expected.loc[13, "A"] = 5 tm.assert_frame_equal(result, expected, check_dtype=False) result = df.interpolate(method="quadratic") - expected.A.loc[3] = 2.82150771 - expected.A.loc[13] = 6.12648668 + expected.loc[3, "A"] = 2.82150771 + expected.loc[13, "A"] = 6.12648668 tm.assert_frame_equal(result, expected) result = df.interpolate(method="slinear") - expected.A.loc[3] = 2.66666667 - expected.A.loc[13] = 5.76923077 + expected.loc[3, "A"] = 2.66666667 + expected.loc[13, "A"] = 5.76923077 tm.assert_frame_equal(result, expected) result = df.interpolate(method="zero") - expected.A.loc[3] = 2.0 - expected.A.loc[13] = 5 + expected.loc[3, "A"] = 2.0 + expected.loc[13, "A"] = 5 tm.assert_frame_equal(result, expected, check_dtype=False) @td.skip_if_no_scipy @@ -218,7 +218,7 @@ def test_interp_leading_nans(self, check_scipy): ) result = df.interpolate() expected = df.copy() - expected["B"].loc[3] = -3.75 + expected.loc[3, "B"] = -3.75 tm.assert_frame_equal(result, expected) if check_scipy: diff --git a/pandas/tests/frame/methods/test_isin.py b/pandas/tests/frame/methods/test_isin.py index d2ebd09c4cc48..e924963f588f3 100644 --- a/pandas/tests/frame/methods/test_isin.py +++ b/pandas/tests/frame/methods/test_isin.py @@ -79,8 +79,8 @@ def test_isin_df(self): df2 = DataFrame({"A": [0, 2, 12, 4], "B": [2, np.nan, 4, 5]}) expected = DataFrame(False, df1.index, df1.columns) result = df1.isin(df2) - expected["A"].loc[[1, 3]] = True - expected["B"].loc[[0, 2]] = True + expected.loc[[1, 3], "A"] = True + expected.loc[[0, 2], "B"] = True tm.assert_frame_equal(result, expected) # partial overlapping columns @@ -133,7 +133,7 @@ def test_isin_against_series(self): ) s = Series([1, 3, 11, 4], index=["a", "b", "c", "d"]) expected = DataFrame(False, index=df.index, columns=df.columns) - expected["A"].loc["a"] = True + expected.loc["a", "A"] = True expected.loc["d"] = True result = df.isin(s) tm.assert_frame_equal(result, expected)