Skip to content

Commit 74e50ec

Browse files
TST: remove chained assignment outside indexing tests (pandas-dev#42882)
1 parent 074b8ba commit 74e50ec

File tree

3 files changed

+17
-17
lines changed

3 files changed

+17
-17
lines changed

pandas/tests/frame/methods/test_dropna.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,7 @@ def test_dropIncompleteRows(self, float_frame):
6666

6767
def test_dropna(self):
6868
df = DataFrame(np.random.randn(6, 4))
69-
df[2][:2] = np.nan
69+
df.iloc[:2, 2] = np.nan
7070

7171
dropped = df.dropna(axis=1)
7272
expected = df.loc[:, [0, 1, 3]]

pandas/tests/frame/methods/test_interpolate.py

+13-13
Original file line numberDiff line numberDiff line change
@@ -102,34 +102,34 @@ def test_interp_various(self):
102102
expected = df.copy()
103103
result = df.interpolate(method="polynomial", order=1)
104104

105-
expected.A.loc[3] = 2.66666667
106-
expected.A.loc[13] = 5.76923076
105+
expected.loc[3, "A"] = 2.66666667
106+
expected.loc[13, "A"] = 5.76923076
107107
tm.assert_frame_equal(result, expected)
108108

109109
result = df.interpolate(method="cubic")
110110
# GH #15662.
111-
expected.A.loc[3] = 2.81547781
112-
expected.A.loc[13] = 5.52964175
111+
expected.loc[3, "A"] = 2.81547781
112+
expected.loc[13, "A"] = 5.52964175
113113
tm.assert_frame_equal(result, expected)
114114

115115
result = df.interpolate(method="nearest")
116-
expected.A.loc[3] = 2
117-
expected.A.loc[13] = 5
116+
expected.loc[3, "A"] = 2
117+
expected.loc[13, "A"] = 5
118118
tm.assert_frame_equal(result, expected, check_dtype=False)
119119

120120
result = df.interpolate(method="quadratic")
121-
expected.A.loc[3] = 2.82150771
122-
expected.A.loc[13] = 6.12648668
121+
expected.loc[3, "A"] = 2.82150771
122+
expected.loc[13, "A"] = 6.12648668
123123
tm.assert_frame_equal(result, expected)
124124

125125
result = df.interpolate(method="slinear")
126-
expected.A.loc[3] = 2.66666667
127-
expected.A.loc[13] = 5.76923077
126+
expected.loc[3, "A"] = 2.66666667
127+
expected.loc[13, "A"] = 5.76923077
128128
tm.assert_frame_equal(result, expected)
129129

130130
result = df.interpolate(method="zero")
131-
expected.A.loc[3] = 2.0
132-
expected.A.loc[13] = 5
131+
expected.loc[3, "A"] = 2.0
132+
expected.loc[13, "A"] = 5
133133
tm.assert_frame_equal(result, expected, check_dtype=False)
134134

135135
@td.skip_if_no_scipy
@@ -218,7 +218,7 @@ def test_interp_leading_nans(self, check_scipy):
218218
)
219219
result = df.interpolate()
220220
expected = df.copy()
221-
expected["B"].loc[3] = -3.75
221+
expected.loc[3, "B"] = -3.75
222222
tm.assert_frame_equal(result, expected)
223223

224224
if check_scipy:

pandas/tests/frame/methods/test_isin.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -79,8 +79,8 @@ def test_isin_df(self):
7979
df2 = DataFrame({"A": [0, 2, 12, 4], "B": [2, np.nan, 4, 5]})
8080
expected = DataFrame(False, df1.index, df1.columns)
8181
result = df1.isin(df2)
82-
expected["A"].loc[[1, 3]] = True
83-
expected["B"].loc[[0, 2]] = True
82+
expected.loc[[1, 3], "A"] = True
83+
expected.loc[[0, 2], "B"] = True
8484
tm.assert_frame_equal(result, expected)
8585

8686
# partial overlapping columns
@@ -133,7 +133,7 @@ def test_isin_against_series(self):
133133
)
134134
s = Series([1, 3, 11, 4], index=["a", "b", "c", "d"])
135135
expected = DataFrame(False, index=df.index, columns=df.columns)
136-
expected["A"].loc["a"] = True
136+
expected.loc["a", "A"] = True
137137
expected.loc["d"] = True
138138
result = df.isin(s)
139139
tm.assert_frame_equal(result, expected)

0 commit comments

Comments
 (0)