Skip to content

Commit 9827ce0

Browse files
authored
Tst verify return none in tests/frame (pandas-dev#35232)
1 parent 37662ea commit 9827ce0

17 files changed

+230
-97
lines changed

pandas/tests/frame/indexing/test_categorical.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -326,7 +326,10 @@ def test_assigning_ops(self):
326326
df = DataFrame({"cats": catsf, "values": valuesf}, index=idxf)
327327

328328
exp_fancy = exp_multi_row.copy()
329-
exp_fancy["cats"].cat.set_categories(["a", "b", "c"], inplace=True)
329+
return_value = exp_fancy["cats"].cat.set_categories(
330+
["a", "b", "c"], inplace=True
331+
)
332+
assert return_value is None
330333

331334
df[df["cats"] == "c"] = ["b", 2]
332335
# category c is kept in .categories

pandas/tests/frame/indexing/test_mask.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -36,12 +36,14 @@ def test_mask_inplace(self):
3636

3737
rdf = df.copy()
3838

39-
rdf.where(cond, inplace=True)
39+
return_value = rdf.where(cond, inplace=True)
40+
assert return_value is None
4041
tm.assert_frame_equal(rdf, df.where(cond))
4142
tm.assert_frame_equal(rdf, df.mask(~cond))
4243

4344
rdf = df.copy()
44-
rdf.where(cond, -df, inplace=True)
45+
return_value = rdf.where(cond, -df, inplace=True)
46+
assert return_value is None
4547
tm.assert_frame_equal(rdf, df.where(cond, -df))
4648
tm.assert_frame_equal(rdf, df.mask(~cond, -df))
4749

pandas/tests/frame/indexing/test_where.py

+28-14
Original file line numberDiff line numberDiff line change
@@ -162,7 +162,8 @@ def _check_set(df, cond, check_dtypes=True):
162162
econd = cond.reindex_like(df).fillna(True)
163163
expected = dfi.mask(~econd)
164164

165-
dfi.where(cond, np.nan, inplace=True)
165+
return_value = dfi.where(cond, np.nan, inplace=True)
166+
assert return_value is None
166167
tm.assert_frame_equal(dfi, expected)
167168

168169
# dtypes (and confirm upcasts)x
@@ -303,7 +304,8 @@ def test_where_bug(self):
303304
tm.assert_frame_equal(result, expected)
304305

305306
result = df.copy()
306-
result.where(result > 2, np.nan, inplace=True)
307+
return_value = result.where(result > 2, np.nan, inplace=True)
308+
assert return_value is None
307309
tm.assert_frame_equal(result, expected)
308310

309311
def test_where_bug_mixed(self, sint_dtype):
@@ -324,7 +326,8 @@ def test_where_bug_mixed(self, sint_dtype):
324326
tm.assert_frame_equal(result, expected)
325327

326328
result = df.copy()
327-
result.where(result > 2, np.nan, inplace=True)
329+
return_value = result.where(result > 2, np.nan, inplace=True)
330+
assert return_value is None
328331
tm.assert_frame_equal(result, expected)
329332

330333
def test_where_bug_transposition(self):
@@ -417,7 +420,8 @@ def create():
417420
result = df.where(pd.notna(df), df.mean(), axis="columns")
418421
tm.assert_frame_equal(result, expected)
419422

420-
df.where(pd.notna(df), df.mean(), inplace=True, axis="columns")
423+
return_value = df.where(pd.notna(df), df.mean(), inplace=True, axis="columns")
424+
assert return_value is None
421425
tm.assert_frame_equal(df, expected)
422426

423427
df = create().fillna(0)
@@ -453,15 +457,17 @@ def test_where_axis(self):
453457
tm.assert_frame_equal(result, expected)
454458

455459
result = df.copy()
456-
result.where(mask, s, axis="index", inplace=True)
460+
return_value = result.where(mask, s, axis="index", inplace=True)
461+
assert return_value is None
457462
tm.assert_frame_equal(result, expected)
458463

459464
expected = DataFrame([[0, 1], [0, 1]], dtype="float64")
460465
result = df.where(mask, s, axis="columns")
461466
tm.assert_frame_equal(result, expected)
462467

463468
result = df.copy()
464-
result.where(mask, s, axis="columns", inplace=True)
469+
return_value = result.where(mask, s, axis="columns", inplace=True)
470+
assert return_value is None
465471
tm.assert_frame_equal(result, expected)
466472

467473
# Upcast needed
@@ -474,7 +480,8 @@ def test_where_axis(self):
474480
tm.assert_frame_equal(result, expected)
475481

476482
result = df.copy()
477-
result.where(mask, s, axis="index", inplace=True)
483+
return_value = result.where(mask, s, axis="index", inplace=True)
484+
assert return_value is None
478485
tm.assert_frame_equal(result, expected)
479486

480487
expected = DataFrame([[0, np.nan], [0, np.nan]])
@@ -488,7 +495,8 @@ def test_where_axis(self):
488495
}
489496
)
490497
result = df.copy()
491-
result.where(mask, s, axis="columns", inplace=True)
498+
return_value = result.where(mask, s, axis="columns", inplace=True)
499+
assert return_value is None
492500
tm.assert_frame_equal(result, expected)
493501

494502
# Multiple dtypes (=> multiple Blocks)
@@ -511,7 +519,8 @@ def test_where_axis(self):
511519
tm.assert_frame_equal(result, expected)
512520

513521
result = df.copy()
514-
result.where(mask, s1, axis="columns", inplace=True)
522+
return_value = result.where(mask, s1, axis="columns", inplace=True)
523+
assert return_value is None
515524
tm.assert_frame_equal(result, expected)
516525

517526
result = df.where(mask, s2, axis="index")
@@ -521,7 +530,8 @@ def test_where_axis(self):
521530
tm.assert_frame_equal(result, expected)
522531

523532
result = df.copy()
524-
result.where(mask, s2, axis="index", inplace=True)
533+
return_value = result.where(mask, s2, axis="index", inplace=True)
534+
assert return_value is None
525535
tm.assert_frame_equal(result, expected)
526536

527537
# DataFrame vs DataFrame
@@ -534,10 +544,12 @@ def test_where_axis(self):
534544
result = df.where(mask, d1, axis="index")
535545
tm.assert_frame_equal(result, expected)
536546
result = df.copy()
537-
result.where(mask, d1, inplace=True)
547+
return_value = result.where(mask, d1, inplace=True)
548+
assert return_value is None
538549
tm.assert_frame_equal(result, expected)
539550
result = df.copy()
540-
result.where(mask, d1, inplace=True, axis="index")
551+
return_value = result.where(mask, d1, inplace=True, axis="index")
552+
assert return_value is None
541553
tm.assert_frame_equal(result, expected)
542554

543555
d2 = df.copy().drop(1, axis=1)
@@ -549,10 +561,12 @@ def test_where_axis(self):
549561
result = df.where(mask, d2, axis="columns")
550562
tm.assert_frame_equal(result, expected)
551563
result = df.copy()
552-
result.where(mask, d2, inplace=True)
564+
return_value = result.where(mask, d2, inplace=True)
565+
assert return_value is None
553566
tm.assert_frame_equal(result, expected)
554567
result = df.copy()
555-
result.where(mask, d2, inplace=True, axis="columns")
568+
return_value = result.where(mask, d2, inplace=True, axis="columns")
569+
assert return_value is None
556570
tm.assert_frame_equal(result, expected)
557571

558572
def test_where_callable(self):

pandas/tests/frame/methods/test_clip.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,8 @@ def test_inplace_clip(self, float_frame):
2222
median = float_frame.median().median()
2323
frame_copy = float_frame.copy()
2424

25-
frame_copy.clip(upper=median, lower=median, inplace=True)
25+
return_value = frame_copy.clip(upper=median, lower=median, inplace=True)
26+
assert return_value is None
2627
assert not (frame_copy.values != median).any()
2728

2829
def test_dataframe_clip(self):

pandas/tests/frame/methods/test_drop.py

+6-3
Original file line numberDiff line numberDiff line change
@@ -70,8 +70,10 @@ def test_drop_names(self):
7070
df_dropped_b = df.drop("b")
7171
df_dropped_e = df.drop("e", axis=1)
7272
df_inplace_b, df_inplace_e = df.copy(), df.copy()
73-
df_inplace_b.drop("b", inplace=True)
74-
df_inplace_e.drop("e", axis=1, inplace=True)
73+
return_value = df_inplace_b.drop("b", inplace=True)
74+
assert return_value is None
75+
return_value = df_inplace_e.drop("e", axis=1, inplace=True)
76+
assert return_value is None
7577
for obj in (df_dropped_b, df_dropped_e, df_inplace_b, df_inplace_e):
7678
assert obj.index.name == "first"
7779
assert obj.columns.name == "second"
@@ -148,7 +150,8 @@ def test_drop(self):
148150
# GH#5628
149151
df = pd.DataFrame(np.random.randn(10, 3), columns=list("abc"))
150152
expected = df[~(df.b > 0)]
151-
df.drop(labels=df[df.b > 0].index, inplace=True)
153+
return_value = df.drop(labels=df[df.b > 0].index, inplace=True)
154+
assert return_value is None
152155
tm.assert_frame_equal(df, expected)
153156

154157
def test_drop_multiindex_not_lexsorted(self):

pandas/tests/frame/methods/test_interpolate.py

+8-4
Original file line numberDiff line numberDiff line change
@@ -246,11 +246,13 @@ def test_interp_inplace(self):
246246
df = DataFrame({"a": [1.0, 2.0, np.nan, 4.0]})
247247
expected = DataFrame({"a": [1.0, 2.0, 3.0, 4.0]})
248248
result = df.copy()
249-
result["a"].interpolate(inplace=True)
249+
return_value = result["a"].interpolate(inplace=True)
250+
assert return_value is None
250251
tm.assert_frame_equal(result, expected)
251252

252253
result = df.copy()
253-
result["a"].interpolate(inplace=True, downcast="infer")
254+
return_value = result["a"].interpolate(inplace=True, downcast="infer")
255+
assert return_value is None
254256
tm.assert_frame_equal(result, expected.astype("int64"))
255257

256258
def test_interp_inplace_row(self):
@@ -259,7 +261,8 @@ def test_interp_inplace_row(self):
259261
{"a": [1.0, 2.0, 3.0, 4.0], "b": [np.nan, 2.0, 3.0, 4.0], "c": [3, 2, 2, 2]}
260262
)
261263
expected = result.interpolate(method="linear", axis=1, inplace=False)
262-
result.interpolate(method="linear", axis=1, inplace=True)
264+
return_value = result.interpolate(method="linear", axis=1, inplace=True)
265+
assert return_value is None
263266
tm.assert_frame_equal(result, expected)
264267

265268
def test_interp_ignore_all_good(self):
@@ -297,7 +300,8 @@ def test_interp_time_inplace_axis(self, axis):
297300
expected = DataFrame(index=idx, columns=idx, data=data)
298301

299302
result = expected.interpolate(axis=0, method="time")
300-
expected.interpolate(axis=0, method="time", inplace=True)
303+
return_value = expected.interpolate(axis=0, method="time", inplace=True)
304+
assert return_value is None
301305
tm.assert_frame_equal(result, expected)
302306

303307
@pytest.mark.parametrize("axis_name, axis_number", [("index", 0), ("columns", 1)])

pandas/tests/frame/methods/test_rename.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -150,7 +150,8 @@ def test_rename_inplace(self, float_frame):
150150

151151
c_id = id(float_frame["C"])
152152
float_frame = float_frame.copy()
153-
float_frame.rename(columns={"C": "foo"}, inplace=True)
153+
return_value = float_frame.rename(columns={"C": "foo"}, inplace=True)
154+
assert return_value is None
154155

155156
assert "C" not in float_frame
156157
assert "foo" in float_frame

pandas/tests/frame/methods/test_rename_axis.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -10,14 +10,16 @@ def test_rename_axis_inplace(self, float_frame):
1010
# GH#15704
1111
expected = float_frame.rename_axis("foo")
1212
result = float_frame.copy()
13-
no_return = result.rename_axis("foo", inplace=True)
13+
return_value = no_return = result.rename_axis("foo", inplace=True)
14+
assert return_value is None
1415

1516
assert no_return is None
1617
tm.assert_frame_equal(result, expected)
1718

1819
expected = float_frame.rename_axis("bar", axis=1)
1920
result = float_frame.copy()
20-
no_return = result.rename_axis("bar", axis=1, inplace=True)
21+
return_value = no_return = result.rename_axis("bar", axis=1, inplace=True)
22+
assert return_value is None
2123

2224
assert no_return is None
2325
tm.assert_frame_equal(result, expected)

0 commit comments

Comments
 (0)