Skip to content

Commit 8295dc4

Browse files
authored
TST Verifiy that dropna returns none when called inplace (#35179) (#35181)
1 parent 59128e3 commit 8295dc4

File tree

1 file changed

+24
-12
lines changed

1 file changed

+24
-12
lines changed

pandas/tests/frame/test_missing.py

+24-12
Original file line numberDiff line numberDiff line change
@@ -24,14 +24,16 @@ def test_dropEmptyRows(self, float_frame):
2424
smaller_frame = frame.dropna(how="all")
2525
# check that original was preserved
2626
tm.assert_series_equal(frame["foo"], original)
27-
inplace_frame1.dropna(how="all", inplace=True)
27+
return_value = inplace_frame1.dropna(how="all", inplace=True)
2828
tm.assert_series_equal(smaller_frame["foo"], expected)
2929
tm.assert_series_equal(inplace_frame1["foo"], expected)
30+
assert return_value is None
3031

3132
smaller_frame = frame.dropna(how="all", subset=["foo"])
32-
inplace_frame2.dropna(how="all", subset=["foo"], inplace=True)
33+
return_value = inplace_frame2.dropna(how="all", subset=["foo"], inplace=True)
3334
tm.assert_series_equal(smaller_frame["foo"], expected)
3435
tm.assert_series_equal(inplace_frame2["foo"], expected)
36+
assert return_value is None
3537

3638
def test_dropIncompleteRows(self, float_frame):
3739
N = len(float_frame.index)
@@ -45,18 +47,20 @@ def test_dropIncompleteRows(self, float_frame):
4547

4648
smaller_frame = frame.dropna()
4749
tm.assert_series_equal(frame["foo"], original)
48-
inp_frame1.dropna(inplace=True)
50+
return_value = inp_frame1.dropna(inplace=True)
4951

5052
exp = Series(mat[5:], index=float_frame.index[5:], name="foo")
5153
tm.assert_series_equal(smaller_frame["foo"], exp)
5254
tm.assert_series_equal(inp_frame1["foo"], exp)
55+
assert return_value is None
5356

5457
samesize_frame = frame.dropna(subset=["bar"])
5558
tm.assert_series_equal(frame["foo"], original)
5659
assert (frame["bar"] == 5).all()
57-
inp_frame2.dropna(subset=["bar"], inplace=True)
60+
return_value = inp_frame2.dropna(subset=["bar"], inplace=True)
5861
tm.assert_index_equal(samesize_frame.index, float_frame.index)
5962
tm.assert_index_equal(inp_frame2.index, float_frame.index)
63+
assert return_value is None
6064

6165
def test_dropna(self):
6266
df = DataFrame(np.random.randn(6, 4))
@@ -65,31 +69,35 @@ def test_dropna(self):
6569
dropped = df.dropna(axis=1)
6670
expected = df.loc[:, [0, 1, 3]]
6771
inp = df.copy()
68-
inp.dropna(axis=1, inplace=True)
72+
return_value = inp.dropna(axis=1, inplace=True)
6973
tm.assert_frame_equal(dropped, expected)
7074
tm.assert_frame_equal(inp, expected)
75+
assert return_value is None
7176

7277
dropped = df.dropna(axis=0)
7378
expected = df.loc[list(range(2, 6))]
7479
inp = df.copy()
75-
inp.dropna(axis=0, inplace=True)
80+
return_value = inp.dropna(axis=0, inplace=True)
7681
tm.assert_frame_equal(dropped, expected)
7782
tm.assert_frame_equal(inp, expected)
83+
assert return_value is None
7884

7985
# threshold
8086
dropped = df.dropna(axis=1, thresh=5)
8187
expected = df.loc[:, [0, 1, 3]]
8288
inp = df.copy()
83-
inp.dropna(axis=1, thresh=5, inplace=True)
89+
return_value = inp.dropna(axis=1, thresh=5, inplace=True)
8490
tm.assert_frame_equal(dropped, expected)
8591
tm.assert_frame_equal(inp, expected)
92+
assert return_value is None
8693

8794
dropped = df.dropna(axis=0, thresh=4)
8895
expected = df.loc[range(2, 6)]
8996
inp = df.copy()
90-
inp.dropna(axis=0, thresh=4, inplace=True)
97+
return_value = inp.dropna(axis=0, thresh=4, inplace=True)
9198
tm.assert_frame_equal(dropped, expected)
9299
tm.assert_frame_equal(inp, expected)
100+
assert return_value is None
93101

94102
dropped = df.dropna(axis=1, thresh=4)
95103
tm.assert_frame_equal(dropped, df)
@@ -100,9 +108,10 @@ def test_dropna(self):
100108
# subset
101109
dropped = df.dropna(axis=0, subset=[0, 1, 3])
102110
inp = df.copy()
103-
inp.dropna(axis=0, subset=[0, 1, 3], inplace=True)
111+
return_value = inp.dropna(axis=0, subset=[0, 1, 3], inplace=True)
104112
tm.assert_frame_equal(dropped, df)
105113
tm.assert_frame_equal(inp, df)
114+
assert return_value is None
106115

107116
# all
108117
dropped = df.dropna(axis=1, how="all")
@@ -126,12 +135,14 @@ def test_drop_and_dropna_caching(self):
126135
df2 = df.copy()
127136
df["A"].dropna()
128137
tm.assert_series_equal(df["A"], original)
129-
df["A"].dropna(inplace=True)
138+
return_value = df["A"].dropna(inplace=True)
130139
tm.assert_series_equal(df["A"], expected)
140+
assert return_value is None
131141
df2["A"].drop([1])
132142
tm.assert_series_equal(df2["A"], original)
133-
df2["A"].drop([1], inplace=True)
143+
return_value = df2["A"].drop([1], inplace=True)
134144
tm.assert_series_equal(df2["A"], original.drop([1]))
145+
assert return_value is None
135146

136147
def test_dropna_corner(self, float_frame):
137148
# bad input
@@ -251,8 +262,9 @@ def test_fillna_different_dtype(self):
251262
)
252263
tm.assert_frame_equal(result, expected)
253264

254-
df.fillna({2: "foo"}, inplace=True)
265+
return_value = df.fillna({2: "foo"}, inplace=True)
255266
tm.assert_frame_equal(df, expected)
267+
assert return_value is None
256268

257269
def test_fillna_limit_and_value(self):
258270
# limit and value

0 commit comments

Comments
 (0)